go-网络请求之限流

aside section._1OhGeD · · 2065 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

golang自身的net/http网络接口也很方便，但是复杂的业务就没有开源框架来的顺手，比如gin微服务框架
限流用在业务较多的服务中，每秒允许多少请求，延迟最长多少，允许同时多少IP在线等。
1，net/http，简单监听一下本地3050端口，限制只能允许一个IP接入，max=1

func main() {
    l, err := net.Listen("tcp", "192.168.100.8:3050")
    if err != nil {
        fmt.Println("list err")
    }
    defer l.Close()
    const max = 1
    l = netutil.LimitListener(l, 1)

    var open int32
    http.Serve(l, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if n := atomic.AddInt32(&open, 1); n > max {
            fmt.Println("connetct",n,max)
        }
        defer atomic.AddInt32(&open, -1)
        time.Sleep(10 * time.Millisecond)
        fmt.Fprint(w, "some body")
    }))
}

接入成功的客户端显示：

image.png

以后接入的客户端显示：

image.png

无法请求到服务器
如果max值改大一点，那么更多客户端就可以连接到服务器。

2，gin-limiter，gin作为最近golang最流行的服务框架之一，使用起来也非常方便engine := gin.default()，可以使用各种中间件配置，测试限流效果和net/http一致，

func LimitHandler(lmt *limiter.Limiter) gin.HandlerFunc {

 return func(c *gin.Context) {

  httpError := tollbooth.LimitByRequest(lmt, c.Writer, c.Request)

  if httpError != nil {

   c.Data(httpError.StatusCode, lmt.GetMessageContentType(), []byte(httpError.Message))

   c.Abort()

  } else {

   c.Next()

  }

 }

}

lmt := tollbooth.NewLimiter(max,nil)
 lmt.SetMessage("error，，")

engine.POST("/login",tollbooth.LimitHandler(lmt,nil),LoginHandler)

有疑问加站长微信联系（非本文作者）