|
|
|
@@ -98,7 +98,7 @@ func (s *TaskStatus) UpdateTaskStatus(tasklist []*types.TaskModel) { |
|
|
|
task.Status = constants.Succeeded |
|
|
|
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 { |
|
|
|
logx.Errorf("reportStatusMessages Error %s", err.Error()) |
|
|
|
} |
|
|
|
@@ -106,7 +106,7 @@ func (s *TaskStatus) UpdateTaskStatus(tasklist []*types.TaskModel) { |
|
|
|
task.Status = constants.Failed |
|
|
|
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 { |
|
|
|
logx.Errorf("reportStatusMessages Error %s", err.Error()) |
|
|
|
} |
|
|
|
@@ -191,7 +191,7 @@ func (s *TaskStatus) updateAiTask(aiTaskList []*models.TaskAi) { |
|
|
|
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{} |
|
|
|
reportMsg := &jcs.TrainReportMessage{ |
|
|
|
Type: "Train", |
|
|
|
@@ -207,7 +207,7 @@ func (s *TaskStatus) reportStatusMessages(task *types.TaskModel, aiTask *models. |
|
|
|
output = aiTask.Output |
|
|
|
} |
|
|
|
|
|
|
|
reportMsg.Status = true |
|
|
|
reportMsg.Status = status |
|
|
|
reportMsg.Message = "" |
|
|
|
reportMsg.ClusterID = strconv.FormatInt(aiTask.ClusterId, 10) |
|
|
|
reportMsg.Output = output |
|
|
|
|