|
@@ -1,9 +1,11 @@
|
|
|
package com.yomahub.liteflow.script.java;
|
|
|
|
|
|
+import cn.hutool.core.util.ReUtil;
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
import com.yomahub.liteflow.enums.ScriptTypeEnum;
|
|
|
import com.yomahub.liteflow.script.ScriptExecuteWrap;
|
|
|
import com.yomahub.liteflow.script.ScriptExecutor;
|
|
|
+import com.yomahub.liteflow.script.body.JaninoCommonScriptBody;
|
|
|
import com.yomahub.liteflow.script.exception.ScriptLoadException;
|
|
|
import com.yomahub.liteflow.util.CopyOnWriteHashMap;
|
|
|
import org.codehaus.commons.compiler.CompilerFactoryFactory;
|
|
@@ -14,13 +16,15 @@ public class JavaExecutor extends ScriptExecutor {
|
|
|
|
|
|
private final Map<String, IScriptEvaluator> compiledScriptMap = new CopyOnWriteHashMap<>();
|
|
|
|
|
|
+
|
|
|
+
|
|
|
@Override
|
|
|
public void load(String nodeId, String script) {
|
|
|
try{
|
|
|
IScriptEvaluator se = CompilerFactoryFactory.getDefaultCompilerFactory(this.getClass().getClassLoader()).newScriptEvaluator();
|
|
|
se.setReturnType(Object.class);
|
|
|
se.setParameters(new String[] {"_meta"}, new Class[] {ScriptExecuteWrap.class});
|
|
|
- se.cook(script);
|
|
|
+ se.cook(convertScript(script));
|
|
|
compiledScriptMap.put(nodeId, se);
|
|
|
}catch (Exception e){
|
|
|
String errorMsg = StrUtil.format("script loading error for node[{}],error msg:{}", nodeId, e.getMessage());
|
|
@@ -48,4 +52,23 @@ public class JavaExecutor extends ScriptExecutor {
|
|
|
public ScriptTypeEnum scriptType() {
|
|
|
return ScriptTypeEnum.JAVA;
|
|
|
}
|
|
|
+
|
|
|
+ private String convertScript(String script){
|
|
|
+ //替换掉public,private,protected等修饰词
|
|
|
+ String script1 = script.replaceAll("public class", "class")
|
|
|
+ .replaceAll("private class", "class")
|
|
|
+ .replaceAll("protected class", "class");
|
|
|
+
|
|
|
+ //分析出class的具体名称
|
|
|
+ String className = ReUtil.getGroup1("class\\s+(\\w+)\\s+implements", script1);
|
|
|
+
|
|
|
+ if (StrUtil.isBlank(className)){
|
|
|
+ throw new RuntimeException("cannot find class defined");
|
|
|
+ }
|
|
|
+
|
|
|
+ return "import com.yomahub.liteflow.script.body.JaninoCommonScriptBody;\n" +
|
|
|
+ script1 + "\n" +
|
|
|
+ StrUtil.format("{} item = new {}();\n", className, className) +
|
|
|
+ "if (item instanceof JaninoCommonScriptBody){item.body(_meta);return null;}else{return item.body(_meta);}";
|
|
|
+ }
|
|
|
}
|