From 676300ce9aaa35ea56220fd6cb42f8da38434db1 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Mon, 14 Feb 2022 17:04:25 +0800
Subject: [PATCH] manage notebook status

---
 models/cloudbrain.go      | 1 +
 routers/repo/modelarts.go | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index 8938525db..0a14ea7b4 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -755,6 +755,7 @@ type NotebookActionResult struct {
 	ErrorMsg      string `json:"error_msg"`
 	CurrentStatus string `json:"current_status"`
 	PreviousState string `json:"previous_state"`
+	Status        string `json:"status"`
 }
 
 type NotebookGetJobTokenResult struct {
diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 52c7cd561..8479af93d 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -390,7 +390,7 @@ func NotebookManage(ctx *context.Context) {
 			break
 		}
 
-		task.Status = res.CurrentStatus
+		task.Status = res.Status
 		err = models.UpdateJob(task)
 		if err != nil {
 			log.Error("UpdateJob(%s) failed:%v", task.JobName, err.Error(), ctx.Data["MsgID"])