diff --git a/internal/scheduler/service/aiService.go b/internal/scheduler/service/aiService.go index 9d6064f7..ce3479c1 100644 --- a/internal/scheduler/service/aiService.go +++ b/internal/scheduler/service/aiService.go @@ -14,6 +14,7 @@ import ( "gitlink.org.cn/JointCloud/pcm-modelarts/client/modelartsservice" "gitlink.org.cn/JointCloud/pcm-octopus/octopusclient" "strconv" + "sync" ) const ( @@ -30,6 +31,7 @@ type AiService struct { Storage *database.AiStorage LocalCache map[string]interface{} Conf *config.Config + TaskSyncLock sync.Mutex } func NewAiService(conf *config.Config, storages *database.AiStorage, localCache map[string]interface{}) (*AiService, error) { diff --git a/internal/scheduler/service/utils/status/taskStatusSync.go b/internal/scheduler/service/utils/status/taskStatusSync.go index e63bdd9d..4e4ff92b 100644 --- a/internal/scheduler/service/utils/status/taskStatusSync.go +++ b/internal/scheduler/service/utils/status/taskStatusSync.go @@ -159,11 +159,19 @@ func reportStatusMessages(svc *svc.ServiceContext, task *types.TaskModel, aiTask Messages: make([]*jcs.ReportMessage, 0), } //add report msg + var output string + switch aiTask.Name { + case "openI": + output = aiTask.JobId + case "modelarts": + output = aiTask.Output + } + jobMsg := &jcs.ReportMessage{ Status: true, Message: "", ClusterID: strconv.FormatInt(aiTask.ClusterId, 10), - Output: aiTask.JobId, + Output: output, } report.Messages = append(report.Messages, jobMsg)