Include rate limiter as a middleware in Gitaly

Add a rate limiting middleware into the interceptor chain for a Gitaly
server.

Changelog: added

1 files changed, 19 insertions, 3 deletions

diff --git a/cmd/gitaly/main.go b/cmd/gitaly/main.go
index 818f5ce2f..6fcda7527 100644
--- a/cmd/gitaly/main.go
+++ b/cmd/gitaly/main.go
@@ -213,10 +213,26 @@ func run(cfg config.Cfg) error {
 		return fmt.Errorf("disk cache walkers: %w", err)
 	}
 
-	limitHandler := limithandler.New(cfg, limithandler.LimitConcurrencyByRepo, limithandler.WithConcurrencyLimiters)
-	prometheus.MustRegister(limitHandler)
+	concurrencyLimitHandler := limithandler.New(
+		cfg,
+		limithandler.LimitConcurrencyByRepo,
+		limithandler.WithConcurrencyLimiters,
+	)
 
-	gitalyServerFactory := server.NewGitalyServerFactory(cfg, glog.Default(), registry, diskCache, []*limithandler.LimiterMiddleware{limitHandler})
+	rateLimitHandler := limithandler.New(
+		cfg,
+		limithandler.LimitConcurrencyByRepo,
+		limithandler.WithRateLimiters(ctx),
+	)
+	prometheus.MustRegister(concurrencyLimitHandler, rateLimitHandler)
+
+	gitalyServerFactory := server.NewGitalyServerFactory(
+		cfg,
+		glog.Default(),
+		registry,
+		diskCache,
+		[]*limithandler.LimiterMiddleware{concurrencyLimitHandler, rateLimitHandler},
+	)
 	defer gitalyServerFactory.Stop()
 
 	ling, err := linguist.New(cfg, gitCmdFactory)