关于gin-gonic中的中间件使用说明以及原理

今天在写一个网关拦截中间件的时候,发现一个关于 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 ,这个是在注册的时候判断的,所以索引超过 abortIndexhandler 就不会被执行到。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值