Browse Source

Merge pull request '新建gpu调试任务页面排队任务数有时候为空' (#2768) from fix-2392 into V20220830

Reviewed-on: https://git.openi.org.cn/OpenI/aiforge/pulls/2768
Reviewed-by: lewis <747342561@qq.com>
fix-2781
lewis 2 years ago
parent
commit
c1b969ac54
1 changed files with 8 additions and 1 deletions
  1. +8
    -1
      modules/cloudbrain/resty.go

+ 8
- 1
modules/cloudbrain/resty.go View File

@@ -81,7 +81,8 @@ func GetQueuesDetail() (*map[string]int, error) {
var jobResult models.QueueDetailResult

var result = make(map[string]int, 0)

retry := 0
sendjob:
res, err := client.R().
SetHeader("Content-Type", "application/json").
SetAuthToken(TOKEN).
@@ -92,6 +93,12 @@ func GetQueuesDetail() (*map[string]int, error) {
return nil, fmt.Errorf("resty get queues detail failed: %s", err)
}

if jobResult.Code == errInvalidToken && retry < 1 {
retry++
_ = loginCloudbrain()
goto sendjob
}

if jobResult.Code != Success {
return nil, fmt.Errorf("jobResult err: %s", res.String())
}


Loading…
Cancel
Save