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();

參考鏈接：http://docs.codehaus.org/display/GROOVY/Embedding+Groovy

當然對底層服務的封裝就根據不同業務系統有所不同，只要底層服務封裝的好，那么在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);  //自定義FileManager

List<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;
    }
}

參考鏈接: http://www.javablogging.com/dynamic-in-memory-compilation/