工控网涉密网联调

yangys

2025-09-29 4c7296d45efe849dc70a3b2e2240c905481a91c9

 blade-service/blade-mdm/src/main/java/org/springblade/mdm/flow/controller/DispatchController.java

@@ -1,48 +1,29 @@
package org.springblade.mdm.flow.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.SerializationException;
import org.apache.commons.lang3.StringUtils;
import org.flowable.engine.RuntimeService;
import org.flowable.engine.TaskService;
import org.flowable.engine.runtime.ProcessInstance;
import org.flowable.task.api.Task;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.api.IResultCode;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.support.Kv;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.mdm.flow.constants.FlowContants;
import org.springblade.mdm.flow.entity.ApproveRecord;
import org.springblade.mdm.flow.constants.FlowConstant;
import org.springblade.mdm.flow.constants.FlowVariableConstant;
import org.springblade.mdm.flow.entity.TaskDispatch;
import org.springblade.mdm.flow.excution.StartDispatcher;
import org.springblade.mdm.flow.service.ApproveRecordService;
import org.springblade.mdm.flow.service.FlowCommonService;
import org.springblade.mdm.flow.service.FlowProgramFileService;
import org.springblade.mdm.flow.service.TaskDispatchService;
import org.springblade.mdm.flow.service.execute.AbstractFlowCompleteService;
import org.springblade.mdm.flow.service.execute.BatchDispatchService;
import org.springblade.mdm.flow.service.execute.DefaultFlowCompleteService;
import org.springblade.mdm.flow.service.execute.TryFlowCompleteService;
import org.springblade.mdm.flow.service.execute.*;
import org.springblade.mdm.flow.vo.BatchDispatchVO;
import org.springblade.mdm.flow.vo.FlowStartResult;
import org.springblade.mdm.flow.vo.TaskAssignVO;
import org.springblade.mdm.program.entity.ProcessProgRef;
import org.springblade.mdm.program.service.NcNodeAutoCreateService;
import org.springblade.mdm.program.service.ProcessProgRefService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@Slf4j
@@ -54,19 +35,16 @@

   private final StartDispatcher dispatcher;

   private final TaskService taskService;
   private final RuntimeService runtimeService;
   private final ProcessProgRefService processProgRefService;

   private final ApproveRecordService approveRecordService;
   private final FlowProgramFileService flowProgramFileService;


   private final TryFlowCompleteService tryFlowCompleteService;
   private final DefaultFlowCompleteService defaultFlowCompleteService;
   private final TaskDispatchService taskDispatchService;
   private final BatchDispatchService dispatchService;

   private final ReplaceFlowCompleteService replaceFlowCompleteService;
   private final TempFlowCompleteService tempFlowCompleteService;
   private final CureFlowCompleteService  cureFlowCompleteService;
   /**
    * 发起派工流程
    */
