Browse Source

#1654

update
tags/v1.22.3.2^2
chenyifan01 4 years ago
parent
commit
e46903c414
2 changed files with 12 additions and 1 deletions
  1. +8
    -1
      routers/api/v1/repo/modelarts.go
  2. +4
    -0
      routers/repo/cloudbrain.go

+ 8
- 1
routers/api/v1/repo/modelarts.go View File

@@ -6,6 +6,7 @@
package repo

import (
"code.gitea.io/gitea/modules/timeutil"
"net/http"
"strconv"
"strings"
@@ -67,8 +68,14 @@ func GetModelArtsNotebook2(ctx *context.APIContext) {
ctx.NotFound(err)
return
}

if job.StartTime == 0 {
job.StartTime = timeutil.TimeStamp(result.Lease.CreateTime / 1000)
}
job.Status = result.Status
if job.EndTime == 0 && job.Status == string(models.ModelArtsStopped) {
job.EndTime = timeutil.TimeStampNow()
job.Duration = job.ComputeDuration()
}
err = models.UpdateJob(job)
if err != nil {
log.Error("UpdateJob failed:", err)


+ 4
- 0
routers/repo/cloudbrain.go View File

@@ -1011,6 +1011,10 @@ func SyncCloudbrainStatus() {
if task.StartTime == 0 {
task.StartTime = timeutil.TimeStamp(result.Lease.CreateTime / 1000)
}
if task.EndTime == 0 && task.Status == string(models.ModelArtsStopped) {
task.EndTime = timeutil.TimeStampNow()
task.Duration = task.ComputeDuration()
}
err = models.UpdateJob(task)
if err != nil {
log.Error("UpdateJob(%s) failed:%v", task.JobName, err)


Loading…
Cancel
Save