两种限流器:
- 漏桶限流, 可应对最大QPS场景
- 请求的时间间隔需要大于PerRequest
- 等待时间wait最大为PerRequest
- 令牌桶限流器, 可应对突发流量场景
- 等待时间wait会累加
两种参数介质:
- using gin
package main
import (
"fmt"
"time"
"github.com/gin-gonic/gin"
lt "github.com/guanhg/ratelimit"
)
func main() {
// 使用token限流器,每分钟最大100请求
gl := lt.NewLimiter("MyLimiter", lt.Token, 100)
/* 分布式服务使用redis介质存储参数,不同主机服务需要相同的name
err := gl.SetRedis("127.0.0.1:6379")
if err != nil {
fmt.Errorf(err)
return
}
*/
LimiteHandler := func() gin.HandlerFunc {
return func(ctx *gin.Context) {
wt, b, err := gl.Acquired()
if err != nil {
ctx.Abort()
ctx.JSON(500, "Limiter error")
return
}
if !b {
ctx.Abort()
waitResp := fmt.Sprintf("Server is busy! Wait for about %f secords", wt.Seconds())
ctx.JSON(500, waitResp)
return
}
ctx.Next()
}
}
serv := gin.Default()
serv.Use(LimiteHandler())
serv.Any("/ping", func(c *gin.Context) {
c.JSON(200, "Pong!")
})
serv.Run(":8080")
}
# test
for ((i=1;i<100;i++)) do echo "$i: $(curl -s https://localhost:8080/ping) \n----"; sleep 0.05; done;