防止爬取数据Go+Gin 接口防止用户频繁访问

在Go+Gin框架中，可以利用中间件实现API防洪（防止用户频繁访问）功能。一种常见的防洪技术是利用Go的漏桶算法或令牌桶算法进行流量控制。
以下面的Go代码为例，演示了一个简单的令牌桶方式的中间件：

package main
 
import (
    "net/http"
    "time"
 
    "github.com/gin-gonic/gin"
    "golang.org/x/time/rate"
)
 
// 创建一个 map 来存储每个 IP 的访问限制
var visitors = make(map[string]*rate.Limiter)
 
// 获取用户 IP 为 key 的令牌桶，若不存在则创建一个
func getVisitor(ip string) *rate.Limiter {
    limiter, exists := visitors[ip]
    if !exists {
        limiter = rate.NewLimiter(1, 5)  // 每秒生成1个令牌，且桶中最多存储5个令牌
        visitors[ip] = limiter
    }
 
    return limiter
}
 
// 控制访问频率的中间件
func limitMiddleware(c *gin.Context) {
    limiter := getVisitor(c.ClientIP())
    // Allow方法返回是否可以接收请求，此处是否可以拿取令牌
    if !limiter.Allow() {
        c.String(http.StatusTooManyRequests, "Too many requests.")
        c.Abort()
        return
    }
 
    c.Next()
}
 
func main() {
    router := gin.Default()
    router.Use(limitMiddleware)
 
    router.GET("/", func(c *gin.Context) {
        c.String(http.StatusOK, "Welcome!")
    })
 
    router.Run(":8080")
}