向后加签

snow-flowable/src/main/java/com/snow/flowable/cmd/AfterSignUserTaskCmd.java

@@ -0,0 +1,114 @@
+package com.snow.flowable.cmd;
+
+import lombok.AllArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections.CollectionUtils;
+import org.flowable.bpmn.model.*;
+import org.flowable.bpmn.model.Process;
+import org.flowable.common.engine.api.FlowableException;
+import org.flowable.common.engine.api.FlowableObjectNotFoundException;
+import org.flowable.common.engine.impl.interceptor.Command;
+import org.flowable.common.engine.impl.interceptor.CommandContext;
+import org.flowable.engine.impl.cmd.AbstractDynamicInjectionCmd;
+import org.flowable.engine.impl.dynamic.BaseDynamicSubProcessInjectUtil;
+import org.flowable.engine.impl.dynamic.DynamicUserTaskBuilder;
+import org.flowable.engine.impl.persistence.entity.DeploymentEntity;
+import org.flowable.engine.impl.persistence.entity.ExecutionEntity;
+import org.flowable.engine.impl.persistence.entity.ProcessDefinitionEntity;
+import org.flowable.engine.impl.util.CommandContextUtil;
+import org.flowable.task.service.TaskService;
+import org.flowable.task.service.impl.persistence.entity.TaskEntity;
+
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * @author qimingjin
+ * @Title: 动态加签
+ * @Description:
+ * @date 2021/6/16 14:57
+ */
+@Slf4j
+@AllArgsConstructor
+public class AfterSignUserTaskCmd extends AbstractDynamicInjectionCmd implements Command<Void> {
+    //流程实例id
+   protected String processInstanceId;
+
+   //后加签的节点信息
+   private DynamicUserTaskBuilder signUserTaskBuilder;
+
+   private String taskId;
+
+
+    @Override
+    public Void execute(CommandContext commandContext) {
+        createDerivedProcessDefinitionForProcessInstance(commandContext,processInstanceId);
+        return null;
+    }
+
+    @Override
+    protected void updateBpmnProcess(CommandContext commandContext, Process process, BpmnModel bpmnModel, ProcessDefinitionEntity processDefinitionEntity, DeploymentEntity deploymentEntity) {
+
+        TaskService taskService = CommandContextUtil.getTaskService(commandContext);
+
+        TaskEntity task = taskService.getTask(taskId);
+
+        if(task==null){
+            throw new FlowableObjectNotFoundException("task:"+taskId+" not found");
+        }
+        //查找当前节点对应的执行执行实体（感觉兴趣的可以搜下流程执行的实例信息 表为ACT_RU_EXECUTION）
+        ExecutionEntity currentExecutionEntity = CommandContextUtil.getExecutionEntityManager(commandContext).findById(task.getExecutionId());
+        if(currentExecutionEntity==null){
+             throw new FlowableObjectNotFoundException("task:"+taskId+",execution:"+task.getExecutionId()+" not found");
+         }
+        //定义的节点id
+        String activityId = currentExecutionEntity.getActivityId();
+        FlowElement flowElement = process.getFlowElement(activityId,true);
+        if(!(flowElement instanceof Task)){
+             throw new FlowableException("task type error");
+         }
+        Activity activity = (Activity) flowElement;
+        //由于是后加签所以需要获取当前流程节点流出的sequence
+        SequenceFlow currentTaskOutSequenceFlow = activity.getOutgoingFlows().get(0);
+        //定义新的节点
+        UserTask addUserTask = new UserTask();
+        //流程的id 可以在调用的时候传入 也可以根据flwoable提供的方法生成
+        String id=signUserTaskBuilder.getId();
+        if (id == null) {
+            id=signUserTaskBuilder.nextTaskId(process.getFlowElementMap());
+        }
+        addUserTask.setId(id);
+        addUserTask.setName(signUserTaskBuilder.getName());
+        addUserTask.setAssignee(signUserTaskBuilder.getAssignee());
+        process.addFlowElement(addUserTask);
+        signUserTaskBuilder.setDynamicTaskId(id);
+
+        String flowId = "sequenceFlow-"+CommandContextUtil.getProcessEngineConfiguration(commandContext).getIdGenerator().getNextId();
+        SequenceFlow newSequenceFlow = this.createSequenceFlow(addUserTask.getId(),currentTaskOutSequenceFlow.getTargetFlowElement().getId());
+        newSequenceFlow.setId(flowId);
+        process.addFlowElement(newSequenceFlow);
+
+        activity.setOutgoingFlows(Collections.singletonList(newSequenceFlow));
+        //修改原有sequence 将原来的sequence的taget改为新的节点
+        currentTaskOutSequenceFlow.setTargetFlowElement(addUserTask);
+        currentTaskOutSequenceFlow.setTargetRef(addUserTask.getId());
+
+        bpmnModel.getLocationMap().clear();
+        bpmnModel.getFlowLocationMap().clear();
+        bpmnModel.getLabelLocationMap().clear();
+        //当前流程的重新定义
+        BaseDynamicSubProcessInjectUtil.processFlowElements(commandContext, process, bpmnModel, processDefinitionEntity, deploymentEntity);
+    }
+
+    @Override
+    protected void updateExecutions(CommandContext commandContext, ProcessDefinitionEntity processDefinitionEntity, ExecutionEntity executionEntity, List<ExecutionEntity> list) {
+
+    }
+
+    public SequenceFlow createSequenceFlow(String from ,String to){
+        SequenceFlow flow=new SequenceFlow();
+        flow.setSourceRef(from);
+        flow.setTargetRef(to);
+        return flow;
+    }
+}