Groovy&Java动态编译执行

2017-06-23  本文已影响3278人  wangliang938

http://www.youzitool.com/blog/15.html
工作中,遇到部分业务经常动态变化,或者在不发布系统的前提下,对业务规则进行调整。那么可以将这部分业务逻辑改写成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;
    }
}
上一篇下一篇

猜你喜欢

热点阅读