chore: prevent too many requests

lucassabreu · Mar 29, 2024 · b1ae40b · b1ae40b
1 parent c4e4ce1
commit b1ae40b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/api/client.go b/api/client.go
@@ -445,7 +445,18 @@ func (c *client) GetUsersHydratedTimeEntries(p GetUserTimeEntriesParam) ([]dto.T
 		return timeEntries, err
 	}
 
-	user, err := c.GetUser(GetUser{p.Workspace, p.UserID})
+	var user dto.User
+	tries := 0
+	for tries < 5 {
+		tries++
+		user, err = c.GetUser(GetUser{p.Workspace, p.UserID})
+		if err == nil || !errors.Is(err, ErrorTooManyRequests) {
+			break
+		}
+
+		time.Sleep(time.Duration(5))
+	}
+
 	if err != nil {
 		return timeEntries, err
 	}

diff --git a/api/httpClient.go b/api/httpClient.go
@@ -22,6 +22,9 @@ var ErrorNotFound = dto.Error{Message: "Nothing was found", Code: 404}
 // ErrorForbidden Forbidden
 var ErrorForbidden = dto.Error{Message: "Forbidden", Code: 403}
 
+// ErrorTooManyRequests Too Many Requests
+var ErrorTooManyRequests = dto.Error{Message: "Too Many Requests", Code: 429}
+
 type transport struct {
 	apiKey string
 	next   http.RoundTripper
@@ -112,6 +115,10 @@ func (c *client) Do(
 			apiErr = ErrorForbidden
 		}
 
+		if r.StatusCode == 429 && apiErr.Message == "" {
+			apiErr = ErrorTooManyRequests
+		}
+
 		if apiErr.Message == "" {
 			apiErr.Message = "No response"
 		}