| @@ -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 | ||||