|
|
@@ -25,6 +25,7 @@ var ( |
|
|
|
|
|
|
|
const ( |
|
|
|
JobHasBeenStopped = "S410" |
|
|
|
errInvalidToken = "S401" |
|
|
|
Public = "public" |
|
|
|
Custom = "custom" |
|
|
|
LogPageSize = 500 |
|
|
@@ -59,7 +60,7 @@ func loginCloudbrain() error { |
|
|
|
|
|
|
|
res, err := client.R(). |
|
|
|
SetHeader("Content-Type", "application/json"). |
|
|
|
SetBody(map[string]interface{}{"username": username, "password": password, "expiration": "604800"}). |
|
|
|
SetBody(map[string]interface{}{"username": username, "password": password, "expiration": conf.Expiration}). |
|
|
|
SetResult(&loginResult). |
|
|
|
Post(HOST + "/rest-server/api/v1/token") |
|
|
|
if err != nil { |
|
|
@@ -122,16 +123,12 @@ sendjob: |
|
|
|
Post(HOST + "/rest-server/api/v1/jobs/") |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
if res != nil { |
|
|
|
var response models.CloudBrainResult |
|
|
|
json.Unmarshal(res.Body(), &response) |
|
|
|
log.Error("code(%s), msg(%s)", response.Code, response.Msg) |
|
|
|
return nil, fmt.Errorf(response.Msg) |
|
|
|
} |
|
|
|
return nil, fmt.Errorf("resty create job: %s", err) |
|
|
|
} |
|
|
|
|
|
|
|
if jobResult.Code == "S401" && retry < 1 { |
|
|
|
var response models.CloudBrainResult |
|
|
|
json.Unmarshal(res.Body(), &response) |
|
|
|
if response.Code == errInvalidToken && retry < 1 { |
|
|
|
retry++ |
|
|
|
_ = loginCloudbrain() |
|
|
|
goto sendjob |
|
|
@@ -163,7 +160,9 @@ sendjob: |
|
|
|
return nil, fmt.Errorf("resty GetJob: %v", err) |
|
|
|
} |
|
|
|
|
|
|
|
if getJobResult.Code == "S401" && retry < 1 { |
|
|
|
var response models.CloudBrainResult |
|
|
|
json.Unmarshal(res.Body(), &response) |
|
|
|
if response.Code == errInvalidToken && retry < 1 { |
|
|
|
retry++ |
|
|
|
_ = loginCloudbrain() |
|
|
|
goto sendjob |
|
|
@@ -196,13 +195,8 @@ sendjob: |
|
|
|
} |
|
|
|
|
|
|
|
var response models.CloudBrainResult |
|
|
|
err = json.Unmarshal(res.Body(), &response) |
|
|
|
if err != nil { |
|
|
|
log.Error("json.Unmarshal failed: %s", err.Error()) |
|
|
|
return &getImagesResult, fmt.Errorf("json.Unmarshal failed: %s", err.Error()) |
|
|
|
} |
|
|
|
|
|
|
|
if response.Code == "S401" && retry < 1 { |
|
|
|
json.Unmarshal(res.Body(), &response) |
|
|
|
if response.Code == errInvalidToken && retry < 1 { |
|
|
|
retry++ |
|
|
|
_ = loginCloudbrain() |
|
|
|
goto sendjob |
|
|
@@ -290,7 +284,9 @@ sendjob: |
|
|
|
return fmt.Errorf("resty CommitImage: %v", err) |
|
|
|
} |
|
|
|
|
|
|
|
if result.Code == "S401" && retry < 1 { |
|
|
|
var response models.CloudBrainResult |
|
|
|
json.Unmarshal(res.Body(), &response) |
|
|
|
if response.Code == errInvalidToken && retry < 1 { |
|
|
|
retry++ |
|
|
|
_ = loginCloudbrain() |
|
|
|
goto sendjob |
|
|
@@ -442,7 +438,9 @@ sendjob: |
|
|
|
return fmt.Errorf("resty StopJob: %v", err) |
|
|
|
} |
|
|
|
|
|
|
|
if result.Code == "S401" && retry < 1 { |
|
|
|
var response models.CloudBrainResult |
|
|
|
json.Unmarshal(res.Body(), &response) |
|
|
|
if response.Code == errInvalidToken && retry < 1 { |
|
|
|
retry++ |
|
|
|
_ = loginCloudbrain() |
|
|
|
goto sendjob |
|
|
@@ -522,7 +520,7 @@ func GetJobAllLog(scrollID string) (*models.GetJobLogResult, error) { |
|
|
|
return &result, nil |
|
|
|
} |
|
|
|
|
|
|
|
func DeleteJobLogToken(scrollID string) (error) { |
|
|
|
func DeleteJobLogToken(scrollID string) error { |
|
|
|
checkSetting() |
|
|
|
client := getRestyClient() |
|
|
|
var result models.DeleteJobLogTokenResult |
|
|
|