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

controller/model_gateway_logs_stat_controller.go

@@ -7,12 +7,12 @@ import (
 	"model-gateway/model/dto"
 )
 
-type stat struct{}
+// ModelGatewayLogsStat 统计控制器
+var ModelGatewayLogsStat = new(stat)
 
-// Stat 统计控制器
-var Stat = new(stat)
+type stat struct{}
 
 // ListModelStat 统计列表
 func (c *stat) ListModelStat(ctx context.Context, req *dto.ListModelStatReq) (res *dto.ListModelStatRes, err error) {
-	return statService.Stat.List(ctx, req)
+	return statService.ModelGatewayLogsStat.List(ctx, req)
 }

controller/model_gateway_models_controller.go

@@ -7,36 +7,36 @@ import (
 	"model-gateway/service/queue"
 )
 
-type model struct{}
+// ModelGatewayModels 模型配置控制器
+var ModelGatewayModels = new(model)
 
-// Model 模型配置控制器
-var Model = new(model)
+type model struct{}
 
 // CreateModel 添加配置
 func (c *model) CreateModel(ctx context.Context, req *dto.CreateModelReq) (res *dto.CreateModelRes, err error) {
-	return modelService.Model.Create(ctx, req)
+	return modelService.ModelGatewayModels.Create(ctx, req)
 }
 
 // UpdateModel 更改配置
 func (c *model) UpdateModel(ctx context.Context, req *dto.UpdateModelReq) (res *dto.UpdateModelRes, err error) {
-	err = modelService.Model.Update(ctx, req)
+	err = modelService.ModelGatewayModels.Update(ctx, req)
 	return
 }
 
 // DeleteModel 删除配置
 func (c *model) DeleteModel(ctx context.Context, req *dto.DeleteModelReq) (res *dto.DeleteModelRes, err error) {
-	err = modelService.Model.Delete(ctx, req)
+	err = modelService.ModelGatewayModels.Delete(ctx, req)
 	return
 }
 
 // GetModel 获取配置详情
 func (c *model) GetModel(ctx context.Context, req *dto.GetModelReq) (res *dto.GetModelRes, err error) {
-	return modelService.Model.Get(ctx, req)
+	return modelService.ModelGatewayModels.Get(ctx, req)
 }
 
 // ListModel 配置列表
 func (c *model) ListModel(ctx context.Context, req *dto.ListModelReq) (res *dto.ListModelRes, err error) {
-	return modelService.Model.List(ctx, req)
+	return modelService.ModelGatewayModels.List(ctx, req)
 }
 
 // AutoTune 动态调参（由上层定时任务每小时触发一次）
@@ -56,11 +56,11 @@ func (c *model) ListOperator(ctx context.Context, req *dto.ListOperatorReq) (res
 
 // UpdateChatModel 更新是否为聊天模型
 func (c *model) UpdateChatModel(ctx context.Context, req *dto.UpdateChatModelReq) (res *dto.UpdateChatModelRes, err error) {
-	err = modelService.Model.UpdateChatModel(ctx, req)
+	err = modelService.ModelGatewayModels.UpdateChatModel(ctx, req)
 	return
 }
 
 // GetIsChatModel 获取当前会话模型
 func (c *model) GetIsChatModel(ctx context.Context, req *dto.GetIsChatModelReq) (res *dto.GetIsChatModelRes, err error) {
-	return modelService.Model.GetIsChatModel(ctx)
+	return modelService.ModelGatewayModels.GetIsChatModel(ctx)
 }

controller/model_gateway_task_controller.go

@@ -2,48 +2,42 @@ package controller
 
 import (
 	"context"
-	"model-gateway/service/job"
 	taskService "model-gateway/service/task"
 
 	"model-gateway/model/dto"
 )
 
-type task struct{}
+// ModelGatewayTask 任务控制器
+var ModelGatewayTask = new(task)
 
-// Task 任务控制器
-var Task = 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.Task.Create(ctx, req)
+	return taskService.ModelGatewayTask.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）
+// GetTaskResult 获取单条任务结果（返回 *dto.GetTaskResultRes）
 func (c *task) GetTaskResult(ctx context.Context, req *dto.GetTaskResultReq) (res *dto.GetTaskResultRes, err error) {
-	return taskService.Task.GetResult(ctx, req.TaskID)
+	return taskService.ModelGatewayTask.GetResult(ctx, req.TaskID)
 }
 
-// GetTaskBatch 批量查询任务（成功任务标记为已下载）
+// GetTaskBatch 批量查询任务（返回 *[]dto.GetTaskBatchItem）
 func (c *task) GetTaskBatch(ctx context.Context, req *dto.GetTaskBatchReq) (res *dto.GetTaskBatchRes, err error) {
-	return taskService.Task.GetBatch(ctx, req)
+	return taskService.ModelGatewayTask.GetBatch(ctx, req)
 }
 
 // ListTask 任务列表分页查询
 func (c *task) ListTask(ctx context.Context, req *dto.ListTaskReq) (res *dto.ListTaskRes, err error) {
-	return taskService.Task.List(ctx, req)
+	return taskService.ModelGatewayTask.List(ctx, req)
 }
 
-// CleanWork 手动触发一次 cleaner（由上层定时任务调用）
-func (c *task) CleanWork(ctx context.Context, req *dto.CleanWorkReq) (res *dto.CleanWorkRes, err error) {
-	return job.Cleaner.RunOnce(ctx)
+// 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)
 }