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

refactor(asynch): 重构异步模型配置和队列管理

Browse Source
This commit is contained in:
王利钊
2026-06-02 20:26:45 +08:00
parent c7e9eb889b
commit 52124385a1
18 changed files with 726 additions and 1006 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
controller/task_controller.go
View File

@@ -18,6 +18,16 @@ func (c *task) CreateTask(ctx context.Context, req *dto.CreateTaskReq) (res *dto
return taskService.Task.Create(ctx, req)
}
// ModelTaskCallback 接收模型异步任务的回调通知
func (c *task) ModelTaskCallback(ctx context.Context, req *dto.ModelTaskCallbackReq) (res *dto.ModelTaskCallbackRes, err error) {
return taskService.Task.ModelTaskCallback(ctx, req)
}
// QueryPendingTasks 批量轮询进行中的异步任务
func (c *task) QueryPendingTasks(ctx context.Context, req *dto.QueryPendingTasksReq) (res *dto.QueryPendingTasksRes, err error) {
return taskService.Task.QueryPendingTasks(ctx, req)
}
// GetTaskResult 获取任务结果（只返回 oss 地址 + state）
func (c *task) GetTaskResult(ctx context.Context, req *dto.GetTaskResultReq) (res *dto.GetTaskResultRes, err error) {
return taskService.Task.GetResult(ctx, req.TaskID)