mirror of
https://gitee.com/dromara/liteFlow.git
synced 2026-05-14 20:22:07 +08:00
enhancement 新增graaljs脚本插件
This commit is contained in:
27
liteflow-script-plugin/liteflow-script-graaljs/pom.xml
Normal file
27
liteflow-script-plugin/liteflow-script-graaljs/pom.xml
Normal 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>
|
||||
@@ -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();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,2 @@
|
||||
# JavaScript的实现
|
||||
com.yomahub.liteflow.script.graaljs.GraalJavaScriptExecutor
|
||||
@@ -1,2 +1,2 @@
|
||||
# JavaScript的实现
|
||||
com.yomahub.liteflow.script.javascript.JavaScriptExecutor
|
||||
com.yomahub.liteflow.script.graaljs.GraalJavaScriptExecutor
|
||||
@@ -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>
|
||||
Reference in New Issue
Block a user