Add HTTP retry for 429 (#55)

hi-artem · web-flow · commit 7ab66710d6e4 · 2022-06-29T15:30:26.000-05:00
* (feat) add retry for 429s
diff --git a/internal/api/client.go b/internal/api/client.go
@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"net/url"
 	"path"
+	"time"
 )
 
 type APIClientConfig struct {
@@ -65,6 +66,13 @@ func (c *Client) Request(method, endpoint string, query, data, response interfac
 	}
 	defer res.Body.Close()
 
+	// Retry in case backend responds with HTTP 429
+	// sleep for 3 seconds before retry
+	if res.StatusCode == 429 {
+		time.Sleep(3 * time.Second)
+		return c.Request(method, endpoint, query, data, &response)
+	}
+
 	if res.StatusCode != http.StatusOK {
 		return fmt.Errorf("Non-OK status: %d", res.StatusCode)
 	}

Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,7 @@ import (`
`9`	`9`	`"net/http"`
`10`	`10`	`"net/url"`
`11`	`11`	`"path"`
	`12`	`+ "time"`
`12`	`13`	`)`
`13`	`14`
`14`	`15`	`type APIClientConfig struct {`
`@@ -65,6 +66,13 @@ func (c *Client) Request(method, endpoint string, query, data, response interfac`
`65`	`66`	`}`
`66`	`67`	`defer res.Body.Close()`
`67`	`68`
	`69`	`+ // Retry in case backend responds with HTTP 429`
	`70`	`+ // sleep for 3 seconds before retry`
	`71`	`+ if res.StatusCode == 429 {`
	`72`	`+ time.Sleep(3 * time.Second)`
	`73`	`+ return c.Request(method, endpoint, query, data, &response)`
	`74`	`+ }`
	`75`	`+`
`68`	`76`	`if res.StatusCode != http.StatusOK {`
`69`	`77`	`return fmt.Errorf("Non-OK status: %d", res.StatusCode)`
`70`	`78`	`}`