监控启智智算任务状态

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/mapper/ZSServiceVersionMapper.java

@@ -60,4 +60,9 @@ public interface ZSServiceVersionMapper {
      * 查询所有runState为COMPLETED的版本列表
      */
     List<ServiceVersionZs> queryAllComplList();
+
+    /**
+     * 查询所有runState为Running的版本列表
+     */
+    List<ServiceVersionZs> queryAllRunList();
 }

ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/scheduling/ServiceVersionStatusTask.java

@@ -164,7 +164,7 @@ public class ServiceVersionStatusTask {
      * 新增定时任务：处理已完成的任务，获取URL
      */
     @Scheduled(cron = "0 * * * * ?") // 每分钟执行一次
-    public void executeServiceVersionUrlTask() {
+    public void executeSetUrl() {
         try {
             // 1. 调用queryAllComplList接口获取所有runState为COMPLETED且url为空的数据
             List<ServiceVersionZs> completedVersions = serviceVersionMapper.queryAllComplList();
@@ -200,6 +200,44 @@ public class ServiceVersionStatusTask {
         }
     }
 
+    /**
+     * 新增定时任务：处理已完成的任务，获取URL
+     */
+    @Scheduled(cron = "0 * * * * ?") // 每分钟执行一次
+    public void executeServiceStatusChange() {
+        try {
+            // 1. 调用queryAllComplList接口获取所有runState为COMPLETED且url为空的数据
+            List<ServiceVersionZs> completedVersions = serviceVersionMapper.queryAllComplList();
+
+            if (completedVersions != null && !completedVersions.isEmpty()) {
+                for (ServiceVersionZs version : completedVersions) {
+                    // 2. 过滤获取数据中的jobSetId字段
+                    String jobSetId = version.getJobSetId();
+                    if (jobSetId != null && !jobSetId.isEmpty()) {
+                        try {
+                            // 3. 调用远程接口获取状态和URL
+                            String taskStatusUrl = zsServiceVersionUrl + "/api/v1/getTaskStatus";
+                            Map<String, Object> taskStatusInfo = callGetTaskStatus(taskStatusUrl, jobSetId);
+
+                            // 4. 如果url有值，则将url写入对应的数据中，然后更新数据库
+                            if (taskStatusInfo != null) {
+                                String status = taskStatusInfo.get("status").toString();
+                                if ("STOPPED".equals(status)) {
+                                    version.setRunState(capitalizeFirstLetter(status));
+                                    serviceVersionMapper.updateVersion(version);
+                                }
+                            }
+                        } catch (Exception e) {
+                            logger.error("Error processing service version URL for jobSetId: {}", jobSetId, e);
+                        }
+                    }
+                }
+            }
+        } catch (Exception e) {
+            logger.error("Error executing service version URL task", e);
+        }
+    }
+
     /**
      * 将字符串转换为首字母大写的格式
      * @param str 原始字符串

ruoyi-modules/management-platform/src/main/resources/mapper/managementPlatform/ZSServiceVersionMapper.xml

@@ -200,4 +200,10 @@
         order by create_time desc
     </select>
 
+    <select id="queryAllRunList" resultType="com.ruoyi.platform.domain.service.ServiceVersionZs">
+        <include refid="selectServiceVersionZsVo" />
+        where run_state = 'Running'
+        order by create_time desc
+    </select>
+
 </mapper>