com.yomahub.liteflow.script.graaljs.GraalJavaScriptExecutor Maven / Gradle / Ivy
package com.yomahub.liteflow.script.graaljs;
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.exception.ScriptLoadException;
import com.yomahub.liteflow.util.CopyOnWriteHashMap;
import org.graalvm.polyglot.Context;
import org.graalvm.polyglot.Engine;
import org.graalvm.polyglot.Source;
import org.graalvm.polyglot.Value;
import java.util.Map;
/**
* GraalVM JavaScript脚本语言的执行器实现
*
* @author zendwang
* @since 2.9.4
*/
public class GraalJavaScriptExecutor extends ScriptExecutor {
private final Map scriptMap = new CopyOnWriteHashMap<>();
private Engine engine;
@Override
public ScriptExecutor init() {
engine = Engine.create();
return this;
}
@Override
public void load(String nodeId, String script) {
try {
String wrapScript = StrUtil.format("function process(){{}} process();", script);
scriptMap.put(nodeId, Source.create("js", wrapScript));
}
catch (Exception e) {
String errorMsg = StrUtil.format("script loading error for node[{}], error msg:{}", nodeId, e.getMessage());
throw new ScriptLoadException(errorMsg);
}
}
@Override
public Object executeScript(ScriptExecuteWrap wrap) {
if (!scriptMap.containsKey(wrap.getNodeId())) {
String errorMsg = StrUtil.format("script for node[{}] is not loaded", wrap.getNodeId());
throw new ScriptLoadException(errorMsg);
}
try (Context context = Context.newBuilder().allowAllAccess(true).engine(this.engine).build()) {
Value bindings = context.getBindings("js");
bindParam(wrap, bindings::putMember, (s, o) -> {
if (!bindings.hasMember(s)) {
bindings.putMember(s, o);
}
});
Value value = context.eval(scriptMap.get(wrap.getNodeId()));
if (value.isBoolean()) {
return value.asBoolean();
}
else if (value.isNumber()) {
return value.asInt();
}
else if (value.isString()) {
return value.asString();
}
return value;
}
catch (Exception e) {
throw e;
}
}
@Override
public void cleanCache() {
scriptMap.clear();
}
@Override
public ScriptTypeEnum scriptType() {
return ScriptTypeEnum.JS;
}
}