enhancement 新增graaljs脚本插件

This commit is contained in:
zendwang
2022-11-22 11:54:11 +08:00
parent d73f8269f3
commit 7956a3014c
58 changed files with 194 additions and 248 deletions

View File

@@ -0,0 +1,27 @@
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<parent>
<artifactId>liteflow-script-plugin</artifactId>
<groupId>com.yomahub</groupId>
<version>${revision}</version>
<relativePath>../pom.xml</relativePath>
</parent>
<modelVersion>4.0.0</modelVersion>
<artifactId>liteflow-script-graaljs</artifactId>
<dependencies>
<dependency>
<groupId>com.yomahub</groupId>
<artifactId>liteflow-core</artifactId>
<version>${revision}</version>
<scope>provided</scope>
</dependency>
<dependency>
<groupId>org.graalvm.js</groupId>
<artifactId>js</artifactId>
</dependency>
</dependencies>
</project>

View File

@@ -0,0 +1,105 @@
package com.yomahub.liteflow.script.graaljs;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.yomahub.liteflow.script.ScriptBeanManager;
import com.yomahub.liteflow.script.ScriptExecuteWrap;
import com.yomahub.liteflow.script.ScriptExecutor;
import com.yomahub.liteflow.script.exception.ScriptLoadException;
import com.yomahub.liteflow.slot.DataBus;
import com.yomahub.liteflow.slot.Slot;
import com.yomahub.liteflow.util.CopyOnWriteHashMap;
import org.graalvm.polyglot.Context;
import org.graalvm.polyglot.Value;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Map;
/**
* GraalVM JavaScript脚本语言的执行器实现
* @author zendwang
* @since 2.9.4
*/
public class GraalJavaScriptExecutor implements ScriptExecutor {
private final Logger log = LoggerFactory.getLogger(this.getClass());
private final Map<String, String> scriptMap = new CopyOnWriteHashMap<>();
@Override
public ScriptExecutor init() {
return this;
}
@Override
public void load(String nodeId, String script) {
try{
String wrapScript = StrUtil.format("function process(){{}} process();",script);
scriptMap.put(nodeId, 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 execute(ScriptExecuteWrap wrap) throws Exception{
try{
if (!scriptMap.containsKey(wrap.getNodeId())){
String errorMsg = StrUtil.format("script for node[{}] is not loaded", wrap.getNodeId());
throw new ScriptLoadException(errorMsg);
}
Context context = Context.newBuilder().allowAllAccess(true).build();
Value bindings = context.getBindings("js");
//往脚本语言绑定表里循环增加绑定上下文的key
//key的规则为自定义上下文的simpleName
//比如你的自定义上下文为AbcContext那么key就为:abcContext
//这里不统一放一个map的原因是考虑到有些用户会调用上下文里的方法而不是参数所以脚本语言的绑定表里也是放多个上下文
DataBus.getContextBeanList(wrap.getSlotIndex()).forEach(o -> {
String key = StrUtil.lowerFirst(o.getClass().getSimpleName());
bindings.putMember(key, o);
});
//把wrap对象转换成元数据map
Map<String, Object> metaMap = BeanUtil.beanToMap(wrap);
//在元数据里放入主Chain的流程参数
Slot slot = DataBus.getSlot(wrap.getSlotIndex());
metaMap.put("requestData", slot.getRequestData());
//如果有隐式流程,则放入隐式流程的流程参数
Object subRequestData = slot.getChainReqData(wrap.getCurrChainName());
if (ObjectUtil.isNotNull(subRequestData)){
metaMap.put("subRequestData", subRequestData);
}
//往脚本上下文里放入元数据
bindings.putMember("_meta", metaMap);
//放入用户自己定义的bean
ScriptBeanManager.getScriptBeanMap().entrySet().stream().forEach( e ->{
bindings.putMember(e.getKey(), e.getValue());
});
Value value = context.eval("js", 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){
log.error(e.getMessage(), e);
throw e;
}
}
@Override
public void cleanCache() {
scriptMap.clear();
}
}

View File

@@ -0,0 +1,2 @@
# JavaScript的实现
com.yomahub.liteflow.script.graaljs.GraalJavaScriptExecutor

View File

@@ -1,2 +1,2 @@
# JavaScript的实现
com.yomahub.liteflow.script.javascript.JavaScriptExecutor
com.yomahub.liteflow.script.graaljs.GraalJavaScriptExecutor

View File

@@ -18,6 +18,7 @@
<module>liteflow-script-qlexpress</module>
<module>liteflow-script-groovy</module>
<module>liteflow-script-javascript</module>
<module>liteflow-script-graaljs</module>
</modules>
</project>