Merge pull request #1096 from seokho-son/main

Trials for inspectResourcesOverview to prevent rateLimitExceeded from CSP
cloud-barista · May 7, 2022 · 3e56a9b · 3e56a9b
2 parents 59dc654 + d11af31
commit 3e56a9b
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/core/mcis/utility.go b/src/core/mcis/utility.go
@@ -617,6 +617,22 @@ func InspectResourcesOverview() (InspectResourceAllResult, error) {
 				common.CBLog.Error(err)
 				temp.SystemMessage = err.Error()
 			}
+			// retry if request rateLimitExceeded occurs. (GCP has ratelimiting)
+			rateLimitMessage := "limit"
+			maxTrials := 5
+			if strings.Contains(temp.SystemMessage, rateLimitMessage) {
+				for i := 0; i < maxTrials; i++ {
+					common.RandomSleep(40, 80)
+					inspectResult, err = InspectResources(k.ConfigName, common.StrVNet)
+					if err != nil {
+						common.CBLog.Error(err)
+						temp.SystemMessage = err.Error()
+					} else {
+						temp.SystemMessage = ""
+						break
+					}
+				}
+			}
 			temp.TumblebugOverview.VNet = inspectResult.ResourceOverview.OnTumblebug
 			temp.CspTotalOverview.VNet = inspectResult.ResourceOverview.OnCspTotal