Browse Source

get job status

tags/v0.1.8
yuyuanshifu 4 years ago
parent
commit
60f1fc41b4
1 changed files with 8 additions and 1 deletions
  1. +8
    -1
      routers/api/v1/repo/modelarts.go

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

@@ -8,6 +8,7 @@ package repo
import (
"code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/context"
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/modelarts"
"net/http"
)
@@ -19,7 +20,7 @@ func GetModelArtsTask(ctx *context.APIContext) {

jobID := ctx.Params(":jobid")
repoID := ctx.Repo.Repository.ID
_, err = models.GetRepoCloudBrainByJobID(repoID, jobID)
job, err := models.GetRepoCloudBrainByJobID(repoID, jobID)
if err != nil {
ctx.NotFound(err)
return
@@ -30,6 +31,12 @@ func GetModelArtsTask(ctx *context.APIContext) {
return
}

job.Status = result.Status
err = models.UpdateJob(job)
if err != nil {
log.Error("UpdateJob failed:", err)
}

ctx.JSON(http.StatusOK, map[string]interface{}{
"JobID": jobID,
"JobStatus": result.Status,


Loading…
Cancel
Save