From 2e7af72003cf28d8a37ed85b973b2df35717a323 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Wed, 22 Dec 2021 10:24:55 +0800
Subject: [PATCH] add job_id

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

diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 6c0169b11..34647a146 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -300,6 +300,7 @@ func CloudBrainRestart(ctx *context.Context) {
 		}
 
 		status = task.Status
+		jobID = task.JobID
 
 		break
 	}
@@ -308,6 +309,7 @@ func CloudBrainRestart(ctx *context.Context) {
 		"result_code": resultCode,
 		"error_msg":   errorMsg,
 		"status":      status,
+		"job_id":      jobID,
 	})
 }
 
@@ -460,6 +462,7 @@ func CloudBrainStop(ctx *context.Context) {
 		"result_code": resultCode,
 		"error_msg":   errorMsg,
 		"status":      status,
+		"job_id":      jobID,
 	})
 }
 
diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 112593c47..8c34eb0b2 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -312,6 +312,7 @@ func NotebookManage(ctx *context.Context) {
 		"result_code": resultCode,
 		"error_msg":   errorMsg,
 		"status":      status,
+		"job_id":      jobID,
 	})
 }