Browse Source

Merge pull request 'fix statusSync err' (#482) from tzwang/pcm-coordinator:master into master

pull/484/head
tzwang 6 months ago
parent
commit
db00179bde
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      internal/scheduler/service/utils/status/statusSync.go

+ 4
- 4
internal/scheduler/service/utils/status/statusSync.go View File

@@ -98,7 +98,7 @@ func (s *TaskStatus) UpdateTaskStatus(tasklist []*types.TaskModel) {
task.Status = constants.Succeeded task.Status = constants.Succeeded
logx.Errorf("############ Report Status Message Before Sending %s", task.Status) logx.Errorf("############ Report Status Message Before Sending %s", task.Status)


err = s.reportStatusMessages(task, aiTask[0])
err = s.reportStatusMessages(task, aiTask[0], true)
if err != nil { if err != nil {
logx.Errorf("reportStatusMessages Error %s", err.Error()) logx.Errorf("reportStatusMessages Error %s", err.Error())
} }
@@ -106,7 +106,7 @@ func (s *TaskStatus) UpdateTaskStatus(tasklist []*types.TaskModel) {
task.Status = constants.Failed task.Status = constants.Failed
logx.Errorf("############ Report Status Message Before Sending %s", task.Status) logx.Errorf("############ Report Status Message Before Sending %s", task.Status)


err = s.reportStatusMessages(task, aiTask[0])
err = s.reportStatusMessages(task, aiTask[0], false)
if err != nil { if err != nil {
logx.Errorf("reportStatusMessages Error %s", err.Error()) logx.Errorf("reportStatusMessages Error %s", err.Error())
} }
@@ -191,7 +191,7 @@ func (s *TaskStatus) updateAiTask(aiTaskList []*models.TaskAi) {
wg.Wait() wg.Wait()
} }


func (s *TaskStatus) reportStatusMessages(task *types.TaskModel, aiTask *models.TaskAi) error {
func (s *TaskStatus) reportStatusMessages(task *types.TaskModel, aiTask *models.TaskAi, status bool) error {
report := &jcs.JobStatusReportReq{} report := &jcs.JobStatusReportReq{}
reportMsg := &jcs.TrainReportMessage{ reportMsg := &jcs.TrainReportMessage{
Type: "Train", Type: "Train",
@@ -207,7 +207,7 @@ func (s *TaskStatus) reportStatusMessages(task *types.TaskModel, aiTask *models.
output = aiTask.Output output = aiTask.Output
} }


reportMsg.Status = true
reportMsg.Status = status
reportMsg.Message = "" reportMsg.Message = ""
reportMsg.ClusterID = strconv.FormatInt(aiTask.ClusterId, 10) reportMsg.ClusterID = strconv.FormatInt(aiTask.ClusterId, 10)
reportMsg.Output = output reportMsg.Output = output


Loading…
Cancel
Save