Add rate limits

Signed-off-by: Dmitry Kisler <[email protected]>
opentofu · Nov 28, 2023 · 1d92861 · 1d92861
1 parent b335ee4
commit 1d92861
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/internal/github/client.go b/src/internal/github/client.go
@@ -6,6 +6,7 @@ import (
 	"log/slog"
 	"net/http"
 	"os"
+	"time"
 
 	"github.com/hashicorp/go-cleanhttp"
 	"github.com/hashicorp/go-retryablehttp"
@@ -51,10 +52,10 @@ func NewClient(ctx context.Context, log *slog.Logger, token string) Client {
 		httpClient: retryClient,
 		ghClient:   githubv4.NewClient(httpClient),
 
-		cliThrottle:   NewDummyThrottle(),
-		apiThrottle:   NewDummyThrottle(),
-		assetThrottle: NewDummyThrottle(),
-		rssThrottle:   NewDummyThrottle(),
+		cliThrottle:   NewThrottle(ctx, time.Second/30, 30),
+		apiThrottle:   NewThrottle(ctx, time.Second, 3),
+		assetThrottle: NewThrottle(ctx, time.Second/30, 30),
+		rssThrottle:   NewThrottle(ctx, time.Second/30, 30),
 	}
 	/* TODO
 	retryClient := retryablehttp.NewClient()