feat: add missing htt StatusCode and make it settable

Author: Marcel S. Henselin

core/wait/wait.go

@@ -10,7 +10,7 @@ import (
 	"github.com/stackitcloud/stackit-sdk-go/core/utils"
 )
 
-var RetryHttpErrorStatusCodes = []int{http.StatusBadGateway, http.StatusGatewayTimeout}
+var RetryHttpErrorStatusCodes = []int{http.StatusBadGateway, http.StatusGatewayTimeout, http.StatusServiceUnavailable}
 
 // AsyncActionCheck reports whether a specific async action has finished.
 //   - waitFinished == true if the async action is finished, false otherwise.
@@ -20,25 +20,33 @@ type AsyncActionCheck[T any] func() (waitFinished bool, response *T, err error)
 
 // AsyncActionHandler handles waiting for a specific async action to be finished.
 type AsyncActionHandler[T any] struct {
-	checkFn                  AsyncActionCheck[T]
-	sleepBeforeWait          time.Duration
-	throttle                 time.Duration
-	timeout                  time.Duration
-	tempErrRetryLimit        int
-	IntermediateStateReached bool
+	checkFn                   AsyncActionCheck[T]
+	sleepBeforeWait           time.Duration
+	throttle                  time.Duration
+	timeout                   time.Duration
+	tempErrRetryLimit         int
+	IntermediateStateReached  bool
+	retryHttpErrorStatusCodes []int
 }
 
 // New initializes an AsyncActionHandler
 func New[T any](f AsyncActionCheck[T]) *AsyncActionHandler[T] {
 	return &AsyncActionHandler[T]{
-		checkFn:           f,
-		sleepBeforeWait:   0 * time.Second,
-		throttle:          5 * time.Second,
-		timeout:           30 * time.Minute,
-		tempErrRetryLimit: 5,
+		checkFn:                   f,
+		sleepBeforeWait:           0 * time.Second,
+		throttle:                  5 * time.Second,
+		timeout:                   30 * time.Minute,
+		tempErrRetryLimit:         5,
+		retryHttpErrorStatusCodes: RetryHttpErrorStatusCodes,
 	}
 }
 
+// SetRetryHttpErrorStatusCodes sets the retry codes to use for retry.
+func (h *AsyncActionHandler[T]) SetRetryHttpErrorStatusCodes(c []int) *AsyncActionHandler[T] {
+	h.retryHttpErrorStatusCodes = c
+	return h
+}
+
 // SetThrottle sets the time interval between each check of the async action.
 func (h *AsyncActionHandler[T]) SetThrottle(d time.Duration) *AsyncActionHandler[T] {
 	h.throttle = d
@@ -114,7 +122,7 @@ func (h *AsyncActionHandler[T]) handleError(retryTempErrorCounter int, err error
 		return retryTempErrorCounter, fmt.Errorf("found non-GenericOpenApiError: %w", err)
 	}
 	// Some APIs may return temporary errors and the request should be retried
-	if !utils.Contains(RetryHttpErrorStatusCodes, oapiErr.StatusCode) {
+	if !utils.Contains(h.retryHttpErrorStatusCodes, oapiErr.StatusCode) {
 		return retryTempErrorCounter, err
 	}
 	retryTempErrorCounter++

core/wait/wait_test.go

@@ -22,11 +22,12 @@ func TestNew(t *testing.T) {
 	checkFn := func() (waitFinished bool, res *interface{}, err error) { return true, nil, nil }
 	got := New(checkFn)
 	want := &AsyncActionHandler[interface{}]{
-		checkFn:           checkFn,
-		sleepBeforeWait:   0 * time.Second,
-		throttle:          5 * time.Second,
-		timeout:           30 * time.Minute,
-		tempErrRetryLimit: 5,
+		checkFn:                   checkFn,
+		sleepBeforeWait:           0 * time.Second,
+		throttle:                  5 * time.Second,
+		timeout:                   30 * time.Minute,
+		tempErrRetryLimit:         5,
+		retryHttpErrorStatusCodes: RetryHttpErrorStatusCodes,
 	}
 
 	diff := cmp.Diff(got, want, cmpOpts...)
@@ -159,7 +160,41 @@ func TestSetTempErrRetryLimit(t *testing.T) {
 			got := New(checkFn)
 			got.SetTempErrRetryLimit(tt.tempErrRetryLimit)
 
-			diff := cmp.Diff(got, want, cmpOpts...)
+			diff := cmp.Diff(want, got, cmpOpts...)
+			if diff != "" {
+				t.Errorf("Data does not match: %s", diff)
+			}
+		})
+	}
+}
+
+func TestSetRetryHttpErrorStatusCodes(t *testing.T) {
+	checkFn := func() (waitFinished bool, res *interface{}, err error) { return true, nil, nil }
+
+	for _, tt := range []struct {
+		desc           string
+		setRetryCodes  []int
+		wantRetryCodes []int
+	}{
+		{
+			"default",
+			[]int{},
+			[]int{http.StatusBadGateway, http.StatusGatewayTimeout, http.StatusServiceUnavailable},
+		},
+		{
+			"base_3",
+			[]int{http.StatusTooManyRequests, http.StatusInternalServerError},
+			[]int{http.StatusTooManyRequests, http.StatusInternalServerError},
+		},
+	} {
+		t.Run(tt.desc, func(t *testing.T) {
+			want := New(checkFn)
+			want.retryHttpErrorStatusCodes = tt.wantRetryCodes
+			got := New(checkFn)
+			if len(tt.setRetryCodes) != 0 {
+				got.SetRetryHttpErrorStatusCodes(tt.setRetryCodes)
+			}
+			diff := cmp.Diff(want, got, cmpOpts...)
 			if diff != "" {
 				t.Errorf("Data does not match: %s", diff)
 			}