第一次提交

controller/base.go

@@ -0,0 +1,24 @@
+package controller
+
+import (
+	"context"
+
+	"gitea.com/red-future/common/beans"
+)
+
+// ensureUser 用于本地/无网关环境下的兜底用户信息，避免 gfdb Hook 因缺少用户上下文而报错。
+// 生产环境建议由网关透传 X-User-Info 或鉴权中间件注入 ctx.Value("user")。
+func ensureUser(ctx context.Context) context.Context {
+	if ctx == nil {
+		ctx = context.Background()
+	}
+	if ctx.Value("user") != nil {
+		return ctx
+	}
+	u := &beans.User{
+		UserName: "admin",
+		TenantId: 1,
+	}
+	return context.WithValue(ctx, "user", u)
+}
+

controller/model_controller.go

@@ -0,0 +1,85 @@
+package controller
+
+import (
+	"context"
+
+	"model-asynch/model/dto"
+	"model-asynch/service"
+
+	"gitea.com/red-future/common/beans"
+)
+
+type model struct{}
+
+// Model 模型配置控制器
+var Model = new(model)
+
+// CreateModel 添加配置
+func (c *model) CreateModel(ctx context.Context, req *dto.CreateModelReq) (res *dto.CreateModelRes, err error) {
+	ctx = ensureUser(ctx)
+	return service.Model.Create(ctx, req)
+}
+
+// UpdateModel 更改配置
+func (c *model) UpdateModel(ctx context.Context, req *dto.UpdateModelReq) (res *beans.ResponseEmpty, err error) {
+	ctx = ensureUser(ctx)
+	err = service.Model.Update(ctx, req)
+	return
+}
+
+// DeleteModel 删除配置
+func (c *model) DeleteModel(ctx context.Context, req *dto.DeleteModelReq) (res *beans.ResponseEmpty, err error) {
+	ctx = ensureUser(ctx)
+	err = service.Model.Delete(ctx, req.ID)
+	return
+}
+
+// GetModel 获取配置详情（按 modelName）
+func (c *model) GetModel(ctx context.Context, req *dto.GetModelReq) (res *dto.GetModelRes, err error) {
+	ctx = ensureUser(ctx)
+	m, err := service.Model.Get(ctx, req.ID)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.GetModelRes{Model: m}, nil
+}
+
+// ListModel 配置列表
+func (c *model) ListModel(ctx context.Context, req *dto.ListModelReq) (res *dto.ListModelRes, err error) {
+	ctx = ensureUser(ctx)
+	pageNum, pageSize := 1, 10 //默认分页参数
+	if req != nil && req.Page != nil {
+		if req.Page.PageNum > 0 {
+			pageNum = int(req.Page.PageNum)
+		}
+		if req.Page.PageSize > 0 {
+			pageSize = int(req.Page.PageSize)
+		}
+	}
+	modelName := ""
+	if req != nil {
+		modelName = req.ModelName
+	}
+	list, total, err := service.Model.List(ctx, pageNum, pageSize, modelName)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.ListModelRes{
+		List:  list,
+		Total: total,
+	}, nil
+}
+
+// AutoTune 动态调参（由上层定时任务每小时触发一次）
+func (c *model) AutoTune(ctx context.Context, req *dto.AutoTuneReq) (res *dto.AutoTuneRes, err error) {
+	ctx = ensureUser(ctx)
+	windowSeconds := 3600
+	if req != nil && req.WindowSeconds > 0 {
+		windowSeconds = req.WindowSeconds
+	}
+	list, err := service.AutoTune(ctx, windowSeconds)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.AutoTuneRes{List: list}, nil
+}

controller/model_type_controller.go

