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