From 8bf5b6898ef26f705c4e7faad88e046cef8a3976 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Thu, 23 Jun 2022 15:17:42 +0800
Subject: [PATCH] refresh ai center

---
 routers/api/v1/repo/modelarts.go          | 18 +++++++++++++++++-
 templates/repo/grampus/trainjob/show.tmpl |  3 ++-
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/routers/api/v1/repo/modelarts.go b/routers/api/v1/repo/modelarts.go
index e9bf9a1f5..2832add00 100755
--- a/routers/api/v1/repo/modelarts.go
+++ b/routers/api/v1/repo/modelarts.go
@@ -127,7 +127,8 @@ func GetModelArtsTrainJob(ctx *context.APIContext) {
 
 func GetModelArtsTrainJobVersion(ctx *context.APIContext) {
 	var (
-		err error
+		err          error
+		aiCenterName string
 	)
 
 	jobID := ctx.Params(":jobid")
@@ -210,6 +211,20 @@ func GetModelArtsTrainJobVersion(ctx *context.APIContext) {
 			job.EndTime = job.StartTime.Add(job.Duration)
 		}
 		job.CorrectCreateUnix()
+
+		if len(job.AiCenter) == 0 {
+			if len(result.JobInfo.Tasks) > 0 {
+				if len(result.JobInfo.Tasks[0].CenterID[0]) > 0 && len(result.JobInfo.Tasks[0].CenterName[0]) > 0 {
+					job.AiCenter = result.JobInfo.Tasks[0].CenterID[0] + "+" + result.JobInfo.Tasks[0].CenterName[0]
+					aiCenterName = result.JobInfo.Tasks[0].CenterName[0]
+				}
+			}
+		} else {
+			temp := strings.Split(job.AiCenter, "+")
+			if len(temp) > 1 {
+				aiCenterName = temp[1]
+			}
+		}
 		err = models.UpdateTrainJobVersion(job)
 		if err != nil {
 			log.Error("UpdateJob failed:", err)
@@ -220,6 +235,7 @@ func GetModelArtsTrainJobVersion(ctx *context.APIContext) {
 		"JobID":       jobID,
 		"JobStatus":   job.Status,
 		"JobDuration": job.TrainJobDuration,
+		"AiCenter":    aiCenterName,
 	})
 
 }
diff --git a/templates/repo/grampus/trainjob/show.tmpl b/templates/repo/grampus/trainjob/show.tmpl
index af924e5cd..579e83693 100755
--- a/templates/repo/grampus/trainjob/show.tmpl
+++ b/templates/repo/grampus/trainjob/show.tmpl
@@ -430,7 +430,7 @@
                                                     </td>
 
                                                     <td class="ti-text-form-content">
-                                                        <div class="text-span text-span-w">
+                                                        <div class="text-span text-span-w" id="{{.VersionName}}-ai_center">
                                                             {{$.ai_center}}
                                                         </div>
                                                     </td>
@@ -679,6 +679,7 @@
             // detail status and duration
             $('#' + version_name + '-duration').text(data.JobDuration)
             $('#' + version_name + '-status').text(data.JobStatus)
+            $('#' + version_name + '-ai_center').text(data.AiCenter)
             loadLog(version_name)