diff --git a/api/internal/logic/inference/imageinferencelogic.go b/api/internal/logic/inference/imageinferencelogic.go index 665f2ff7..b80de115 100644 --- a/api/internal/logic/inference/imageinferencelogic.go +++ b/api/internal/logic/inference/imageinferencelogic.go @@ -229,6 +229,7 @@ func infer(opt *option.InferOption, clusters []*strategy.AssignedCluster, ts []s if len(cs) == 0 { for _, t := range aiTaskList { t.Status = constants.Failed + t.EndTime = time.Now().Format(time.RFC3339) err := svcCtx.Scheduler.AiStorages.UpdateAiTask(t) if err != nil { logx.Errorf(tx.Error.Error()) @@ -256,6 +257,7 @@ func infer(opt *option.InferOption, clusters []*strategy.AssignedCluster, ts []s for _, t := range aiTaskList { if ac.ClusterId == strconv.Itoa(int(t.ClusterId)) { t.Status = constants.Failed + t.EndTime = time.Now().Format(time.RFC3339) err := svcCtx.Scheduler.AiStorages.UpdateAiTask(t) if err != nil { logx.Errorf(tx.Error.Error()) @@ -321,6 +323,7 @@ func infer(opt *option.InferOption, clusters []*strategy.AssignedCluster, ts []s for _, t := range aiTaskList { if c.clusterId == strconv.Itoa(int(t.ClusterId)) { t.Status = constants.Completed + t.EndTime = time.Now().Format(time.RFC3339) err := svcCtx.Scheduler.AiStorages.UpdateAiTask(t) if err != nil { logx.Errorf(tx.Error.Error()) diff --git a/api/internal/scheduler/database/aiStorage.go b/api/internal/scheduler/database/aiStorage.go index a812f7e2..8e4105e7 100644 --- a/api/internal/scheduler/database/aiStorage.go +++ b/api/internal/scheduler/database/aiStorage.go @@ -95,6 +95,7 @@ func (s *AiStorage) GetAiTasksByAdapterId(adapterId string) ([]*models.TaskAi, e } func (s *AiStorage) SaveTask(name string, strategyCode int64, synergyStatus int64, aiType string) (int64, error) { + startTime := time.Now() // 构建主任务结构体 taskModel := models.Task{ Status: constants.Saved, @@ -104,6 +105,7 @@ func (s *AiStorage) SaveTask(name string, strategyCode int64, synergyStatus int6 Strategy: strategyCode, AdapterTypeDict: "1", TaskTypeDict: aiType, + StartTime: &startTime, CommitTime: time.Now(), } // 保存任务数据到数据库 @@ -152,6 +154,7 @@ func (s *AiStorage) SaveAiTask(taskId int64, opt option.Option, adapterName stri Status: status, Msg: msg, Card: aiOpt.ComputeCard, + StartTime: time.Now().Format(time.RFC3339), CommitTime: time.Now(), } // 保存任务数据到数据库