1. 程式人生 > >Groovy&Java動態編譯執行

Groovy&Java動態編譯執行

工作中,遇到部分業務經常動態變化,或者在不釋出系統的前提下,對業務規則進行調整。那麼可以將這部分業務邏輯改寫成Groovy指令碼來執行,那麼就可以在業務執行過程中動態更改業務規則,達到快速響應。

Case1: Groovy動態編譯執行

閒話少說,直接上程式碼:

static Compilable engine;  //獲取groovy編譯類	static {		ScriptEngineManager manager = new ScriptEngineManager();		engine = (Compilable) manager.getEngineByName("groovy");	}public static Object execute(String scriptText, Map<?, ?> context) {   CompiledScript script = engine.compile(scriptText); // 這個地方需要使用快取,達到編譯一次,多次執行。if (script != null) {	  ScriptContext scriptContext = new SimpleScriptContext();	  Set<?> its = context.entrySet();   //上下文引數	  for(Object o:its){		@SuppressWarnings("rawtypes")		Entry entry = (Entry)o;		scriptContext.setAttribute(		LangUtil.getString(entry.getKey()),	    entry.getValue(), ScriptContext.ENGINE_SCOPE); //copy到groovy執行環境上下文	 }	return script.eval(scriptContext);  //執行groovy指令碼}

那麼腳本里面就可以使用很靈活的規則,如果需要使用bean,需要在上文下引數將bean工作傳遞進去。

以上是使用java的ScriptEngineManager來動態編譯、執行Groovy。

如果你覺得這種執行方式效率比較低,那麼還可以使用groovy的ClassLoade將其編譯成物件來執行。

ClassLoader parent = getClass().getClassLoader();GroovyClassLoader loader = new GroovyClassLoader(parent);Class groovyClass = loader.parseClass(new File("src/test/groovy/script/HelloWorld.groovy")); //也可以直接使用String(指令碼內容)GroovyObject groovyObject = (GroovyObject) groovyClass.newInstance();Object[] args = {};groovyObject.invokeMethod("run", args);如果你希望這個groovy指令碼是嵌入在自己的抽象類中執行的,那麼也可以使用如下方法:GroovyClassLoader gcl = new GroovyClassLoader();Class clazz = gcl.parseClass(myStringwithGroovyClassSource, "SomeName.groovy");Object aScript = clazz.newInstance();MyInterface myObject = (MyInterface) aScript;myObject.interfaceMethod();

當然對底層服務的封裝就根據不同業務系統有所不同,只要底層服務封裝的好,那麼在groovy腳本里就可以只使用一套封裝之後的API來完成各種業務實現,避免了各種直接呼叫bean的情況出現。

Case2: Java動態編譯執行

以上執行方式已經在生產環境得到驗證,但最近壓測發現,不管使用Case1中的哪種方法執行的QPS都沒有直接使用Java程式碼的執行效率高。那麼是否可以將Java程式碼動態編譯。結果當然是suer,在JDK6及以上中已經支援了java程式碼動態編譯(JavaCompiler)

Java的動態編譯對源提供了多個渠道。可以是字串,可以是文字檔案,也可以是編譯過的位元組碼檔案(.class檔案),甚至可以是存放在資料庫中的明文程式碼或是位元組碼。彙總成一句話,只要是符合Java規範的就都可以在執行期動態載入,其實現方式就是實現JavaFileObject介面,重寫getCharContent、openInputStream、openOutputStream,或者實現JDK已經提供的兩個SimpleJavaFileObject、ForwardingJavaFileObject

public void compileClass(String name, String code) {private final JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();private final FileManager manager = new FileManager(this.compiler);  //自定義FileManagerList<Source> list = new ArrayList<Source>();        list.add(new Source(name, JavaFileObject.Kind.SOURCE, code)); //輸入StringWriter stringWriter = new StringWriter();        this.compiler.getTask(stringWriter, this.manager, null, null, null, list).call();Output mc = this.manager.getOutput(name);    //輸出if (mc != null) {            byte[] array = mc.toByteArray();  //轉換成byte[]            return defineClass(name, array, 0, array.length);  //需要繼承ClassLoader}}
public class FileManager extends ForwardingJavaFileManager<StandardJavaFileManager> {    protected final Map<String, Output> map = new HashMap<String, Output>();    public FileManager(JavaCompiler compiler) {        super(compiler.getStandardFileManager(null, null, null));  }  //取得檔案管理器    @Override    public Output getJavaFileForOutput            (Location location, String name, JavaFileObject.Kind kind, FileObject source) {        Output mc = new Output(name, kind);   //與檔案連線        this.map.put(name, mc);        return mc;    }    public Output getOutput(String name) {        return map.get(name);    }}
public class Output extends SimpleJavaFileObject {    private final ByteArrayOutputStream baos = new ByteArrayOutputStream();    public Output(String name, Kind kind) {        super(URI.create("dynamic:///" + name.replace('.', '/') + kind.extension), kind); } //URI    byte[] toByteArray() {  //外部呼叫,生成Class        return this.baos.toByteArray();    }    @Override    public ByteArrayOutputStream openOutputStream() {        return this.baos;    }}public class Source extends SimpleJavaFileObject {    private final String content;    public Source(String name, Kind kind, String content) {        super(URI.create("dynamic:///" + name.replace('.', '/') + kind.extension), kind);        this.content = content;    }    @Override    public CharSequence getCharContent(boolean ignore) {        return this.content;    }}