(BIDS-2872) copy ratelimit over from v1

gobitfly · Jul 30, 2024 · 012c097 · 012c097
1 parent cf035e3
commit 012c097
Show file tree

Hide file tree

Showing 4 changed files with 1,223 additions and 8 deletions.
diff --git a/backend/cmd/api/main.go b/backend/cmd/api/main.go
@@ -11,6 +11,7 @@ import (
 
 	"github.com/gobitfly/beaconchain/pkg/commons/log"
 	"github.com/gobitfly/beaconchain/pkg/commons/metrics"
+	"github.com/gobitfly/beaconchain/pkg/commons/ratelimit"
 	"github.com/gobitfly/beaconchain/pkg/commons/types"
 	"github.com/gobitfly/beaconchain/pkg/commons/utils"
 	"github.com/gobitfly/beaconchain/pkg/commons/version"
@@ -62,6 +63,9 @@ func main() {
 		}(cfg.Metrics.Address)
 	}
 
+	ratelimit.Init()
+	router.Use(ratelimit.HttpMiddleware)
+
 	srv := &http.Server{
 		Handler:      router,
 		Addr:         net.JoinHostPort(cfg.Frontend.Server.Host, cfg.Frontend.Server.Port),

diff --git a/backend/pkg/commons/metrics/metrics.go b/backend/pkg/commons/metrics/metrics.go
@@ -70,6 +70,10 @@ var (
 		Name: "state",
 		Help: "Gauge for various states",
 	}, []string{"state"})
+	Counter = promauto.NewCounterVec(prometheus.CounterOpts{
+		Name: "counter",
+		Help: "Generic counter of events with name in labels",
+	}, []string{"name"})
 )
 
 func init() {