gin 框架进行流量限制

通过 golang.org/x/time/rate 包并结合 gin 中间件来实现

package middleware

import (
	"github.com/gin-gonic/gin"
	"golang.org/x/time/rate"
	"net/http"
)

var limiter *rate.Limiter

func init()  {

	// TODO 根据配置文件配置limiter
	limiter = rate.NewLimiter(1, 2)
}

func RateLimit(c *gin.Context) {
	if limiter.Allow() {
		c.Next()
	}else {
		c.AbortWithStatusJSON(http.StatusOK, struct {
			Message string `json:"message"`
		}{"服务器繁忙，请稍后再试"})
	}
}