refactor(service): 重构模型网关服务结构

controller/model_gateway_task_controller.go

@@ -0,0 +1,43 @@
+package controller
+
+import (
+	"context"
+	taskService "model-gateway/service/task"
+
+	"model-gateway/model/dto"
+)
+
+// ModelGatewayTask 任务控制器
+var ModelGatewayTask = new(task)
+
+type task struct{}
+
+// CreateTask 根据 modelName 创建异步任务，返回 taskId
+func (c *task) CreateTask(ctx context.Context, req *dto.CreateTaskReq) (res *dto.CreateTaskRes, err error) {
+	return taskService.ModelGatewayTask.Create(ctx, req)
+}
+
+// GetTaskResult 获取单条任务结果（返回 *dto.GetTaskResultRes）
+func (c *task) GetTaskResult(ctx context.Context, req *dto.GetTaskResultReq) (res *dto.GetTaskResultRes, err error) {
+	return taskService.ModelGatewayTask.GetResult(ctx, req.TaskID)
+}
+
+// GetTaskBatch 批量查询任务（返回 *[]dto.GetTaskBatchItem）
+func (c *task) GetTaskBatch(ctx context.Context, req *dto.GetTaskBatchReq) (res *dto.GetTaskBatchRes, err error) {
+	return taskService.ModelGatewayTask.GetBatch(ctx, req)
+}
+
+// ListTask 任务列表分页查询
+func (c *task) ListTask(ctx context.Context, req *dto.ListTaskReq) (res *dto.ListTaskRes, err error) {
+	return taskService.ModelGatewayTask.List(ctx, req)
+}
+
+// ModelTaskCallback 接收模型异步任务的回调通知 —— 待调整
+func (c *task) ModelTaskCallback(ctx context.Context, req *dto.ModelTaskCallbackReq) (res *dto.ModelTaskCallbackRes, err error) {
+	return taskService.ModelGatewayTask.ModelTaskCallback(ctx, req)
+}
+
+// QueryPendingTasks 批量轮询进行中的异步任务 —— 待调整
+func (c *task) QueryPendingTasks(ctx context.Context, req *dto.QueryPendingTasksReq) (res *dto.QueryPendingTasksRes, err error) {
+	return taskService.ModelGatewayTask.QueryPendingTasks(ctx, req)
+}