golang-base-project/middleware/throttle.go

package middleware

import (
	"github.com/gin-gonic/gin"
	"github.com/ulule/limiter/v3"
	middlewareGin "github.com/ulule/limiter/v3/drivers/middleware/gin"
	"github.com/ulule/limiter/v3/drivers/store/memory"
	"time"
)

// Throttle middleware takes a limit per minute and blocks any additional requests that go over this limit
func Throttle(limit int) gin.HandlerFunc {
	store := memory.NewStore()

	// Create a new middleware with the limiter instance.
	return middlewareGin.NewMiddleware(limiter.New(store, limiter.Rate{
		Period: time.Minute,
		Limit:  int64(limit),
	}))
}