Java ScriptEngine 经验分享 - 1.nashorn的适配

先来一段代码，把脚本引擎实例化出来

ScriptEngineManager scriptEngineManager = new ScriptEngineManager();
ScriptEngine  _eng = scriptEngineManager.getEngineByName("nashorn");
Invocable _eng_call = (Invocable)_eng;

一、传参的方式

• 传参方式一（和脚本拼到一起。这种只能传普通类型）

//定义函数
_eng.eval("function fun1(arg){}");

//调用
String arg = "test"; 
_eng.eval("fun1('"+arg+"')");

• 传参方式二（通过Invocable，调用函数传参。可以传一切）

//定义函数
_eng.eval("function fun1(arg){}");

//调用
XContext ctx = XContext.current();
_eng_call.invokeFunction("fun1",ctx);

• 传参方式三（通过ThreadLocal，应该算是新套路）

//XContext.current()，是一个从ThreadLocal取值的函数

//定义函数
_eng.eval("function fun1(arg){}");
_eng.eval("function fun2(){ fun1(XContext.current()); /*...*/ }");

//调用
_eng_call.invokeFunction("fun2");

/**
 比如，http 上下文我们就可以这么搞
*/

方式二和方式三较好。

二、为跨文件调用做处理

像node.js一样，加个require接口，供相互调用

//此处只讲意图，__JTEAPI具体看后面
_eng.eval("function require(path){__JTEAPI.require(path);return __global.lib[path]}")

三、添加mvc支持（或视图支持）

//此处只讲意图，__JTEAPI具体看后面
_eng.eval("function modelAndView(tml,mod){return __JTEAPI.modelAndView(tml,mod);};");

四、更好的处理返回结果（json 或 具体的值）

• 问题1：js 引擎的复杂对象会被转换，类型不明JSON化处理不方便
• 问题2：先用js转好再返回。但是JSON.stringify，不能转java的复杂类型
• 本案采用混合方式处理。在js代码里转换，同时为JSON.stringify 提供 java类型转换支持

设计一个协助处理的函数

_eng.eval("function stringify_java(k,v){if(v){if(v.getTicks){return v.getTicks()}if(v.getTime){return v.getTime()}if(v.putAll){var obj={};v.forEach(function(k2,v2){obj[k2]=v2});return obj}if(v.addAll){var ary=[];v.forEach(function(v2){ary.push(v2)});return ary}}return v};");

设计一个执行代理，由它调用具体代码并按情况转换

_eng.eval("function API_RUN(api){var rst=api(XContext.current());if(rst){if(typeof(rst)=='object'){return JSON.stringify(rst,stringify_java)}else{return rst}}else{return null}};");

用这两个函数进行代码执行处理

Object api = _eng.get("API_"+name2);
Object tmp = _eng_call.invokeFunction("API_RUN",api);

if (tmp == null) {
    return null;
} else {
    return tmp.toString();
}

附：一份完整的 nashorn 的适配示例

此示例为嵌入式FaaS引擎 SolonJT 的一个执行器的适配

public class JavascriptJtExecutor implements IJtExecutor {
    private static final ThData _tlBuilder = new ThData(new StringBuilder(1024*5));
    private static final String _lock ="";
    private static JavascriptJtExecutor _g;
    public static JavascriptJtExecutor singleton(){
        if(_g == null){
            synchronized (_lock){
                if(_g == null){
                    _g = new JavascriptJtExecutor();
                }
            }
        }

        return _g;
    }


    private final ScriptEngine _eng;
    private final Invocable    _eng_call;
    private final Set  _loaded_names;

