From 215b67061fa500858a627fc514f33f4a57e2bc7d Mon Sep 17 00:00:00 2001
From: chenyifan01 <gitea@fake.local>
Date: Mon, 28 Mar 2022 11:45:52 +0800
Subject: [PATCH] #1654 fix bug

---
 routers/repo/cloudbrain.go | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 2d4b4f279..1b83c86ab 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -1084,7 +1084,7 @@ func HandleTaskWithNoDuration(ctx *context.Context) {
 		}
 	}
 	log.Info("HandleTaskWithNoTrainJobDuration:count=%d", count)
-
+	ctx.JSON(200, "success")
 }
 
 func handleNoDurationTask(cloudBrains []*models.Cloudbrain) {
@@ -1241,13 +1241,16 @@ func CloudBrainBenchmarkIndex(ctx *context.Context) {
 		ciTasks[i].CanDel = cloudbrain.CanDeleteJob(ctx, &task.Cloudbrain)
 		ciTasks[i].Cloudbrain.ComputeResource = task.ComputeResource
 		if ciTasks[i].TrainJobDuration == "" {
-			var duration int64
-			if task.Status == string(models.JobRunning) {
-				duration = time.Now().Unix() - int64(task.Cloudbrain.CreatedUnix)
-			} else {
-				duration = int64(task.Cloudbrain.UpdatedUnix) - int64(task.Cloudbrain.CreatedUnix)
+			if ciTasks[i].Duration == 0 {
+				var duration int64
+				if task.Status == string(models.JobRunning) {
+					duration = time.Now().Unix() - int64(task.Cloudbrain.CreatedUnix)
+				} else {
+					duration = int64(task.Cloudbrain.UpdatedUnix) - int64(task.Cloudbrain.CreatedUnix)
+				}
+				ciTasks[i].Duration = duration
 			}
-			ciTasks[i].TrainJobDuration = models.ConvertDurationToStr(duration)
+			ciTasks[i].TrainJobDuration = models.ConvertDurationToStr(ciTasks[i].Duration)
 		}
 
 		ciTasks[i].BenchmarkTypeName = ""