notebook restart record

routers/repo/modelarts.go

@@ -424,6 +424,33 @@ func NotebookManage(ctx *context.Context) {
 		}
 
 		status = task.Status
+		if action == models.ActionRestart {
+			//todo: add a record
+			newTask := &models.Cloudbrain{
+				Status:          string(models.JobWaiting),
+				UserID:          task.UserID,
+				RepoID:          task.RepoID,
+				JobID:           task.JobID,
+				JobName:         task.JobName,
+				DisplayJobName:  task.DisplayJobName,
+				SubTaskName:     task.SubTaskName,
+				JobType:         task.JobType,
+				Type:            task.Type,
+				Uuid:            task.Uuid,
+				Image:           task.Image,
+				GpuQueue:        task.GpuQueue,
+				ResourceSpecId:  task.ResourceSpecId,
+				ComputeResource: task.ComputeResource,
+			}
+
+			err = models.RestartCloudbrain(task, newTask)
+			if err != nil {
+				log.Error("RestartCloudbrain(%s) failed:%v", task.JobName, err.Error(), ctx.Data["MsgID"])
+				resultCode = "-1"
+				errorMsg = "system error"
+				break
+			}
+		}
 
 		break
 	}