From 372be23544073a207feab7ecce0632f3564dbd03 Mon Sep 17 00:00:00 2001 From: ychao_1983 Date: Thu, 25 Aug 2022 17:41:27 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=BB=BAgpu=E8=B0=83=E8=AF=95?= =?UTF-8?q?=E4=BB=BB=E5=8A=A1=E9=A1=B5=E9=9D=A2=E6=8E=92=E9=98=9F=E4=BB=BB?= =?UTF-8?q?=E5=8A=A1=E6=95=B0=E6=9C=89=E6=97=B6=E5=80=99=E4=B8=BA=E7=A9=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- modules/cloudbrain/resty.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/modules/cloudbrain/resty.go b/modules/cloudbrain/resty.go index d9db3bbb5..8387d481a 100755 --- a/modules/cloudbrain/resty.go +++ b/modules/cloudbrain/resty.go @@ -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()) }