Repository navigation
Expand file tree
/
Copy pathJSR223ScriptExecutor.java
More file actions
105 lines (86 loc) · 3.59 KB
/
Copy pathJSR223ScriptExecutor.java
File metadata and controls
105 lines (86 loc) · 3.59 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package apijson.orm.script;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import javax.script.Bindings;
import javax.script.Compilable;
import javax.script.CompiledScript;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.SimpleBindings;
import apijson.Log;
import apijson.orm.AbstractFunctionParser;
/**
* JSR223 script engine的统一实现抽象类
*/
public abstract class JSR223ScriptExecutor<T, M extends Map<String, Object>, L extends List<Object>> implements ScriptExecutor<T, M, L> {
private static final String TAG = "JSR223ScriptExecutor";
protected ScriptEngine scriptEngine;
private final Map<String, CompiledScript> compiledScriptMap = new ConcurrentHashMap<>();
@Override
public ScriptExecutor<T, M, L> init() {
scriptEngine = createScriptEngine();
return this;
}
protected ScriptEngine createScriptEngine() {
String name = scriptEngineName();
if ("nashorn".equalsIgnoreCase(name) || "javascript".equalsIgnoreCase(name)
|| "js".equalsIgnoreCase(name) || "ecmascript".equalsIgnoreCase(name)) {
try {
Class<?> factoryClass = Class.forName("jdk.nashorn.api.scripting.NashornScriptEngineFactory");
Class<?> filterClass = Class.forName("jdk.nashorn.api.scripting.ClassFilter");
Object filter = java.lang.reflect.Proxy.newProxyInstance(
filterClass.getClassLoader(),
new Class<?>[]{filterClass},
(proxy, method, methodArgs) -> isClassExposureAllowed((String) methodArgs[0]));
Object factory = factoryClass.getDeclaredConstructor().newInstance();
return (ScriptEngine) factoryClass.getMethod("getScriptEngine", filterClass).invoke(factory, filter);
} catch (Throwable e) {
Log.e(TAG, "create sandboxed Nashorn engine failed, falling back: " + e);
}
}
return new ScriptEngineManager().getEngineByName(name);
}
protected boolean isClassExposureAllowed(String className) {
return false;
}
protected abstract String scriptEngineName();
protected abstract Object extendParameter(AbstractFunctionParser<T, M, L> parser, Map<String, Object> currentObject, String methodName, Object[] args);
protected abstract boolean isLockScript(String methodName);
protected String convertScript(String script) {
return script;
}
@Override
public void load(String name, String script) {
try {
CompiledScript compiledScript = ((Compilable) scriptEngine).compile(convertScript(script));
compiledScriptMap.put(name, compiledScript);
} catch (Exception e) {
Log.e(TAG, "compile script failed: " + name, e);
}
}
@Override
public Object execute(AbstractFunctionParser<T, M, L> parser, Map<String, Object> currentObject, String methodName, Object[] args) throws Exception {
CompiledScript compiledScript = compiledScriptMap.get(methodName);
Bindings bindings = new SimpleBindings();
// 往脚本上下文里放入元数据
// 把 RequestMethod method, String tag, int version, @NotNull JSONMap request,
// HttpSession session 等参数作为全局参数传进去供脚本使用
// 加载扩展属性
Object extendParameter = this.extendParameter(parser, currentObject, methodName, args);
if(extendParameter != null) {
bindings.put("extParam", extendParameter);
}
Map<String, Object> metaMap = new HashMap<>();
metaMap.put("version", parser == null ? 0 : parser.getVersion());
metaMap.put("tag", parser == null ? null : parser.getTag());
metaMap.put("args", args);
bindings.put("_meta", metaMap);
return compiledScript.eval(bindings);
}
@Override
public void cleanCache() {
compiledScriptMap.clear();
}
}