rate limit

Author: mstgnz

.env.example

@@ -20,7 +20,9 @@ AWS_SESSION_TOKEN=
 AWS_REGION=eu-central-1
 AWS_BUCKET=
 
+RATE_LIMIT=100
 UPLOAD_RATE_LIMIT=50
+RATE_LIMIT_DURATION=1
 
 # File Validation
 VALIDATE_FILE=true

pkg/middleware/limiter.go

@@ -5,12 +5,13 @@ import (
 
 	"github.com/gofiber/fiber/v2"
 	"github.com/gofiber/fiber/v2/middleware/limiter"
+	"github.com/mstgnz/cdn/pkg/config"
 )
 
 // RateLimiterConfig default rate limiter configuration
 var RateLimiterConfig = limiter.Config{
-	Max:        100,             // Maximum number of requests per duration
-	Expiration: 1 * time.Minute, // Duration for rate limit window
+	Max:        config.GetEnvAsIntOrDefault("RATE_LIMIT", 100),
+	Expiration: time.Duration(config.GetEnvAsIntOrDefault("RATE_LIMIT_DURATION", 1)) * time.Minute, // Duration for rate limit window
 	KeyGenerator: func(c *fiber.Ctx) string {
 		return c.IP() // Use client IP as key
 	},

pkg/middleware/security.go

@@ -9,6 +9,7 @@ import (
 	"github.com/gofiber/fiber/v2"
 	"github.com/gofiber/fiber/v2/middleware/cors"
 	"github.com/gofiber/fiber/v2/middleware/limiter"
+	"github.com/mstgnz/cdn/pkg/config"
 )
 
 // SecurityConfig represents security middleware configuration
@@ -220,5 +221,5 @@ func NewAdvancedRateLimiter(max int, duration time.Duration) fiber.Handler {
 
 // DefaultAdvancedRateLimiter returns a default rate limiter middleware (100 requests per minute)
 func DefaultAdvancedRateLimiter() fiber.Handler {
-	return NewAdvancedRateLimiter(100, time.Minute)
+	return NewAdvancedRateLimiter(config.GetEnvAsIntOrDefault("RATE_LIMIT", 100), time.Minute)
 }