Merge pull request #7 from speakeasy-api/add_retry_after_and_paginatino_url

feat: add retry after and pagination url

Author: ryan-timothy-albert

cmd/server/main.go

@@ -33,6 +33,7 @@ func main() {
 	r.HandleFunc("/pagination/limitoffset/page", pagination.HandleLimitOffsetPage).Methods(http.MethodGet, http.MethodPut)
 	r.HandleFunc("/pagination/limitoffset/offset", pagination.HandleLimitOffsetOffset).Methods(http.MethodGet, http.MethodPut)
 	r.HandleFunc("/pagination/cursor", pagination.HandleCursor).Methods(http.MethodGet, http.MethodPut)
+	r.HandleFunc("/pagination/url", pagination.HandleURL).Methods(http.MethodGet)
 	r.HandleFunc("/retries", retries.HandleRetries).Methods(http.MethodGet, http.MethodPost)
 	r.HandleFunc("/errors/{status_code}", errors.HandleErrors).Methods(http.MethodGet)
 	r.HandleFunc("/optional", acceptHeaders.HandleAcceptHeaderMultiplexing).Methods(http.MethodGet)

internal/pagination/service.go

@@ -2,6 +2,7 @@ package pagination
 
 import (
 	"encoding/json"
+	"fmt"
 	"math"
 	"net/http"
 	"strconv"
@@ -18,8 +19,9 @@ type CursorRequest struct {
 }
 
 type PaginationResponse struct {
-	NumPages    int   `json:"numPages"`
-	ResultArray []int `json:"resultArray"`
+	NumPages    int     `json:"numPages"`
+	ResultArray []int   `json:"resultArray"`
+	Next        *string `json:"next"`
 }
 
 const total = 20
@@ -116,6 +118,41 @@ func HandleCursor(w http.ResponseWriter, r *http.Request) {
 	}
 }
 
+func HandleURL(w http.ResponseWriter, r *http.Request) {
+	attemptsString := r.FormValue("attempts")
+	var attempts int
+	if attemptsString != "" {
+		var err error
+		attempts, err = strconv.Atoi(attemptsString)
+		if err != nil {
+			w.WriteHeader(http.StatusBadRequest)
+			_, _ = w.Write([]byte("attempts must be an integer"))
+			return
+		}
+	}
+
+	res := PaginationResponse{
+		NumPages:    0,
+		ResultArray: make([]int, 0),
+	}
+
+	if attempts > 0 {
+		baseURL := fmt.Sprintf("%s://%s%s", r.URL.Scheme, r.Host, r.URL.Path)
+		if r.URL.Scheme == "" { // Fallback if Scheme is not available
+			baseURL = fmt.Sprintf("http://%s%s", r.Host, r.URL.Path)
+		}
+
+		nextUrl := fmt.Sprintf("%s?attempts=%d", baseURL, attempts-1)
+		res.Next = &nextUrl
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	err := json.NewEncoder(w).Encode(res)
+	if err != nil {
+		w.WriteHeader(500)
+	}
+}
+
 func getValue(queryValue string, hasBody bool, paginationValue int) int {
 	if hasBody {
 		return paginationValue

internal/retries/service.go

@@ -15,6 +15,7 @@ type retriesResponse struct {
 func HandleRetries(w http.ResponseWriter, r *http.Request) {
 	requestID := r.URL.Query().Get("request-id")
 	numRetriesStr := r.URL.Query().Get("num-retries")
+	includeHeaderTimeout := r.URL.Query().Get("included-header-timeout")
 
 	numRetries := 3
 	if numRetriesStr != "" {
@@ -27,6 +28,17 @@ func HandleRetries(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
+	var headerTimeout int
+	if includeHeaderTimeout != "" {
+		var err error
+		headerTimeout, err = strconv.Atoi(includeHeaderTimeout)
+		if err != nil {
+			w.WriteHeader(http.StatusBadRequest)
+			_, _ = w.Write([]byte("included-header-timeout must be an integer"))
+			return
+		}
+	}
+
 	if requestID == "" {
 		w.WriteHeader(http.StatusBadRequest)
 		_, _ = w.Write([]byte("request-id is required"))
@@ -40,6 +52,9 @@ func HandleRetries(w http.ResponseWriter, r *http.Request) {
 	callCounts[requestID]++
 
 	if callCounts[requestID] < numRetries {
+		if headerTimeout > 0 {
+			w.Header().Set("Retry-After", strconv.Itoa(headerTimeout))
+		}
 		w.WriteHeader(http.StatusServiceUnavailable)
 		_, _ = w.Write([]byte("request failed please retry"))
 		return