add status

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,
 	})
 }
 

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,
 	})
 }