Fix task status

db/comfyui_task.sql

@@ -10,7 +10,8 @@ CREATE TABLE `comfyui_task` (
     `id` BIGINT AUTO_INCREMENT COMMENT 'ID',
     `user_id` BIGINT NOT NULL COMMENT '用户ID',
     `prompt_id` VARCHAR(255) NOT NULL COMMENT '任务ID',
-    `status` TINYINT DEFAULT '-1' COMMENT '任务状态 (-1:未开始, 1:进行中, 2:成功, 3:失败)',
+#     `status` TINYINT DEFAULT '-1' COMMENT '任务状态 (-1:未开始, 1:进行中, 2:成功, 3:失败)',
+    `status` TINYINT DEFAULT '-1' COMMENT '任务状态 (-1:未完成, 1:已完成)',
     `msg` VARCHAR(2000) COMMENT '任务状态信息，当任务失败时展示失败原因（如触发平台的内容风控等）',
     `generate_request` TEXT COMMENT '任务请求原始参数 (JSONString)',
     `generate_response` TEXT COMMENT '任务生成原始结果 (JSONString)',

src/main/java/com/backendsys/modules/sdk/comfyui/entity/ComfyuiTask.java

@@ -13,7 +13,7 @@ public class ComfyuiTask {
     private Long id;
     private Long user_id;                 // 用户ID
     private String prompt_id;             // 任务ID
-    private Integer status;               // 任务状态 (-1:未开始, 1:进行中, 2:成功, 3:失败)
+    private Integer status;               // 任务状态 (-1:未完成, 1:已完成)
     private String msg;                   // 任务状态信息，当任务失败时展示失败原因（如触发平台的内容风控等）
     private String generate_request;      // 任务请求原始参数 (JSONString)
     private String generate_response;     // 任务生成原始结果 (JSONString)

src/main/java/com/backendsys/modules/sdk/comfyui/service/impl/ComfyuiSocketServiceImpl.java

@@ -184,7 +184,7 @@ public class ComfyuiSocketServiceImpl implements ComfyuiSocketService {
                                     data.put("data", dataChildren);
 
                                     // [db] 新增任务记录
-                                    comfyUtil.asyncUpdateTaskRecord(prompt_id, JSONUtil.toJsonStr(dataChildren));
+                                    comfyUtil.asyncUpdateTaskRecord(prompt_id, JSONUtil.toJsonStr(dataChildren), 1);
 
                                 }
 

src/main/java/com/backendsys/modules/sdk/comfyui/utils/ComfyUtil.java

@@ -37,7 +37,7 @@ public class ComfyUtil {
     /**
      * [更新] 生成任务的记录 (异步)
      */
-    public void asyncUpdateTaskRecord(String prompt_id, String response) {
+    public void asyncUpdateTaskRecord(String prompt_id, String response, Integer status) {
         CompletableFuture.runAsync(() -> {
             // [db] 新增任务记录
             try {
@@ -46,10 +46,7 @@ public class ComfyUtil {
                 wrapper.eq(ComfyuiTask::getPrompt_id, prompt_id);
 
                 ComfyuiTask comfyuiTask = new ComfyuiTask();
-
-
-                // 目前 [进行中、失败] 的任务状态 还没有做记录？
-                comfyuiTask.setStatus(2);
+                comfyuiTask.setStatus(status);
                 comfyuiTask.setGenerate_response(response);
                 comfyuiTaskDao.update(comfyuiTask, wrapper);
             } catch (Exception e) {