Add a rebuild recovery tool (#176)

Add a tool to run rebuild recoveries
google · Feb 19, 2025 · 5ef7529 · 5ef7529
1 parent fcd6b22
commit 5ef7529
Show file tree

Hide file tree

Showing 2 changed files with 470 additions and 0 deletions.
diff --git a/internal/httpx/http.go b/internal/httpx/http.go
@@ -9,6 +9,7 @@ import (
 	"bytes"
 	"errors"
 	"net/http"
+	"time"
 
 	"github.com/google/oss-rebuild/internal/cache"
 )
@@ -72,3 +73,15 @@ func (cc *CachedClient) Do(req *http.Request) (*http.Response, error) {
 }
 
 var _ BasicClient = &CachedClient{}
+
+type RateLimitedClient struct {
+	BasicClient
+	Ticker *time.Ticker
+}
+
+func (c *RateLimitedClient) Do(req *http.Request) (*http.Response, error) {
+	<-c.Ticker.C // Wait for next tick
+	return c.BasicClient.Do(req)
+}
+
+var _ BasicClient = &RateLimitedClient{}