merge

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)