From b27c724f4ac6e30c5a0d1186d6a5b4d5175da293 Mon Sep 17 00:00:00 2001
From: lewis <747342561@qq.com>
Date: Mon, 14 Feb 2022 11:30:24 +0800
Subject: [PATCH] debug

---
 models/cloudbrain.go      |  9 +++++----
 routers/repo/modelarts.go | 11 +++++++++--
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index 5f3b47dc8..8938525db 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -674,12 +674,13 @@ type GetNotebook2Result struct {
 	Name             string `json:"name"`
 	Description      string `json:"description"`
 	Status           string `json:"status"`
-	Url              string `json:"url"` //实例访问的URL
+	Url              string `json:"url"`   //实例访问的URL
+	Token            string `json:"token"` //notebook鉴权使用的token信息
 	Flavor           string `json:"flavor"`
 	CreateTime       string
 	LatestUpdateTime string
-	CreateAt int64 `json:"create_at"` //实例创建的时间，UTC毫秒
-	UpdateAt int64 `json:"update_at"` //实例最后更新（不包括保活心跳）的时间，UTC毫秒
+	CreateAt         int64 `json:"create_at"` //实例创建的时间，UTC毫秒
+	UpdateAt         int64 `json:"update_at"` //实例最后更新（不包括保活心跳）的时间，UTC毫秒
 	Image            struct {
 		Name          string `json:"name"`
 		Status        string `json:"status"`
@@ -689,7 +690,7 @@ type GetNotebook2Result struct {
 	} `json:"image"`
 	Lease struct {
 		CreateTime int64 `json:"create_at"` //实例创建的时间，UTC毫秒
-		Duration   int64 `json:"duration"`    //实例运行时长，以创建时间为起点计算，即“创建时间+duration > 当前时刻”时，系统会自动停止实例
+		Duration   int64 `json:"duration"`  //实例运行时长，以创建时间为起点计算，即“创建时间+duration > 当前时刻”时，系统会自动停止实例
 		UpdateTime int64 `json:"update_at"` //实例最后更新（不包括保活心跳）的时间，UTC毫秒
 	} `json:"lease"` //实例自动停止的倒计时信息
 	VolumeRes struct {
diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 834d6dcf6..9226ddbb4 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -284,8 +284,15 @@ func NotebookDebug(ctx *context.Context) {
 }
 
 func NotebookDebug2(ctx *context.Context) {
-	debugUrl := setting.DebugHost + "/" + ctx.Cloudbrain.JobID + "/lab"
-	ctx.Redirect(debugUrl)
+	var jobID = ctx.Params(":jobid")
+
+	result, err := modelarts.GetNotebook2(jobID)
+	if err != nil {
+		ctx.RenderWithErr(err.Error(), tplModelArtsNotebookIndex, nil)
+		return
+	}
+	
+	ctx.Redirect(result.Url)
 }
 
 func NotebookManage(ctx *context.Context) {