@@ -0,0 +1,68 @@
+package controller
+
+import (
+	"context"
+
+	"model-asynch/model/dto"
+	"model-asynch/service"
+
+	"gitea.com/red-future/common/beans"
+)
+
+type modelType struct{}
+
+// ModelType 模型类型控制器
+var ModelType = new(modelType)
+
+func (c *modelType) CreateModelType(ctx context.Context, req *dto.CreateModelTypeReq) (res *dto.CreateModelTypeRes, err error) {
+	ctx = ensureUser(ctx)
+	return service.ModelType.Create(ctx, req)
+}
+
+func (c *modelType) UpdateModelType(ctx context.Context, req *dto.UpdateModelTypeReq) (res *beans.ResponseEmpty, err error) {
+	ctx = ensureUser(ctx)
+	err = service.ModelType.Update(ctx, req)
+	return
+}
+
+func (c *modelType) DeleteModelType(ctx context.Context, req *dto.DeleteModelTypeReq) (res *beans.ResponseEmpty, err error) {
+	ctx = ensureUser(ctx)
+	err = service.ModelType.Delete(ctx, req.ID)
+	return
+}
+
+func (c *modelType) GetModelType(ctx context.Context, req *dto.GetModelTypeReq) (res *dto.GetModelTypeRes, err error) {
+	ctx = ensureUser(ctx)
+	t, err := service.ModelType.Get(ctx, req.ID)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.GetModelTypeRes{Type: t}, nil
+}
+
+func (c *modelType) ListModelType(ctx context.Context, req *dto.ListModelTypeReq) (res *dto.ListModelTypeRes, err error) {
+	ctx = ensureUser(ctx)
+	pageNum, pageSize := 1, 10
+	if req != nil && req.Page != nil {
+		if req.Page.PageNum > 0 {
+			pageNum = int(req.Page.PageNum)
+		}
+		if req.Page.PageSize > 0 {
+			pageSize = int(req.Page.PageSize)
+		}
+	}
+	typeName := ""
+	if req != nil {
+		typeName = req.TypeName
+	}
+	list, total, err := service.ModelType.List(ctx, pageNum, pageSize, typeName)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.ListModelTypeRes{List: list, Total: total}, nil
+}
+
+func (c *modelType) ListModelTypeWithModels(ctx context.Context, req *dto.ListModelTypeWithModelsReq) (res []dto.ModelTypeWithModelsItem, err error) {
+	ctx = ensureUser(ctx)
+	return service.ModelType.ListWithModels(ctx, req)
+}

controller/stat_controller.go

@@ -0,0 +1,20 @@
+package controller
+
+import (
+	"context"
+
+	"model-asynch/model/dto"
+	"model-asynch/service"
+)
+
+type stat struct{}
+
+// Stat 统计控制器
+var Stat = new(stat)
+
+// ListModelStat 统计列表
+func (c *stat) ListModelStat(ctx context.Context, req *dto.ListModelStatReq) (res *dto.ListModelStatRes, err error) {
+	ctx = ensureUser(ctx)
+	return service.Stat.List(ctx, req)
+}
+

controller/task_controller.go

@@ -0,0 +1,63 @@
+package controller
+
+import (
+	"context"
+
+	"model-asynch/model/dto"
+	"model-asynch/service"
+)
+
+type task struct{}
+
+// Task 任务控制器
+var Task = new(task)
+
+// CreateTask 根据 modelName 创建异步任务，返回 taskId
+func (c *task) CreateTask(ctx context.Context, req *dto.CreateTaskReq) (res *dto.CreateTaskRes, err error) {
+	ctx = ensureUser(ctx)
+	return service.Task.Create(ctx, req)
+}
+
+// GetTaskResult 获取任务结果（只返回 oss 地址 + state）
+func (c *task) GetTaskResult(ctx context.Context, req *dto.GetTaskResultReq) (res *dto.GetTaskResultRes, err error) {
+	ctx = ensureUser(ctx)
+	return service.Task.GetResult(ctx, req.TaskID)
+}
+
+// GetTaskBatch 批量查询任务（成功任务标记为已下载）
+func (c *task) GetTaskBatch(ctx context.Context, req *dto.GetTaskBatchReq) (res *dto.GetTaskBatchRes, err error) {
+	ctx = ensureUser(ctx)
+	return service.Task.GetBatch(ctx, req)
+}
+
+// ListTask 任务列表分页查询
+func (c *task) ListTask(ctx context.Context, req *dto.ListTaskReq) (res *dto.ListTaskRes, err error) {
+	ctx = ensureUser(ctx)
+	return service.Task.List(ctx, req)
+}
+
+// RunWork 手动触发一次 worker（由上层定时任务调用）
+func (c *task) RunWork(ctx context.Context, req *dto.RunWorkReq) (res *dto.RunWorkRes, err error) {
+	ctx = ensureUser(ctx)
+	batchSize, goroutines := 10, 1
+	if req != nil {
+		if req.BatchSize > 0 {
+			batchSize = req.BatchSize
+		}
+		if req.Goroutines > 0 {
+			goroutines = req.Goroutines
+		}
+	}
+	n, err := service.AsyncWorker.RunOnce(ctx, batchSize, goroutines)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.RunWorkRes{Claimed: n}, nil
+}
+
+// CleanWork 手动触发一次 cleaner（由上层定时任务调用）
+func (c *task) CleanWork(ctx context.Context, req *dto.CleanWorkReq) (res *dto.CleanWorkRes, err error) {
+	ctx = ensureUser(ctx)
+	service.Cleaner.RunOnce(ctx)
+	return &dto.CleanWorkRes{Ok: true}, nil
+}