ci: regenerated with OpenAPI Doc 0.0.3, Speakeasy CLI 1.129.1

internal/sdk/pkg/utils/retries.go

@@ -0,0 +1,117 @@
+// Code generated by Speakeasy (https://speakeasyapi.dev). DO NOT EDIT.
+
+package utils
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"net/http"
+	"net/url"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/cenkalti/backoff/v4"
+)
+
+var errRequestFailed = errors.New("request failed")
+
+type BackoffStrategy struct {
+	InitialInterval int
+	MaxInterval     int
+	Exponent        float64
+	MaxElapsedTime  int
+}
+
+type RetryConfig struct {
+	Strategy              string
+	Backoff               *BackoffStrategy
+	RetryConnectionErrors bool
+}
+
+type Retries struct {
+	Config      *RetryConfig
+	StatusCodes []string
+}
+
+func Retry(ctx context.Context, r Retries, action func() (*http.Response, error)) (*http.Response, error) {
+	switch r.Config.Strategy {
+	case "backoff":
+		if r.Config.Backoff == nil {
+			return action()
+		}
+
+		config := backoff.NewExponentialBackOff()
+		config.InitialInterval = time.Duration(r.Config.Backoff.InitialInterval) * time.Millisecond
+		config.MaxInterval = time.Duration(r.Config.Backoff.MaxInterval) * time.Millisecond
+		config.Multiplier = r.Config.Backoff.Exponent
+		config.MaxElapsedTime = time.Duration(r.Config.Backoff.MaxElapsedTime) * time.Millisecond
+		config.Reset()
+
+		var resp *http.Response
+
+		err := backoff.Retry(func() error {
+			if resp != nil {
+				resp.Body.Close()
+			}
+
+			select {
+			case <-ctx.Done():
+				return backoff.Permanent(ctx.Err())
+			default:
+			}
+
+			res, err := action()
+			if err != nil {
+				urlError := new(url.Error)
+				if errors.As(err, &urlError) {
+					if (urlError.Temporary() || urlError.Timeout()) && r.Config.RetryConnectionErrors {
+						return err
+					}
+				}
+
+				return backoff.Permanent(err)
+			}
+			resp = res
+			if res == nil {
+				return fmt.Errorf("no response")
+			}
+
+			for _, code := range r.StatusCodes {
+				if strings.Contains(strings.ToUpper(code), "X") {
+					codeRange, err := strconv.Atoi(code[:1])
+					if err != nil {
+						continue
+					}
+
+					s := res.StatusCode / 100
+
+					if s >= codeRange && s < codeRange+1 {
+						return errRequestFailed
+					}
+				} else {
+					parsedCode, err := strconv.Atoi(code)
+					if err != nil {
+						continue
+					}
+
+					if res.StatusCode == parsedCode {
+						return errRequestFailed
+					}
+				}
+			}
+
+			resp = res
+
+			return nil
+		}, config)
+		if err != nil && !errors.Is(err, errRequestFailed) {
+			return nil, err
+		}
+
+		return resp, nil
+	default:
+		return action()
+	}
+}