mirror of
https://github.com/nprasad2077/NBA_Go.git
synced 2026-09-22 14:05:13 +00:00
limiter changes and import changes
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
@@ -9,8 +10,17 @@ import (
|
||||
|
||||
func RateLimiter() fiber.Handler {
|
||||
return limiter.New(limiter.Config{
|
||||
Max: 30,
|
||||
Max: 20,
|
||||
Expiration: 1 * time.Minute,
|
||||
Next: func(c *fiber.Ctx) bool {
|
||||
// Skip rate limiting for internal services and infra endpoints
|
||||
ip := c.IP()
|
||||
if strings.HasPrefix(ip, "10.") || strings.HasPrefix(ip, "172.") || ip == "127.0.0.1" {
|
||||
return true
|
||||
}
|
||||
path := c.Path()
|
||||
return path == "/metrics" || strings.HasPrefix(path, "/swagger")
|
||||
},
|
||||
KeyGenerator: func(c *fiber.Ctx) string {
|
||||
if ip := c.Get("X-Real-IP"); ip != "" {
|
||||
return ip
|
||||
|
||||
Reference in New Issue
Block a user