gin中使用限流中间件

在Gin框架中，你可以使用第三方库如go-rate来实现限流功能。以下是一个简单的示例，展示了如何在Gin应用中使用限流中间件：

首先，你需要安装go-rate库：

go get -u github.com/go-rate/rate

然后，你可以在Gin应用中这样使用限流中间件：

package main
 
import (
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/go-rate/rate"
    "time"
)
 
func main() {
    r := gin.Default()
 
    // 创建一个每秒限制2个请求的限流器
    limiter := rate.NewLimiter(rate.Every(time.Second), 2)
 
    // 使用Gin的Group来应用限流中间件
    group := r.Group("/")
    group.Use(rateLimit(limiter))
 
    group.GET("/", func(c *gin.Context) {
        c.String(200, "Hello, Gin!")
    })
 
    r.Run()
}
 
// rateLimit 是一个Gin中间件，用于应用限流逻辑
func rateLimit(limiter *rate.Limiter) gin.HandlerFunc {
    return func(c *gin.Context) {
        if err := limiter.Wait(c.Request.Context()); err != nil {
            c.AbortWithStatusJSON(429, gin.H{"error": "Too Many Requests"})
            return
        }
 
        c.Next()
    }
}

在这个示例中，我们创建了一个每秒限制2个请求的限流器，并将其作为中间件应用到了Gin的路由组上。如果请求超过了限制，中间件会返回状态码429（Too Many Requests）。