feature #I5RV4W 加强脚本和java之间的互动

This commit is contained in:
everywhere.z
2022-10-08 18:13:41 +08:00
parent f364bedf90
commit 979c4ea070
11 changed files with 142 additions and 31 deletions

View File

@@ -1,8 +1,10 @@
package com.yomahub.liteflow.script.groovy;
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.slot.DataBus;
import com.yomahub.liteflow.slot.Slot;
import com.yomahub.liteflow.script.ScriptExecutor;
@@ -49,42 +51,48 @@ public class GroovyScriptExecutor implements ScriptExecutor {
}
@Override
public Object execute(String currChainName, String nodeId, int slotIndex) {
public Object execute(ScriptExecuteWrap wrap) {
try{
if (!compiledScriptMap.containsKey(nodeId)){
String errorMsg = StrUtil.format("script for node[{}] is not loaded", nodeId);
if (!compiledScriptMap.containsKey(wrap.getNodeId())){
String errorMsg = StrUtil.format("script for node[{}] is not loaded", wrap.getNodeId());
throw new ScriptLoadException(errorMsg);
}
CompiledScript compiledScript = compiledScriptMap.get(nodeId);
CompiledScript compiledScript = compiledScriptMap.get(wrap.getNodeId());
Bindings bindings = new SimpleBindings();
//往脚本语言绑定表里循环增加绑定上下文的key
//key的规则为自定义上下文的simpleName
//比如你的自定义上下文为AbcContext那么key就为:abcContext
//这里不统一放一个map的原因是考虑到有些用户会调用上下文里的方法而不是参数所以脚本语言的绑定表里也是放多个上下文
DataBus.getContextBeanList(slotIndex).forEach(o -> {
DataBus.getContextBeanList(wrap.getSlotIndex()).forEach(o -> {
String key = StrUtil.lowerFirst(o.getClass().getSimpleName());
bindings.put(key, o);
});
//放入主Chain的流程参数
Slot slot = DataBus.getSlot(slotIndex);
bindings.put("requestData", slot.getRequestData());
//把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(currChainName);
Object subRequestData = slot.getChainReqData(wrap.getCurrChainName());
if (ObjectUtil.isNotNull(subRequestData)){
bindings.put("subRequestData", subRequestData);
metaMap.put("subRequestData", subRequestData);
}
//往脚本上下文里放入元数据
bindings.put("_meta", metaMap);
//放入用户自己定义的bean
bindings.putAll(ScriptBeanManager.getScriptBeanMap());
return compiledScript.eval(bindings);
}catch (Exception e){
log.error(e.getMessage(), e);
String errorMsg = StrUtil.format("script execute error for node[{}]", nodeId);
String errorMsg = StrUtil.format("script execute error for node[{}]", wrap.getNodeId());
throw new ScriptExecuteException(errorMsg);
}
}