模型计算管理开发

ruoyi-api-patform/src/main/java/com/ruoyi/interfaces/controller/MdFlowDispatchController.java

@@ -41,4 +41,14 @@ public class MdFlowDispatchController extends BaseController {
     }
 
 
+    @PutMapping
+    public AjaxResult update(@RequestBody MdFlowDispatchVo flowDispatchVo) {
+        flowDispatchVo.setUpdateBy(getUsername());
+        flowDispatchVo.setJobName(flowDispatchVo.getFlowJobName());
+        return success(mdFlowDispatchService.update(flowDispatchVo));
+    }
+
+
+
+
 }

ruoyi-api-patform/src/main/java/com/ruoyi/interfaces/mapper/MdFlowDispatchMapper.java

@@ -15,4 +15,6 @@ public interface MdFlowDispatchMapper {
     List<MdFlowDispatchVo> selectFlowDispatch(MdFlowDispatchVo par);
 
     MdFlowDispatchVo getInfo(String id);
+
+    Long updateMdFlowJob(MdFlowDispatchVo flowDispatchVo);
 }

ruoyi-api-patform/src/main/java/com/ruoyi/interfaces/service/IMdFlowDispatchService.java

@@ -10,4 +10,6 @@ public interface IMdFlowDispatchService {
     List<MdFlowDispatchVo> selectFlowDispatch(MdFlowDispatchVo par);
 
     MdFlowDispatchVo getInfo(String id);
+
+    Long update(MdFlowDispatchVo flowDispatchVo);
 }

ruoyi-api-patform/src/main/java/com/ruoyi/interfaces/service/impl/MdFlowDispatchServiceImpl.java

@@ -69,4 +69,37 @@ public class MdFlowDispatchServiceImpl implements IMdFlowDispatchService {
         info.initFlowParam();
         return info;
     }
+
+    @Override
+    public Long update(MdFlowDispatchVo flowDispatchVo) {
+        HashMap<String, Object> argsMap = new HashMap<>();
+        argsMap.put("flowId", flowDispatchVo.getFlowId());
+        argsMap.put("flowParam", flowDispatchVo.getFlowParam());
+
+
+        flowDispatchVo.setArgsStrMap(argsMap);
+
+        Boolean updateState = false;
+        switch (flowDispatchVo.getTaskType()) {
+            case 1:
+                updateState = snailJobService.updateClusterJob(flowDispatchVo);
+                break;
+            case 2:
+                updateState = snailJobService.updateBroadcastJob(flowDispatchVo);
+                break;
+            case 3:
+                updateState = snailJobService.updateShardingJob(flowDispatchVo);
+                break;
+            case 4:
+                updateState = snailJobService.updateMapJob(flowDispatchVo);
+                break;
+            case 5:
+                updateState = snailJobService.updateMapReduceJob(flowDispatchVo);
+                break;
+        }
+        if (updateState) {
+            return 0L;
+        }
+        return flowDispatchMapper.updateMdFlowJob(flowDispatchVo);
+    }
 }

ruoyi-api-patform/src/main/resources/mapper/interfaces/MdFlowDispatchMapper.xml

@@ -24,6 +24,18 @@
                 #{chargeBy,jdbcType=VARCHAR},
                 #{chargePhone,jdbcType=VARCHAR})
     </insert>
+    <update id="updateMdFlowJob">
+        update MD_FLOW_JOB
+        set flow_job_name     = #{flowJobName,jdbcType=VARCHAR},
+            flow_id           = #{flowId,jdbcType=INTEGER},
+            id                = #{id,jdbcType=INTEGER},
+            flow_param_string = #{flowParamString,jdbcType=VARCHAR},
+            create_by         = #{createBy,jdbcType=VARCHAR},
+            create_time       = #{createTime,jdbcType=DATE},
+            charge_by         = #{chargeBy,jdbcType=VARCHAR},
+            charge_phone      = #{chargePhone,jdbcType=VARCHAR}
+        where flow_job_id = #{flowJobId}
+    </update>
     <resultMap type="com.ruoyi.interfaces.domain.vo.MdFlowDispatchVo" id="MdFlowDispatchMap">
         <result property="flowJobId" column="FLOW_JOB_ID"/>
         <result property="flowId" column="FLOW_ID"/>