From 22b9bb5610ef75a598fa11b26d37c4dfdeec3af9 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Wed, 22 Dec 2021 09:35:22 +0800
Subject: [PATCH] add status

---
 routers/repo/cloudbrain.go | 4 ++++
 routers/repo/modelarts.go  | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index e12bc816a..4fcb3c5f3 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -258,6 +258,7 @@ func CloudBrainRestart(ctx *context.Context) {
 	var jobID = ctx.Params(":jobid")
 	var resultCode = "0"
 	var errorMsg = ""
+	var status = ""
 
 	for {
 		task, err := models.GetCloudbrainByJobID(jobID)
@@ -298,12 +299,15 @@ func CloudBrainRestart(ctx *context.Context) {
 			break
 		}
 
+		status = task.Status
+
 		break
 	}
 
 	ctx.JSON(200, map[string]string{
 		"result_code": resultCode,
 		"error_msg":   errorMsg,
+		"status":      status,
 	})
 }
 
diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index fd301d861..112593c47 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -234,6 +234,7 @@ func NotebookManage(ctx *context.Context) {
 	var action = ctx.Params(":action")
 	var resultCode = "0"
 	var errorMsg = ""
+	var status = ""
 
 	for {
 		task, err := models.GetCloudbrainByJobID(jobID)
@@ -302,12 +303,15 @@ func NotebookManage(ctx *context.Context) {
 			break
 		}
 
+		status = task.Status
+
 		break
 	}
 
 	ctx.JSON(200, map[string]string{
 		"result_code": resultCode,
 		"error_msg":   errorMsg,
+		"status":      status,
 	})
 }