    private JavascriptJtExecutor(){
        _loaded_names = Collections.synchronizedSet(new HashSet<>());

        ScriptEngineManager scriptEngineManager = new ScriptEngineManager();
        _eng = scriptEngineManager.getEngineByName("nashorn");
        _eng_call = (Invocable)_eng;

        XApp.global().shared().forEach((k, v)->{
            sharedSet(k, v);
        });

        XApp.global().onSharedAdd((k,v)->{
            sharedSet(k, v);
        });

        sharedSet("__JTEAPI", new __JTEAPI_CLZ());

        try {
            StringBuilder sb = new StringBuilder();

            sb.append("var __global={lib:{}};");

            sb.append("Date.prototype.toJSON =function(){ return this.getTime()};");

            sb.append("var XContext = Java.type('org.noear.solon.core.XContext');");
            sb.append("var ONode = Java.type('org.noear.snack.ONode');");

            sb.append("var Datetime = Java.type('org.noear.solonjt.utils.Datetime');");
            sb.append("var Timecount = Java.type('org.noear.solonjt.utils.Timecount');");
            sb.append("var Timespan = Java.type('org.noear.solonjt.utils.Timespan');");

            sb.append("function modelAndView(tml,mod){return __JTEAPI.modelAndView(tml,mod);};");

            sb.append("function require(path){__JTEAPI.require(path);return __global.lib[path]}");

            //为JSON.stringify 添加java的对象处理

            sb.append("function stringify_java(k,v){if(v){if(v.getTicks){return v.getTicks()}if(v.getTime){return v.getTime()}if(v.putAll){var obj={};v.forEach(function(k2,v2){obj[k2]=v2});return obj}if(v.addAll){var ary=[];v.forEach(function(v2){ary.push(v2)});return ary}}return v};");

            sb.append("function API_RUN(api){var rst=api(XContext.current());if(rst){if(typeof(rst)=='object'){return JSON.stringify(rst,stringify_java)}else{return rst}}else{return null}};");

            _eng.eval(sb.toString());

        } catch (Exception ex) {
            ex.printStackTrace();
        }



    }

    public void sharedSet(String name,Object val){
        _eng.put(name, val);
    }

    //
    // IJtEngine 接口
    //
    @Override
    public String language() {
        return "javascript";
    }

    @Override
    public boolean isLoaded(String name2) {
        return _loaded_names.contains(name2);
    }

    @Override
    public boolean preLoad(String name2, AFileModel file) throws Exception {
        if (isLoaded(name2) == false) {
            _loaded_names.add(name2);

            _eng.eval(compilerAsFun(name2, file));
        }

        return true;
    }

    @Override
    public  void del(String name) {
        String name2 = name.replace(".", "_").replace("*","_");
        _loaded_names.remove(name2);
        _loaded_names.remove(name2 + "__lib");
    }

    @Override
    public  void delAll() {
        _loaded_names.clear();
    }

    @Override
    public  Object exec(String name, AFileModel file, XContext ctx, Map model, boolean outString) throws Exception {
        String name2 = name.replace(".","_").replace("*","_");

        preLoad(name2, file);

        if(outString){
            Object api = _eng.get("API_"+name2);
            Object tmp = _eng_call.invokeFunction("API_RUN",api);

            if (tmp == null) {
                return null;
            } else {
                return tmp.toString();
            }
        }else{
            return _eng_call.invokeFunction("API_"+name2, ctx);
        }
    }



    //////////////////////////////////////////////////////////////////


    /**
     * 编译为函数代码
     * */
    public  String compilerAsFun(String name, AFileModel file) {
        StringBuilder sb = _tlBuilder.get();
        sb.setLength(0);

        sb.append("this.API_").append(name).append("=function(ctx){");
        sb.append("\r\n\r\n");
        sb.append(file.content);
        sb.append("\r\n\r\n};");

        if (name.endsWith("__lib")) {
            sb.append("__global.lib['")
                    .append(file.path)
                    .append("']=")
                    .append("new API_")
                    .append(name)
                    .append("();");
        }

        return sb.toString();
    }
}



public class __JTEAPI_CLZ {

    public String serialize_java(Object obj) throws Exception{
        return ONode.serialize(obj);
    }

    public String require(String path) throws Exception {
        String name = path.replace("/", "__");
        String name2 = name.replace(".", "_") + "__lib";

        AFileModel file = ExecutorFactory.fileGet(path);

        JavascriptJtExecutor.singleton().preLoad(name2, file);

        return name2;
    }

    public Object modelAndView(String path, Map model) throws Exception {
        String path2 = path;//AFileUtil.path2(path);//不用转为*
        String name = path2.replace("/", "__");

        AFileModel file = ExecutorFactory.fileGet(path2);

        if (file.file_id > 0) {
            return ExecutorFactory.call(name,file, XContext.current(),model,true);
        } else {
            return "";
        }
    }
}