flowable之驳回多实例驳回并行网关驳回普通节点驳回到多实例

 public ReturnVo<String> backToStep(BackVo backVo) 
        ReturnVo<String> returnVo = new ReturnVo<>(ReturnCode.SUCCESS, "OK");
        if (backVo != null && StringUtils.isNotBlank(backVo.getTaskId())) 
            TaskEntity taskEntity = (TaskEntity) taskService.createTaskQuery().taskId(backVo.getTaskId()).singleResult();
            if (taskEntity != null)
                Activity distActivity = processDefinitionUtils.findFlowElementById(taskEntity.getProcessDefinitionId(), backVo.getDistFlowElementId());
                if (taskEntity != null && distActivity != null) 
                    //1. 判断该节点上一个节点是不是并行网关节点
                    List<SequenceFlow> incomingFlows = distActivity.getIncomingFlows();
                    if (CollectionUtils.isNotEmpty(incomingFlows)) 
                        for (SequenceFlow sequenceFlow : incomingFlows) 
                            FlowElement upNode = sequenceFlow.getSourceFlowElement();
                            if (upNode != null && (upNode instanceof ParallelGateway || upNode instanceof InclusiveGateway)) 
                                returnVo = new ReturnVo<>(ReturnCode.FAIL, "并行节点无法驳回，请选择其他节点!");
                                return returnVo;
                            
                        
                    
                    //2. 如果上一个节点是提交者的话要处理一下
                    if (FlowConstant.FLOW_SUBMITTER.equals(distActivity.getName())) 
                        //查找发起人
                        ExtendHisprocinst extendHisprocinst = this.extendHisprocinstService.findExtendHisprocinstByProcessInstanceId(taskEntity.getProcessInstanceId());
                        if (extendHisprocinst != null) 
                            runtimeService.setVariable(taskEntity.getProcessInstanceId(), FlowConstant.FLOW_SUBMITTER, extendHisprocinst.getCreator());
                        
                    
                    //3. 添加审批意见和修改流程状态
                    this.addCommentAndUpdateProcessStatus(backVo);
                    //4. 删除现有的所有任务
                    managementService.executeCommand(new DeleteTaskCmd(taskEntity.getProcessInstanceId()));
                    //5. 删除节点信息
                    this.deleteHisActivities(distActivity, taskEntity.getProcessInstanceId());
                    //6. 驳回到disk节点
                    Activity currActivity = processDefinitionUtils.findFlowElementById(taskEntity.getProcessDefinitionId(), taskEntity.getTaskDefinitionKey());
                    //6.1 如果当前节点是多实例节点 删除当前多实例 如果目标节点不是多实例我们就创建一个孩子实例
                    boolean flag = false;
                    if (currActivity.getBehavior() instanceof MultiInstanceActivityBehavior)
                        ExecutionEntity executionEntity = (ExecutionEntity)runtimeService.createExecutionQuery().executionId(taskEntity.getExecutionId()).singleResult();
                        managementService.executeCommand(new DeleteMultiInstanceExecutionCmd(executionEntity.getParentId(),false));
                        flag = true;
                    
                    //6.2 处理并行网关的多实例
                    List<Execution> executions = runtimeService.createExecutionQuery().parentId(taskEntity.getProcessInstanceId()).list();
                    if (CollectionUtils.isNotEmpty(executions) && executions.size() >1)
                        executions.forEach(execution -> 
                            ExecutionEntity e = (ExecutionEntity)execution;
                            managementService.executeCommand(new DeleteChildExecutionCmd(e));
                        );
                        flag = true;
                    
                    if (flag) 
                        ExecutionEntity parentExecutionEntity = (ExecutionEntity)runtimeService.createExecutionQuery().executionId(taskEntity.getProcessInstanceId()).singleResult();
                        managementService.executeCommand(new AddChildExecutionCmd(parentExecutionEntity));
                    
                    managementService.executeCommand(new JumpActivityCmd(taskEntity.getProcessInstanceId(),distActivity.getId()));
                    //TODO 7. 处理加签的数据0
                
            else 
                returnVo = new ReturnVo<>(ReturnCode.FAIL, "当前任务不存在!");
            
         else 
            returnVo = new ReturnVo<>(ReturnCode.FAIL, "请设置相关参数!");
        
        return returnVo;
    

@Component
public class ProcessDefinitionUtils 

    @Autowired
    private RepositoryService repositoryService;

    /**
     * 获取end节点
     *
     * @param processDefId
     * @return FlowElement
     */
    public FlowElement findEndFlowElement(String processDefId) 
        Process process = repositoryService.getBpmnModel(processDefId).getMainProcess();
        Collection<FlowElement> list = process.getFlowElements();
        for (FlowElement f : list) 
            if (f instanceof EndEvent) 
                return f;
            
        
        return null;
    

    /**
     * 获取指定节点的节点信息
     *
     * @param processDefId
     * @param flowElementId
     * @return FlowElement
     */
    public Activity findFlowElementById(String processDefId, String flowElementId) 
        Process process = repositoryService.getBpmnModel(processDefId).getMainProcess();
        return (Activity) process.getFlowElement(flowElementId);
    

