updated AiDeployInstanceTask

internal/logic/inference/createdeploytasklogic.go

@@ -50,7 +50,7 @@ func (l *CreateDeployTaskLogic) CreateDeployTask(req *types.CreateDeployTaskReq)
 		return nil, errors.New("TaskName already exists")
 	}
 
-	taskId, err := l.svcCtx.Scheduler.AiStorages.SaveInferDeployTask(req.TaskName, req.ModelName, req.ModelType, req.TaskDesc)
+	taskId, err := l.svcCtx.Scheduler.AiStorages.SaveInferDeployTask(req.TaskName, 0, req.ModelName, req.ModelType, req.TaskDesc)
 	if err != nil {
 		return nil, err
 	}

internal/logic/inference/createinferencetasklogic.go

@@ -68,7 +68,7 @@ func (l *CreateInferenceTaskLogic) CreateInferenceTask(req *types.CreateInferenc
 		Cmd:       "",
 	}
 
-	taskId, err := l.svcCtx.Scheduler.AiStorages.SaveInferDeployTask(taskName, modelName, "", req.Description)
+	taskId, err := l.svcCtx.Scheduler.AiStorages.SaveInferDeployTask(taskName, req.UserId, modelName, "", req.Description)
 	if err != nil {
 		return nil, err
 	}

internal/scheduler/database/aiStorage.go

@@ -628,11 +628,12 @@ func (s *AiStorage) GetTrainingTaskRunningNum() (int32, error) {
 	return total, nil
 }
 
-func (s *AiStorage) SaveInferDeployTask(taskName string, modelName string, modelType string, desc string) (int64, error) {
+func (s *AiStorage) SaveInferDeployTask(taskName string, userId int64, modelName string, modelType string, desc string) (int64, error) {
 	startTime := time.Now().Format(time.RFC3339)
 	// 构建主任务结构体
 	taskModel := models.AiDeployInstanceTask{
 		Name:       taskName,
+		UserId:     userId,
 		ModelName:  modelName,
 		ModelType:  modelType,
 		Desc:       desc,

pkg/models/aideployinstancetaskmodel_gen.go

@@ -37,6 +37,7 @@ type (
 	AiDeployInstanceTask struct {
 		Id         int64  `db:"id"`
 		Name       string `db:"name"`
+		UserId     int64  `db:"user_id"`
 		ModelName  string `db:"model_name"`
 		ModelType  string `db:"model_type"`
 		Desc       string `db:"desc"`