feat(task): 暴露任务取消控制状态

This commit is contained in:
chengcheng
2026-06-25 17:01:32 +08:00
parent 6089aa6085
commit f4e1db1279
4 changed files with 49 additions and 0 deletions
+19
View File
@@ -20,6 +20,25 @@ type TaskCancelResult struct {
Message string `json:"message"`
}
func DescribeTaskCancellation(task store.GatewayTask) TaskCancelResult {
if taskCancelTerminalStatus(task.Status) {
return taskCancelUnavailable(task, "任务已结束,无法取消")
}
if strings.TrimSpace(task.RemoteTaskID) != "" {
return taskCancelUnavailable(task, "任务已提交上游,当前不可取消,请继续查询结果")
}
if strings.TrimSpace(task.Status) != "queued" {
return taskCancelUnavailable(task, "任务已开始执行,当前阶段不可取消,请继续查询结果")
}
return TaskCancelResult{
TaskID: task.ID,
Cancelled: false,
Cancellable: true,
Submitted: false,
Message: "任务仍在本地队列中,可取消",
}
}
func (s *Service) CancelTask(ctx context.Context, taskID string, user *auth.User) (TaskCancelResult, error) {
task, err := s.store.GetTask(ctx, taskID)
if err != nil {