mirror of https://github.com/JustKato/FreePad.git
35 lines
736 B
Go
35 lines
736 B
Go
|
package controllers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/JustKato/FreePad/lib/helper"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/ulule/limiter/v3"
|
||
|
|
"github.com/ulule/limiter/v3/drivers/store/memory"
|
||
|
|
|
||
|
|
mgin "github.com/ulule/limiter/v3/drivers/middleware/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Apply the rate limiter to the gin Engine
|
||
|
|
func DoRateLimit(router *gin.Engine) {
|
||
|
|
|
||
|
|
// Initialize the rate limiter
|
||
|
|
rate := limiter.Rate{
|
||
|
|
Period: 5 * time.Minute,
|
||
|
|
Limit: int64(helper.GetApiBanLimit()),
|
||
|
|
}
|
||
|
|
|
||
|
|
// Initialize the memory storage
|
||
|
|
store := memory.NewStore()
|
||
|
|
|
||
|
|
// Initialize the limiter instance
|
||
|
|
instance := limiter.New(store, rate)
|
||
|
|
|
||
|
|
// Create the gin middleware
|
||
|
|
middleWare := mgin.NewMiddleware(instance)
|
||
|
|
|
||
|
|
// use the middleware in gin
|
||
|
|
router.Use(middleWare)
|
||
|
|
}
|