/**
     * 添加审批意见和修改流程状态
     * @param baseProcessVo 基本流程任务参数
     */
    protected void addCommentAndUpdateProcessStatus(BaseProcessVo baseProcessVo) 
        //兼容处理
        if (StringUtils.isBlank(baseProcessVo.getProcessInstanceId()))
            Task task = taskService.createTaskQuery().taskId(baseProcessVo.getTaskId()).singleResult();
            if (task != null) 
                baseProcessVo.setProcessInstanceId(task.getProcessInstanceId());
            
        
        //1.添加审批意见
        FlowCommentVo flowCommentVo = new FlowCommentVo(baseProcessVo.getTaskId(), baseProcessVo.getUserCode(),
                baseProcessVo.getProcessInstanceId(), baseProcessVo.getMessage(), baseProcessVo.getCommentTypeEnum().toString());
        this.addFlowComment(flowCommentVo);
        //2.修改流程实例的状态
        ExtendHisprocinst extendHisprocinst = new ExtendHisprocinst(baseProcessVo.getProcessInstanceId(), baseProcessVo.getProcessStatusEnum().toString());
        extendHisprocinstService.updateStatusByProcessInstanceId(extendHisprocinst);
        //3.TODO 生成索引
    

    /**
     * 添加审批意见
     * @param flowCommentVo
     */
    private void addFlowComment(FlowCommentVo flowCommentVo) 
        FlowCommentCmd cmd = new FlowCommentCmd(flowCommentVo.getTaskId(), flowCommentVo.getUserId(),
                flowCommentVo.getProcessInstanceId(), flowCommentVo.getType(), flowCommentVo.getMessage());
        managementService.executeCommand(cmd);
    

public class DeleteTaskCmd implements Command<Void> 

    private String processInstanceId;

    public DeleteTaskCmd(String processInstanceId) 
        this.processInstanceId = processInstanceId;
    

    @Override
    public Void execute(CommandContext commandContext) 
        TaskEntityManager taskEntityManager = CommandContextUtil.getTaskEntityManager(commandContext);
        ExecutionEntityManager executionEntityManager = org.flowable.engine.impl.util.CommandContextUtil.getExecutionEntityManager(commandContext);
        List<ExecutionEntity> executionEntities = executionEntityManager.findChildExecutionsByProcessInstanceId(processInstanceId);
        executionEntities.forEach(executionEntity -> taskEntityManager.deleteTasksByExecutionId(executionEntity.getId()));
        return null;
    

public class AddChildExecutionCmd implements Command<Void> 

    private ExecutionEntity parentExecutionEntity;

    public AddChildExecutionCmd(ExecutionEntity parentExecutionEntity) 
        this.parentExecutionEntity = parentExecutionEntity;
    

    @Override
    public Void execute(CommandContext commandContext) 
        ExecutionEntityManager executionEntityManager = CommandContextUtil.getExecutionEntityManager(commandContext);
        executionEntityManager.createChildExecution(parentExecutionEntity);
        return null;
    

public class DeleteChildExecutionCmd implements Command<Void> 

    private ExecutionEntity child;

    public DeleteChildExecutionCmd(ExecutionEntity child) 
        this.child = child;
    

    @Override
    public Void execute(CommandContext commandContext) 
        ExecutionEntityManager executionEntityManager = CommandContextUtil.getExecutionEntityManager(commandContext);
        executionEntityManager.delete(child,true);
        return null;
    

public class JumpActivityCmd implements Command<Void> 

    private String target;
    private String processInstanceId;

    public JumpActivityCmd(String processInstanceId, String target) 
        this.processInstanceId = processInstanceId;
        this.target = target;
    

    @Override
    public Void execute(CommandContext commandContext) 
        ExecutionEntityManager executionEntityManager = CommandContextUtil.getExecutionEntityManager(commandContext);
        List<ExecutionEntity> executionEntities = executionEntityManager.findChildExecutionsByParentExecutionId(processInstanceId);
        Process process = ProcessDefinitionUtil.getProcess(executionEntities.get(0).getProcessDefinitionId());
        FlowNode targetFlowElement = (FlowNode) process.getFlowElement(target);
        FlowableEngineAgenda agenda = CommandContextUtil.getAgenda();
        executionEntities.forEach(execution -> 
            execution.setCurrentFlowElement(targetFlowElement);
            agenda.planContinueProcessInCompensation(execution);
        );
        return null;