Browse Source

fix-1556

tags/v1.22.2.2^2
liuzx 3 years ago
parent
commit
0cf9c5165f
3 changed files with 17 additions and 9 deletions
  1. +4
    -3
      routers/api/v1/api.go
  2. +12
    -5
      routers/api/v1/repo/cloudbrain.go
  3. +1
    -1
      templates/repo/debugjob/index.tmpl

+ 4
- 3
routers/api/v1/api.go View File

@@ -878,9 +878,10 @@ func RegisterRoutes(m *macaron.Macaron) {
}, reqAdmin())
}, reqAnyRepoReader())
m.Group("/cloudbrain", func() {
// m.Get("/:jobid", repo.GetCloudbrainTask)
m.Get("/:jobname", repo.GetCloudbrainTask)
m.Get("/:jobname/log", repo.CloudbrainGetLog)
m.Get("/:jobid", repo.GetCloudbrainTask)
m.Get("/:jobid/log", repo.CloudbrainGetLog)
// m.Get("/:jobname", repo.GetCloudbrainTask)
// m.Get("/:jobname/log", repo.CloudbrainGetLog)
}, reqRepoReader(models.UnitTypeCloudBrain))
m.Group("/modelarts", func() {
m.Group("/notebook", func() {


+ 12
- 5
routers/api/v1/repo/cloudbrain.go View File

@@ -49,8 +49,11 @@ func GetCloudbrainTask(ctx *context.APIContext) {
err error
)

jobName := ctx.Params(":jobname")
job, err := models.GetCloudbrainByName(jobName)
// jobName := ctx.Params(":jobname")
// job, err := models.GetCloudbrainByName(jobName)
jobID := ctx.Params(":jobid")
repoID := ctx.Repo.Repository.ID
job, err := models.GetRepoCloudBrainByJobID(repoID, jobID)
if err != nil {
ctx.Data["error"] = err.Error()
}
@@ -83,6 +86,7 @@ func GetCloudbrainTask(ctx *context.APIContext) {
}

ctx.JSON(http.StatusOK, map[string]interface{}{
"JobID": result.Config.JobID,
"JobName": result.Config.JobName,
"JobStatus": result.JobStatus.State,
"SubState": result.JobStatus.SubState,
@@ -93,8 +97,11 @@ func GetCloudbrainTask(ctx *context.APIContext) {
}

func CloudbrainGetLog(ctx *context.Context) {
jobName := ctx.Params(":jobname")
job, err := models.GetCloudbrainByName(jobName)
// jobName := ctx.Params(":jobname")
// job, err := models.GetCloudbrainByName(jobName)
jobID := ctx.Params(":jobid")
repoID := ctx.Repo.Repository.ID
job, err := models.GetRepoCloudBrainByJobID(repoID, jobID)
if err != nil {
log.Error("GetCloudbrainByJobID failed: %v", err, ctx.Data["MsgID"])
ctx.ServerError(err.Error(), err)
@@ -141,7 +148,7 @@ func CloudbrainGetLog(ctx *context.Context) {
}

ctx.JSON(http.StatusOK, map[string]interface{}{
"JobName": jobName,
"JobID": jobID,
"Content": content,
})



+ 1
- 1
templates/repo/debugjob/index.tmpl View File

@@ -293,7 +293,7 @@
</div>
<div class="two wide column text center">
<!--任务状态 -->
<span class="job-status" id="{{.JobID}}" data-repopath="{{$.RepoRelPath}}{{if eq .ComputeResource "CPU/GPU"}}/cloudbrain{{else}}/modelarts/notebook{{end}}" data-jobid="{{if eq .ComputeResource "CPU/GPU"}}{{.JobName}}{{else}}{{.JobID}}{{end}}" data-resource="{{.ComputeResource}}">
<span class="job-status" id="{{.JobID}}" data-repopath="{{$.RepoRelPath}}{{if eq .ComputeResource "CPU/GPU"}}/cloudbrain{{else}}/modelarts/notebook{{end}}" data-jobid="{{.JobID}}" data-resource="{{.ComputeResource}}">
<span><i id="{{.JobID}}-icon" style="vertical-align: middle;" class="{{.Status}}"></i><span id="{{.JobID}}-text" style="margin-left: 0.4em;font-size: 12px;">{{.Status}}</span></span>
</span>
</div>


Loading…
Cancel
Save