add provider for fiber storage

prismelabs · May 5, 2024 · cdd715a · cdd715a
1 parent 5dc9c73
commit cdd715a
Show file tree

Hide file tree

Showing 9 changed files with 34 additions and 5 deletions.
diff --git a/cmd/server/full/wire.go b/cmd/server/full/wire.go
@@ -47,6 +47,7 @@ func Initialize(logger wired.BootstrapLogger) wired.App {
 		uaparser.ProvideService,
 		wired.ProvideApp,
 		wired.ProvideClickhouseConfig,
+		wired.ProvideFiberStorage,
 		wired.ProvideGrafanaConfig,
 		wired.ProvideLogger,
 		wired.ProvideMinimalFiber,

diff --git a/cmd/server/full/wire_gen.go b/cmd/server/full/wire_gen.go
diff --git a/cmd/server/ingestion/wire.go b/cmd/server/ingestion/wire.go
@@ -42,6 +42,7 @@ func Initialize(logger wired.BootstrapLogger) wired.App {
 		uaparser.ProvideService,
 		wired.ProvideApp,
 		wired.ProvideClickhouseConfig,
+		wired.ProvideFiberStorage,
 		wired.ProvideLogger,
 		wired.ProvideMinimalFiber,
 		wired.ProvideMinimalFiberConfig,

diff --git a/cmd/server/ingestion/wire_gen.go b/cmd/server/ingestion/wire_gen.go
diff --git a/flake.nix b/flake.nix
@@ -35,7 +35,7 @@
               default = pkgs.buildGoModule {
                 pname = "prisme";
                 version = "0.15.0";
-                vendorHash = "sha256-cjt7l+6k+TLuzp5vwXBI5WvrtqRH5IZept/MV5Zv0L8=";
+                vendorHash = "sha256-90NC32QsMnF3rwUmtB2x9aEqpNiRn+hIRuFfaFfv6w4=";
 
                 src = ./.;
                 # Skip go test.

diff --git a/go.mod b/go.mod
@@ -8,12 +8,15 @@ require (
 	github.com/ClickHouse/clickhouse-go/v2 v2.23.2
 	github.com/cespare/xxhash/v2 v2.3.0
 	github.com/gofiber/fiber/v2 v2.52.4
+	github.com/gofiber/storage v1.3.3
+	github.com/gofiber/storage/memory v1.3.4
 	github.com/golang-migrate/migrate/v4 v4.17.1
 	github.com/google/uuid v1.6.0
 	github.com/google/wire v0.6.0
 	github.com/negrel/ringo v0.5.0
 	github.com/oschwald/maxminddb-golang v1.12.0
 	github.com/prometheus/client_golang v1.19.0
+	github.com/prometheus/client_model v0.5.0
 	github.com/rs/zerolog v1.32.0
 	github.com/stretchr/testify v1.9.0
 	github.com/tidwall/gjson v1.17.1
@@ -31,7 +34,6 @@ require (
 	github.com/philhofer/fwd v1.1.2 // indirect
 	github.com/pierrec/lz4/v4 v4.1.21 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
-	github.com/prometheus/client_model v0.5.0 // indirect
 	github.com/prometheus/common v0.48.0 // indirect
 	github.com/prometheus/procfs v0.12.0 // indirect
 	github.com/rogpeppe/go-internal v1.11.0 // indirect

diff --git a/go.sum b/go.sum
@@ -44,6 +44,12 @@ github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre
 github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
 github.com/gofiber/fiber/v2 v2.52.4 h1:P+T+4iK7VaqUsq2PALYEfBBo6bJZ4q3FP8cZ84EggTM=
 github.com/gofiber/fiber/v2 v2.52.4/go.mod h1:KEOE+cXMhXG0zHc9d8+E38hoX+ZN7bhOtgeF2oT6jrQ=
+github.com/gofiber/storage v1.3.3 h1:XvcaYqEVcoXllhnkvH8sTIbuB1IbidIe3UlxzMvzBrE=
+github.com/gofiber/storage v1.3.3/go.mod h1:ZGviBmsCFkBSQInsD4C5e390gXwnH/MKcpKk//Fwv6o=
+github.com/gofiber/storage/memory v1.3.4 h1:VxTq8Vrdvk73VsfvtTgc3LjXClbBrHXanzEo+w0cpgo=
+github.com/gofiber/storage/memory v1.3.4/go.mod h1:pYsCUle/+4exGfsG7IlpmFYBVmNntP8OIDBvmABU8PE=
+github.com/gofiber/utils v1.0.1 h1:knct4cXwBipWQqFrOy1Pv6UcgPM+EXo9jDgc66V1Qio=
+github.com/gofiber/utils v1.0.1/go.mod h1:pacRFtghAE3UoknMOUiXh2Io/nLWSUHtQCi/3QASsOc=
 github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=
 github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
 github.com/golang-migrate/migrate/v4 v4.17.1 h1:4zQ6iqL6t6AiItphxJctQb3cFqWiSpMnX7wLTPnnYO4=

diff --git a/pkg/middlewares/events_rate_limiter.go b/pkg/middlewares/events_rate_limiter.go
@@ -5,18 +5,20 @@ import (
 
 	"github.com/gofiber/fiber/v2"
 	"github.com/gofiber/fiber/v2/middleware/limiter"
+	"github.com/gofiber/storage"
 	"github.com/prismelabs/analytics/pkg/config"
 )
 
 type EventsRateLimiter fiber.Handler
 
 // ProvideEventsRateLimiter is a wire provider for events endpoints rate limiter.
-func ProvideEventsRateLimiter(cfg config.Server) EventsRateLimiter {
+func ProvideEventsRateLimiter(cfg config.Server, storage storage.Storage) EventsRateLimiter {
 	return limiter.New(limiter.Config{
 		Max: 60,
 		KeyGenerator: func(c *fiber.Ctx) string {
 			return c.IP()
 		},
 		Expiration: time.Minute,
+		Storage:    storage,
 	})
 }
diff --git a/pkg/wired/fiber_storage.go b/pkg/wired/fiber_storage.go
@@ -0,0 +1,15 @@
+package wired
+
+import (
+	"time"
+
+	"github.com/gofiber/storage"
+	"github.com/gofiber/storage/memory"
+)
+
+// ProvideFiberStorage is a wire provider for fiber storage.
+func ProvideFiberStorage() storage.Storage {
+	return memory.New(memory.Config{
+		GCInterval: 10 * time.Second,
+	})
+}