FlowEngineService 流程执行引擎核心服务 + 编译通过
This commit is contained in:
root
@@ -0,0 +1,246 @@
|
|||||||
|
package com.ruoyi.wcs.flow.engine;
|
||||||
|
|
||||||
|
import java.util.*;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.stereotype.Service;
|
||||||
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
|
import com.ruoyi.wcs.flow.domain.*;
|
||||||
|
import com.ruoyi.wcs.flow.service.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* WCS 轻量流程执行引擎
|
||||||
|
* 按流程定义依次执行节点，记录实例和节点日志
|
||||||
|
*
|
||||||
|
* @author wcs
|
||||||
|
*/
|
||||||
|
@Service
|
||||||
|
public class FlowEngineService
|
||||||
|
{
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(FlowEngineService.class);
|
||||||
|
|
||||||
|
private Map<String, INodeHandler> handlerMap = new HashMap<>();
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private IWcsFlowDefService flowDefService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private IWcsFlowNodeService flowNodeService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private IWcsFlowLineService flowLineService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private IWcsFlowInstanceService flowInstanceService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private IWcsFlowNodeLogService flowNodeLogService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
public void setHandlers(List<INodeHandler> handlers)
|
||||||
|
{
|
||||||
|
for (INodeHandler handler : handlers)
|
||||||
|
{
|
||||||
|
handlerMap.put(handler.getNodeType(), handler);
|
||||||
|
}
|
||||||
|
log.info("[FlowEngine] 已注册 {} 个节点处理器: {}", handlerMap.size(), handlerMap.keySet());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行流程
|
||||||
|
*
|
||||||
|
* @param flowDefId 流程定义ID
|
||||||
|
* @return 流程实例ID
|
||||||
|
*/
|
||||||
|
@Transactional
|
||||||
|
public Long executeFlow(Long flowDefId)
|
||||||
|
{
|
||||||
|
// 1. 查询流程定义
|
||||||
|
WcsFlowDef flowDef = flowDefService.selectFlowDefById(flowDefId);
|
||||||
|
if (flowDef == null)
|
||||||
|
{
|
||||||
|
throw new RuntimeException("流程定义不存在: " + flowDefId);
|
||||||
|
}
|
||||||
|
if (!"1".equals(flowDef.getStatus()))
|
||||||
|
{
|
||||||
|
throw new RuntimeException("流程未发布，无法执行: " + flowDef.getFlowCode());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. 创建流程实例
|
||||||
|
WcsFlowInstance instance = new WcsFlowInstance();
|
||||||
|
instance.setFlowDefId(flowDefId);
|
||||||
|
instance.setStatus("1"); // 执行中
|
||||||
|
instance.setStartTime(new Date());
|
||||||
|
flowInstanceService.insertFlowInstance(instance);
|
||||||
|
Long instanceId = instance.getId();
|
||||||
|
log.info("[FlowEngine] 流程实例创建: instanceId={}, flowCode={}", instanceId, flowDef.getFlowCode());
|
||||||
|
|
||||||
|
// 3. 加载节点和连线
|
||||||
|
List<WcsFlowNode> nodes = flowNodeService.selectFlowNodeListByFlowId(flowDefId);
|
||||||
|
List<WcsFlowLine> lines = flowLineService.selectFlowLineListByFlowId(flowDefId);
|
||||||
|
|
||||||
|
// 按nodeId建索引
|
||||||
|
Map<Long, WcsFlowNode> nodeMap = nodes.stream()
|
||||||
|
.collect(Collectors.toMap(WcsFlowNode::getNodeId, n -> n));
|
||||||
|
|
||||||
|
// 4. 找到开始节点
|
||||||
|
WcsFlowNode currentNode = findStartNode(nodes);
|
||||||
|
if (currentNode == null)
|
||||||
|
{
|
||||||
|
updateInstanceStatus(instance, "3", "未找到开始节点");
|
||||||
|
throw new RuntimeException("流程缺少开始节点: " + flowDef.getFlowCode());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 5. 执行上下文
|
||||||
|
Map<String, Object> context = new HashMap<>();
|
||||||
|
context.put("flowDefId", flowDefId);
|
||||||
|
context.put("instanceId", instanceId);
|
||||||
|
|
||||||
|
// 6. 按连线顺序执行
|
||||||
|
int maxSteps = 100; // 防止死循环
|
||||||
|
int step = 0;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
while (currentNode != null && step < maxSteps)
|
||||||
|
{
|
||||||
|
step++;
|
||||||
|
log.info("[FlowEngine] 执行节点: step={}, nodeId={}, type={}, name={}",
|
||||||
|
step, currentNode.getNodeId(), currentNode.getNodeType(), currentNode.getNodeName());
|
||||||
|
|
||||||
|
// 记录节点日志-开始
|
||||||
|
WcsFlowNodeLog nodeLog = new WcsFlowNodeLog();
|
||||||
|
nodeLog.setInstanceId(instanceId);
|
||||||
|
nodeLog.setNodeId(String.valueOf(currentNode.getNodeId()));
|
||||||
|
nodeLog.setNodeType(currentNode.getNodeType());
|
||||||
|
nodeLog.setStatus("1"); // 执行中
|
||||||
|
nodeLog.setStartTime(new Date());
|
||||||
|
flowNodeLogService.insertFlowNodeLog(nodeLog);
|
||||||
|
|
||||||
|
// 获取handler
|
||||||
|
INodeHandler handler = getHandler(currentNode.getNodeType());
|
||||||
|
if (handler == null)
|
||||||
|
{
|
||||||
|
nodeLog.setStatus("3"); // 失败
|
||||||
|
nodeLog.setErrorMsg("未找到节点处理器: " + currentNode.getNodeType());
|
||||||
|
nodeLog.setEndTime(new Date());
|
||||||
|
flowNodeLogService.updateFlowNodeLog(nodeLog);
|
||||||
|
updateInstanceStatus(instance, "3", nodeLog.getErrorMsg());
|
||||||
|
return instanceId;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 执行节点
|
||||||
|
NodeResult result = handler.execute(currentNode.getConfigJson(), context);
|
||||||
|
|
||||||
|
// 更新节点日志
|
||||||
|
nodeLog.setEndTime(new Date());
|
||||||
|
if (result.isSuccess())
|
||||||
|
{
|
||||||
|
nodeLog.setStatus("2"); // 成功
|
||||||
|
nodeLog.setOutput(result.getOutput());
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
nodeLog.setStatus("3"); // 失败
|
||||||
|
nodeLog.setErrorMsg(result.getErrorMsg());
|
||||||
|
flowNodeLogService.updateFlowNodeLog(nodeLog);
|
||||||
|
updateInstanceStatus(instance, "3", result.getErrorMsg());
|
||||||
|
return instanceId;
|
||||||
|
}
|
||||||
|
flowNodeLogService.updateFlowNodeLog(nodeLog);
|
||||||
|
|
||||||
|
// end节点，流程结束
|
||||||
|
if ("end".equals(currentNode.getNodeType()))
|
||||||
|
{
|
||||||
|
log.info("[FlowEngine] 流程执行完成: instanceId={}", instanceId);
|
||||||
|
updateInstanceStatus(instance, "2", null); // 已完成
|
||||||
|
return instanceId;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 找下一个节点
|
||||||
|
currentNode = findNextNode(currentNode, result, lines, nodeMap);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 超过最大步数
|
||||||
|
if (step >= maxSteps)
|
||||||
|
{
|
||||||
|
updateInstanceStatus(instance, "3", "超过最大执行步数: " + maxSteps);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
log.error("[FlowEngine] 流程执行异常: instanceId={}", instanceId, e);
|
||||||
|
updateInstanceStatus(instance, "3", e.getMessage());
|
||||||
|
}
|
||||||
|
|
||||||
|
return instanceId;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 找开始节点
|
||||||
|
*/
|
||||||
|
private WcsFlowNode findStartNode(List<WcsFlowNode> nodes)
|
||||||
|
{
|
||||||
|
return nodes.stream()
|
||||||
|
.filter(n -> "start".equals(n.getNodeType()))
|
||||||
|
.findFirst()
|
||||||
|
.orElse(null);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据连线找下一个节点
|
||||||
|
*/
|
||||||
|
private WcsFlowNode findNextNode(WcsFlowNode current, NodeResult result,
|
||||||
|
List<WcsFlowLine> lines, Map<Long, WcsFlowNode> nodeMap)
|
||||||
|
{
|
||||||
|
List<WcsFlowLine> outLines = lines.stream()
|
||||||
|
.filter(l -> current.getNodeId().equals(l.getSourceNodeId()))
|
||||||
|
.sorted(Comparator.comparingInt(l -> l.getSortOrder() != null ? l.getSortOrder() : 0))
|
||||||
|
.collect(Collectors.toList());
|
||||||
|
|
||||||
|
if (outLines.isEmpty())
|
||||||
|
{
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// condition节点：按branchKey匹配conditionExpr
|
||||||
|
if ("condition".equals(current.getNodeType()) && result.getBranchKey() != null)
|
||||||
|
{
|
||||||
|
for (WcsFlowLine line : outLines)
|
||||||
|
{
|
||||||
|
if (result.getBranchKey().equals(line.getConditionExpr()))
|
||||||
|
{
|
||||||
|
return nodeMap.get(line.getTargetNodeId());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 没匹配到，走第一条（默认分支）
|
||||||
|
log.warn("[FlowEngine] 条件分支未匹配, branchKey={}, 走默认分支", result.getBranchKey());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 非condition节点或默认：走第一条连线
|
||||||
|
return nodeMap.get(outLines.get(0).getTargetNodeId());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取节点处理器
|
||||||
|
*/
|
||||||
|
private INodeHandler getHandler(String nodeType)
|
||||||
|
{
|
||||||
|
return handlerMap.get(nodeType);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 更新流程实例状态
|
||||||
|
*/
|
||||||
|
private void updateInstanceStatus(WcsFlowInstance instance, String status, String remark)
|
||||||
|
{
|
||||||
|
instance.setStatus(status);
|
||||||
|
instance.setEndTime(new Date());
|
||||||
|
if (remark != null)
|
||||||
|
{
|
||||||
|
instance.setRemark(remark);
|
||||||
|
}
|
||||||
|
flowInstanceService.updateFlowInstance(instance);
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user