@@ -74,20 +52,72 @@
   @Operation(summary = "任务计划(派工流程)", description = "启动派工流程")
   public R<Boolean> start(@RequestBody TaskAssignVO startVO) {
      try {
         long id= taskDispatchService.saveTask(startVO);
         String instId = dispatcher.start(startVO);
         taskDispatchService.updateSuccess(id,instId);
         return R.data(true);
         boolean isDuplicate = taskDispatchService.checkIsDuplicate(startVO);

         TaskDispatch dispatch = taskDispatchService.saveTask(startVO,isDuplicate);
         startVO.setCreateUser(dispatch.getCreateUser());
         if(isDuplicate){
            throw new ServiceException("重复派工");
         }

         try {
            FlowStartResult result = dispatcher.start(startVO);
            dispatch.setStatus(TaskDispatch.STATUS_STARTED);
            dispatch.setProcessInstanceId(result.getProcessInstanceId());
            dispatch.setProgramNo(result.getProgramNo());
            taskDispatchService.updateById(dispatch);
            return R.data(true);
         } catch (Exception e) {
            dispatch.setStatus(TaskDispatch.STATUS_EXCEPTION);
            dispatch.setErrMsg(e.getMessage());
            taskDispatchService.updateById(dispatch);
            return R.fail(new IResultCode() {
               @Override
               public String getMessage() {
                  return e.getMessage();
               }

               @Override
               public int getCode() {
                  return 201;
               }
            });
         }
      }catch(Exception e){
         return R.fail(e.getMessage());
         return R.fail(new IResultCode() {
            @Override
            public String getMessage() {
               return e.getMessage();
            }

            @Override
            public int getCode() {
               return 201;
            }
         });
      }

   }

   /**
    * 获取实际审批流程执行服务
    * @param processInstanceId
    * @return
    */
   AbstractFlowCompleteService getActualService(String processInstanceId){
      ProcessInstance inst = runtimeService.createProcessInstanceQuery().processInstanceId(processInstanceId).singleResult();
      if(inst.getProcessDefinitionKey().equals(FlowContants.TRY_PROCESS_KEY) || inst.getProcessDefinitionKey().equals(FlowContants.CURE_PROCESS_KEY)){
         return tryFlowCompleteService;
      ProcessInstance inst = runtimeService.createProcessInstanceQuery().processInstanceId(processInstanceId).includeProcessVariables().singleResult();
      if(inst.getProcessDefinitionKey().equals(FlowConstant.TRY_PROCESS_KEY)
         || inst.getProcessDefinitionKey().equals(FlowConstant.UNLOCK_PROCESS_KEY)){
         String isTempFlow = Func.toStr(inst.getProcessVariables().get(FlowVariableConstant.IS_TEMP_FLOW));
         if(FlowConstant.N.equals(isTempFlow)){
            return tryFlowCompleteService;
         }else{
            return tempFlowCompleteService;
         }
      }else if(inst.getProcessDefinitionKey().equals(FlowConstant.CURE_PROCESS_KEY)){
         return cureFlowCompleteService;
      }else if(inst.getProcessDefinitionKey().equals(FlowConstant.REPLACE_PROCESS_KEY)){
         return replaceFlowCompleteService;
      }else{
         return defaultFlowCompleteService;
      }
@@ -98,7 +128,6 @@
      try {
         AbstractFlowCompleteService completeService = getActualService(processInstanceId);
         completeService.completeTask(taskId, processInstanceId, comment, variables);


         return R.success("流程提交成功");
      }catch(Exception e){
@@ -112,7 +141,7 @@
   @PostMapping("batchDispatch")
   public R<Void> batchDispatch(@RequestBody BatchDispatchVO batchDispatchVO) {
      try {
         dispatchService.batchCompleteTask(batchDispatchVO);
         dispatchService.batchDispatchTask(batchDispatchVO);
         return R.success("流程提交成功");
      }catch(Exception e){
         return R.fail(e.getMessage());
@@ -124,21 +153,35 @@
   @PostMapping("batchDispatchAuto")
   public R<Void> batchDispatchAuto(@RequestBody BatchDispatchVO batchDispatchVO) {
      try {
         dispatchService.batchAutoCompleteTask(batchDispatchVO);
         dispatchService.batchAutoDispatchTask(batchDispatchVO);
         return R.success("流程提交成功");
      }catch(Exception e){
         return R.fail(e.getMessage());
      }

   }
   /*
   void addApproveRecord(String taskId,String comment,Map<String, Object> variables){
      String operateResult = variables.get("approve")+"";

      Task task = taskService.createTaskQuery()
         .taskId(taskId)
         .singleResult();
      approveRecordService.saveApproveRecords(task,operateResult,comment);
   @Operation(summary = "批量审批", description = "高师批量审批")
   @PostMapping("batchApprove")
   public R<Void> batchApprove(@RequestBody BatchDispatchVO batchDispatchVO) {
      try {
         dispatchService.batchApprove(batchDispatchVO);
         return R.success("流程提交成功");
      }catch(Exception e){
         return R.fail(e.getMessage());
      }
   }

   }*/
   @Operation(summary = "编程员现场编制批量处理", description = "编程员现场编制批量处理")
   @PostMapping("batch-program-on-machine")
   public R<Void> batchProgramOnMachine(@RequestBody BatchDispatchVO batchDispatchVO) {
      try {
         dispatchService.batchProgramOnMachine(batchDispatchVO);
         return R.success("流程提交成功");
      }catch(Exception e){
         return R.fail(e.getMessage());
      }

   }

}