first commit

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,67 @@
+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)
+		}
+	}
+	list, total, err := service.Model.List(ctx, pageNum, pageSize)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.ListModelRes{
+		List:  list,
+		Total: total,
+	}, nil
+}

controller/task_controller.go

@@ -0,0 +1,31 @@
+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)
+}