Browse Source

#1852

update
tags/v1.22.4.1^2
chenyifan01 3 years ago
parent
commit
954f3939be
3 changed files with 5 additions and 1 deletions
  1. +1
    -1
      models/cloudbrain.go
  2. +1
    -0
      modules/cloudbrain/cloudbrain.go
  3. +3
    -0
      routers/repo/modelarts.go

+ 1
- 1
models/cloudbrain.go View File

@@ -1572,7 +1572,7 @@ func RestartCloudbrain(old *Cloudbrain, new *Cloudbrain) (err error) {
return err
}

if _, err = sess.Insert(new); err != nil {
if _, err = sess.NoAutoTime().Insert(new); err != nil {
sess.Rollback()
return err
}


+ 1
- 0
modules/cloudbrain/cloudbrain.go View File

@@ -438,6 +438,7 @@ func RestartTask(ctx *context.Context, task *models.Cloudbrain, newID *string) e
ResourceSpecId: task.ResourceSpecId,
ComputeResource: task.ComputeResource,
CreatedUnix: createTime,
UpdatedUnix: createTime,
}

err = models.RestartCloudbrain(task, newTask)


+ 3
- 0
routers/repo/modelarts.go View File

@@ -439,6 +439,7 @@ func NotebookManage(ctx *context.Context) {
param := models.NotebookAction{
Action: action,
}
createTime := timeutil.TimeStampNow()
res, err := modelarts.ManageNotebook2(task.JobID, param)
if err != nil {
log.Error("ManageNotebook2(%s) failed:%v", task.JobName, err.Error(), ctx.Data["MsgID"])
@@ -465,6 +466,8 @@ func NotebookManage(ctx *context.Context) {
Image: task.Image,
ComputeResource: task.ComputeResource,
Description: task.Description,
CreatedUnix: createTime,
UpdatedUnix: createTime,
}

err = models.RestartCloudbrain(task, newTask)


Loading…
Cancel
Save