diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go index ab5070262..1591e1dd5 100755 --- a/modules/modelarts/modelarts.go +++ b/modules/modelarts/modelarts.go @@ -931,6 +931,7 @@ func HandleTrainJobInfo(task *models.Cloudbrain) error { } if result != nil { + oldStatus := task.Status task.Status = TransTrainJobStatus(result.IntStatus) task.Duration = result.Duration / 1000 task.TrainJobDuration = result.TrainJobDuration @@ -943,6 +944,9 @@ func HandleTrainJobInfo(task *models.Cloudbrain) error { task.EndTime = task.StartTime.Add(task.Duration) } task.CorrectCreateUnix() + if oldStatus != task.Status { + notification.NotifyChangeCloudbrainStatus(task, oldStatus) + } err = models.UpdateJob(task) if err != nil { log.Error("UpdateJob(%s) failed:%v", task.JobName, err) @@ -1009,6 +1013,7 @@ func HandleNotebookInfo(task *models.Cloudbrain) error { } if result != nil { + oldStatus := task.Status task.Status = result.Status if task.StartTime == 0 && result.Lease.UpdateTime > 0 { task.StartTime = timeutil.TimeStamp(result.Lease.UpdateTime / 1000) @@ -1018,6 +1023,9 @@ func HandleNotebookInfo(task *models.Cloudbrain) error { } task.CorrectCreateUnix() task.ComputeAndSetDuration() + if oldStatus != task.Status { + notification.NotifyChangeCloudbrainStatus(task, oldStatus) + } err = models.UpdateJob(task) if err != nil { log.Error("UpdateJob(%s) failed:%v", task.DisplayJobName, err)