add retry for 4xx and 5xx requests

Signed-off-by: Markus Blaschke <[email protected]>
webdevops · Dec 10, 2022 · 8f59509 · 8f59509
1 parent ba65972
commit 8f59509
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/azuremetadata/metadata.go b/azuremetadata/metadata.go
@@ -29,9 +29,15 @@ func (m *AzureMetadata) Init() {
 	m.restClient.SetHeader("Metadata", "true")
 	m.restClient.SetHeader("Accept", "application/json")
 	m.restClient.SetTimeout(*m.Timeout)
+
+	// retry
 	m.restClient.SetRetryCount(5)
 	m.restClient.SetRetryMaxWaitTime(30 * time.Second)
 	m.restClient.SetRetryWaitTime(5 * time.Second)
+	m.restClient.AddRetryCondition(resty.RetryConditionFunc(func(r *resty.Response, err error) bool {
+		// retry for 4xx and 5xx
+		return r.StatusCode() >= http.StatusBadRequest
+	}))
 }
 
 func (m *AzureMetadata) FetchScheduledEvents() (*AzureScheduledEventResponse, error) {