Add retry on get user

form3tech-oss · Sep 30, 2019 · 690ec1e · 690ec1e
1 parent 4da9dd4
commit 690ec1e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/auth0/auth0_client.go b/auth0/auth0_client.go
@@ -4,6 +4,8 @@ import (
 	"bytes"
 	"encoding/json"
 	"fmt"
+	"net/http"
+	"time"
 
 	"github.com/parnurzeal/gorequest"
 )
@@ -113,7 +115,11 @@ func (config *Config) getAuthenticationHeader() string {
 // User
 func (authClient *AuthClient) GetUserById(id string) (*User, error) {
 
-	resp, body, errs := gorequest.New().Get(authClient.config.apiUri+"users/"+id).Set("Authorization", authClient.config.getAuthenticationHeader()).End()
+	resp, body, errs := gorequest.New().
+		Get(authClient.config.apiUri+"users/"+id).
+		Set("Authorization", authClient.config.getAuthenticationHeader()).
+		Retry(2, 60*time.Second, http.StatusTooManyRequests).
+		End()
 
 	if resp.StatusCode >= 400 && resp.StatusCode != 404 {
 		return nil, fmt.Errorf("bad status code (%d): %s", resp.StatusCode, body)