Initial commit

middleware/throttle.go

@@ -0,0 +1,25 @@
+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"
+	"log"
+)
+
+func Throttle(limit int) gin.HandlerFunc {
+	// Define a limit rate to 3 requests per minute.
+	rate, err := limiter.NewRateFromFormatted("5-M")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	store := memory.NewStore()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// Create a new middleware with the limiter instance.
+	return middlewareGin.NewMiddleware(limiter.New(store, rate))
+}