add error handling

stacklok · Feb 19, 2024 · 22af15e · 22af15e
1 parent eb5ba88
commit 22af15e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/feeds/maven/maven.go b/pkg/feeds/maven/maven.go
@@ -72,6 +72,17 @@ func (feed Feed) fetchPackages(page int) ([]Package, error) {
 	}
 	defer resp.Body.Close()
 
+	// Handle rate limiting (HTTP status code 429).
+	if resp.StatusCode == http.StatusTooManyRequests {
+		time.Sleep(5 * time.Second)
+		return feed.fetchPackages(page) // Retry the request
+	}
+
+	// Handle other HTTP status codes
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("unexpected HTTP status code: %d", resp.StatusCode)
+	}
+
 	// Decode response.
 	var response Response
 	err = json.NewDecoder(resp.Body).Decode(&response)