red-future/model-gateway
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(prompt): 优化任务等待机制并改进数据结构

Browse Source
This commit is contained in:
王利钊
2026-05-21 14:23:34 +08:00
parent 4626d819b5
commit b6cdb8ff1d
2 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
service/gateway/gateway_http_service.go
View File

@@ -45,7 +45,7 @@ func UploadByTask(ctx context.Context, _ *entity.AsynchTask, data []byte, fileEx
return "", err return "", err
} }
contentType := writer.FormDataContentType() contentType := writer.FormDataContentType()
if err := writer.Close(); err != nil { if err = writer.Close(); err != nil {
return "", err return "", err
} }

2
service/task_service.go
View File

@@ -104,7 +104,6 @@ func (s *taskService) Create(ctx context.Context, req *dto.CreateTaskReq) (res *
// 4) 创建成功后立即异步尝试执行当前任务，并仅在任务仍处于 pending(state=0) 时做定向轮询。 // 4) 创建成功后立即异步尝试执行当前任务，并仅在任务仍处于 pending(state=0) 时做定向轮询。
// 一旦任务进入 running/success/failed/downloaded，就停止轮询，避免一直空转。 // 一旦任务进入 running/success/failed/downloaded，就停止轮询，避免一直空转。
go s.pollAndRunUntilPicked(context.WithoutCancel(ctx), taskID, req.EpicycleId) go s.pollAndRunUntilPicked(context.WithoutCancel(ctx), taskID, req.EpicycleId)
return &dto.CreateTaskRes{TaskID: taskID}, nil return &dto.CreateTaskRes{TaskID: taskID}, nil
} }
@@ -162,7 +161,6 @@ func (s *taskService) pollAndRunUntilPicked(ctx context.Context, taskID string,
if stop := tryRun(); stop { if stop := tryRun(); stop {
return return
} }
for { for {
select { select {
case <-ctx.Done(): case <-ctx.Done():