yuyuanshifu
4 years ago
1 changed files with 8 additions and 1 deletions
Unified View
Diff Options
+ 8
- 1
routers/api/v1/repo/modelarts.go
View File
| @@ -8,6 +8,7 @@ package repo | |||||
| import ( | import ( | ||||
| "code.gitea.io/gitea/models" | "code.gitea.io/gitea/models" | ||||
| "code.gitea.io/gitea/modules/context" | "code.gitea.io/gitea/modules/context" | ||||
| "code.gitea.io/gitea/modules/log" | |||||
| "code.gitea.io/gitea/modules/modelarts" | "code.gitea.io/gitea/modules/modelarts" | ||||
| "net/http" | "net/http" | ||||
| ) | ) | ||||
| @@ -19,7 +20,7 @@ func GetModelArtsTask(ctx *context.APIContext) { | |||||
| jobID := ctx.Params(":jobid") | jobID := ctx.Params(":jobid") | ||||
| repoID := ctx.Repo.Repository.ID | repoID := ctx.Repo.Repository.ID | ||||
| _, err = models.GetRepoCloudBrainByJobID(repoID, jobID) | |||||
| job, err := models.GetRepoCloudBrainByJobID(repoID, jobID) | |||||
| if err != nil { | if err != nil { | ||||
| ctx.NotFound(err) | ctx.NotFound(err) | ||||
| return | return | ||||
| @@ -30,6 +31,12 @@ func GetModelArtsTask(ctx *context.APIContext) { | |||||
| return | return | ||||
| } | } | ||||
| job.Status = result.Status | |||||
| err = models.UpdateJob(job) | |||||
| if err != nil { | |||||
| log.Error("UpdateJob failed:", err) | |||||
| } | |||||
| ctx.JSON(http.StatusOK, map[string]interface{}{ | ctx.JSON(http.StatusOK, map[string]interface{}{ | ||||
| "JobID": jobID, | "JobID": jobID, | ||||
| "JobStatus": result.Status, | "JobStatus": result.Status, | ||||