Golang Gin 中间件 Next()方法

Golang Gin 中间件 Next()方法_第1张图片

 

Next()方法


  • 在中间件中调用 next() 方法,会从 next() 方法调用的地方跳转到Handler函数
  • Handler函数执行完成,若中间件还有部分代码未执行(中间件中next()之后的代码),则执行该代

Golang Gin 中间件 Next()方法_第2张图片

如果第一个中间件处理直接使用了next函数,直接跳转到handler函数,处理完之后还会回过来处理剩下的中间件函数。

除了最中心的这块,其他的都是中间件,当我一个中间件进来,当我直接使用了next之后,直接跳到业务处理这部分的handler,然后业务处理的handler处理完毕之后,然后它再回跳到一层一层未处理完的这个中间件代码。

所以它就非常像洋葱,它其实中间件是一层一层,但是前面只执行了一半,处理完中间件之后是响应了,响应的时候还会再去处理剩下的这个中间件一层一层。

// 日志中间件

func MiddleWareLog() gin.HandlerFunc {
	//这里也是context,因为你的中间件也是对请求和响应的处理,只要涉及到请求和响应的处理就用到context上下文
	return func(c *gin.Context) {
		fmt.Println("log start")
		c.Next()
		fmt.Println("log end")
	}
}

func MiddleWareRequestId() gin.HandlerFunc {
	return func(c *gin.Context) {
		fmt.Println("request id start")
		c.Next()
		fmt.Println("request id end")
	}
}


func main() {
	r := gin.Default()
	r.Use(MiddleWareLog(), MiddleWareRequestId())
	r.GET("/", func(c *gin.Context) {
		fmt.Println("app handler.......")
		c.JSON(http.StatusOK, gin.H{
			"msg": "success",
		})
	})
	r.Run(":8000")
}

                                      
[GIN-debug] Listening and serving HTTP on :8000
log start
request id start
app handler.......
request id end
log end
[GIN] 2023/04/08 - 09:51:03 | 200 |      2.2505ms |             ::1 | GET      "/"

可以看到如果有多个中间件,然后调用了next方法,那么就到了下一个中间件了。当中间件后面不再有中间件,只剩下业务代码了,那么就去执行业务的handler。 

Golang Gin 中间件 Next()方法_第3张图片

首先可以看到我先注册了log中间件,然后注册了requestid中间件的id。所以说是按照注册中间件的顺序来的,然后会执行完。

  1. 如果上面中间件没有加入next函数,那么它会将中间件里面所有逻辑执行完
  2. 如果只有一层中间件,那么在中间件中调用 next() 方法,会从 next() 方法调用的地方跳转到Handler函数。

如果有多层中间件,那么会跳转到链表里面下一个中间件,上面就是从log中间件跳转到request中间件了。在requestid中间件执行了next函数,它下面没有中间件了,只有业务代码了,那么就去执行业务的handler,并且执行好了之后我返回出去,最后执行剩下的中间件那部分没执行完的逻辑。

其实和洋葱一样,最开始执行的是最外面的一层,后执行的是最里面的一层。但是业务代码执行完了之后,我从最里面这层开始。

Golang Gin 中间件 Next()方法_第4张图片

业务场景:最里面这一层的中间件,就是requestid了,可以去做一个开始的计时,next()然后里面去执行业务代码,执行完之后返回里面再来一个结束的计时,那么就能够拿到这个方法的执行时间。

在写一些微服务框架,做一个metric的展现的时候,其实都是借助中间件这么做的,然后将数据暴露给一个接口,然后Prometheus去拿一下,最后展现出来,就是这样实现的。

实现token认证


http://127.0.0.1:8080/home home家目录需要对token进行验证,验证通过才可访问

Golang Gin 中间件 Next()方法_第5张图片

中间件最大的存储长度是60多,不能再往下存了。

c.abort()就是直接跳到链表最后一个, 后面没有了,那么也就执行完了,其他的中间件就不执行了。

正常情况下是一个token解析,jwt的包,可以生成token和解析token。

package main

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
)

func AuthMiddleWare() gin.HandlerFunc {
	return func(c *gin.Context) {
		//常用场景,客户端也就是接口的调用方,客户端携带token的方式 1:请求头(最常用) 2:请求体 3:uri
		token := c.Request.Header.Get("token")
		fmt.Println("获取token信息:", token)

		if token != "qazw1234" {
			c.JSON(http.StatusUnauthorized, gin.H{
				"msg": "身份验证不通过",
			})
			//Abort方法用于跳到handler链表最后一个index,相当于直接退出,不执行后面的handler了
			c.Abort()
            return
		}
	}
}

func main() {
	r := gin.Default()
	r.GET("/home", AuthMiddleWare(), func(c *gin.Context) {
		c.JSON(http.StatusOK, gin.H{
			"msg": "home路由",
		})
	})
	r.Run(":8000")
}

你可能感兴趣的:(Golang,Gin,golang)