refactor(model): 优化模型网关的数据解析和任务处理逻辑

service/task/task_service.go

@@ -107,13 +107,27 @@ func (s *taskService) Create(ctx context.Context, req *dto.CreateTaskReq) (res *
 		},
 	})
 
-	// 5) 获取任务信息
-	task, err := dao.ModelGatewayTask.ClaimByID(ctx, id)
+	// 5) 抢占任务：改为执行中
+	rows, err := dao.ModelGatewayTask.Update(ctx, &entity.ModelGatewayTask{
+		SQLBaseDO: beans.SQLBaseDO{Id: id},
+		State:     public.TaskStatusRunning,
+	})
+	if err != nil {
+		return nil, err
+	}
+	if rows == 0 {
+		return nil, fmt.Errorf("任务不存在: id=%d", id)
+	}
+
+	// 6) 查询任务信息
+	task, err := dao.ModelGatewayTask.Get(ctx, &entity.ModelGatewayTask{
+		SQLBaseDO: beans.SQLBaseDO{Id: id},
+	})
 	if err != nil {
 		return nil, err
 	}
 
-	// 5) 创建成功后立即异步尝试执行当前任务
+	// 7) 创建成功后立即异步尝试执行当前任务
 	go AsyncWorker.handleOne(util.AsyncCtx(ctx), task, model, req)
 
 	return &dto.CreateTaskRes{TaskID: taskID}, nil