Repository navigation
Expand file tree
/
Copy pathJSR223ScriptExecutor.java
More file actions
80 lines (63 loc) · 2.44 KB
/
Copy pathJSR223ScriptExecutor.java
File metadata and controls
80 lines (63 loc) · 2.44 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
package apijson.orm.script;
import java.util.HashMap;
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 com.alibaba.fastjson.JSONObject;
import apijson.orm.AbstractFunctionParser;
/**
* JSR223 script engine的统一实现抽象类
*/
public abstract class JSR223ScriptExecutor implements ScriptExecutor {
protected ScriptEngine scriptEngine;
private final Map<String, CompiledScript> compiledScriptMap = new ConcurrentHashMap<>();
@Override
public ScriptExecutor init() {
ScriptEngineManager scriptEngineManager = new ScriptEngineManager();
scriptEngine = scriptEngineManager.getEngineByName(scriptEngineName());
return this;
}
protected abstract String scriptEngineName();
protected abstract Object extendParameter(AbstractFunctionParser parser, JSONObject 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) {
e.printStackTrace();
}
}
@Override
public Object execute(AbstractFunctionParser parser, JSONObject currentObject, String methodName, Object[] args) throws Exception {
CompiledScript compiledScript = compiledScriptMap.get(methodName);
Bindings bindings = new SimpleBindings();
// 往脚本上下文里放入元数据
// 把 RequestMethod method, String tag, int version, @NotNull JSONObject 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();
}
}