本文共 1668 字,大约阅读时间需要 5 分钟。
今天在写一个网关拦截中间件的时候,发现一个关于 c.Next 的问题。
事情是这样的,我的网关中间件校验IP发起请求的频率,过高了就拦截此次请求,并加入黑名单,所以当出发拦截的时候,我要终止程序继续向下传递。于是想到了 c.Next
,各种资料都说它是传递到下一个 handler 去处理,测试后发现不是这么回事,于是我把 c.Next
删掉了,发现并不影响事件往下传递,至此我才明白,这个 c.Next
的意思是下一个中间件在此处被调用,因为中间件是个洋葱模型,先进后出,你可以控制在哪个位置去载入下一个中间件,比如如下中间件:
func StatActive() gin.HandlerFunc { return func(c *gin.Context) { start := time.Now() c.Next() // 在此处载入下个 handle cost := time.Since(start).Seconds() if cost > 3 { logging.Info(fmt.Sprintf("API: %s - 耗时:%.2f秒", c.FullPath(), cost)) } }}
如果当前中间件未调用 c.Next,那么会在当前中间件结束时进入下一个中间件。
明白了这些之后,我还需要知道如果终止程序继续往下执行,可以调用 c.Abort()
如果还想顺带响应一些信息的话可以
c.AbortWithStatusJSON(...)
或者
c.Abort()response.JsonResponseError(c, "当前IP请求过于频繁,暂时被封禁~")return
另外说一句,gin 的中间件个数是有上限的,在 context.go 文件中有定义:
const abortIndex int8 = math.MaxInt8 / 2
在我的机器上的值为 63,实际上这个值为总的 handlers 的个数上限。
func (group *RouterGroup) combineHandlers(handlers HandlersChain) HandlersChain { finalSize := len(group.Handlers) + len(handlers) if finalSize >= int(abortIndex) { panic("too many handlers") } mergedHandlers := make(HandlersChain, finalSize) copy(mergedHandlers, group.Handlers) copy(mergedHandlers[len(group.Handlers):], handlers) return mergedHandlers}
在注册的时候,超出这个个数就会报错。
并且中间件链条的中断也是基于这个参数的,比如上面的 Abort()
和 AbortWithStatusJSON()
方法:
func (c *Context) Abort() { c.index = abortIndex}func (c *Context) AbortWithStatusJSON(code int, jsonObj interface{ }) { c.Abort() c.JSON(code, jsonObj)}
然后在中间件运行的时候:
func (c *Context) Next() { c.index++ for c.index < int8(len(c.handlers)) { c.handlers[c.index](c) c.index++ }}
因为 c.handlers
的总长度不会超过 abortIndex
,这个是在注册的时候判断的,所以索引超过 abortIndex
的 handler
就不会被执行到。
转载地址:http://edaui.baihongyu.com/