详解Gin的中间件执行流程与用法

使用 Gin 框架进行 Web 开发的时候，基本上都会遇到登录拦截的场景。 例如某些接口必须在登录以后才能访问，根据登录用户的信息以及权限，拿到属于自己的数据，反之，没登录过则直接拒绝访问。  那么如何做到这些登录拦截呢？做过 Java spring 开发或者其它框架开发的童鞋知道，这种场景一般都是设置一个登录的拦截器，全局统一在拦截器进行登录权限校验处理， 符合登录条件则放行请求到业务函数，否则拒绝访问。 这样可以方便我们做统一管理，不需要每个业务函数再自己重复写一套一模一样的拦截逻辑。

但是，在 Gin 当中， 中间件和业务处理函数都是一样的类型，都是一种函数，这个函数签名是这样的func(c *gin.Context){}，Gin 的 Egine 对象就是把这些函数func(c *gin.Context){}放到一个 slice 切片数组里面， 然后默认按照顺序执行。 只要你的函数func(c *gin.Context){}，我们都可以把这个函数作为 Gin 的中间件函数。

业务函数这个 handlers 切片数组的最后一个元素, 那么在业务函数前面我们可以增加很多的函数，这些函数我们称为中间件函数.

我们可以调用c.Next()执行下一个中间件函数/后续函数， 也可以调用c.Abort()终止后续的中间件执行。

通过c.Next()，查看源码我们看到这么一个结构：

c.Next()就是将这个 handlers[函数链数组]往下移动一个元素，也就是执行调用下一个中间件函数。

每个函数接收到的*Context, 都可以随时结束响应 HTTP 请求，也可以获取 HTTP 请求参数，由此做到中间件拦截处理的作用。

二、执行流程图

1、总流程图

如果没有调用 Abort()、Next()这些函数，则执行是按照中间件使用的顺序执行。

2、Abort()的使用

调用 Abort()，终止后续中间件的执行，直接响应 HTTP 内容。 默认 Abort()， 响应状态码是 200，并且没有任何 payload 响应内容。  除了 Abort()，还有基于 Abort()演变的几个函数，方便我们更改响应状态码例如 403、内容等等。

源测试代码：

package main
 
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
)
 
func A(c *gin.Context) {
	fmt.Println("A1")
	c.Abort()
}
 
func B(c *gin.Context) {
	fmt.Println("B1")
}
 
func main() {
	c := gin.Default()
	// 使用两个中间件函数, 按照定义的顺序执行 A,B
	c.Use(A, B)
	c.GET("/hello", func(c *gin.Context) {
		fmt.Println("业务函数")
		c.JSON(http.StatusOK, gin.H{
			"message": "hello world",
		})
	})
	c.Run()
}

运行结果：运行到 A 中间件遇到 Abort()，后续的中间件或业务函数也不会被执行。

3、Next()的使用

源测试代码：

package main
 
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
)
 
func A(c *gin.Context) {
	fmt.Println("A1")
	c.Next()
	fmt.Println("A1-end")
}
 
func B(c *gin.Context) {
	fmt.Println("B1")
}
 
func main() {
	c := gin.Default()
	// 使用两个中间件函数, 按照定义的顺序执行 A,B
	c.Use(A, B)
	c.GET("/hello", func(c *gin.Context) {
		fmt.Println("业务函数")
		c.JSON(http.StatusOK, gin.H{
			"message": "hello world",
		})
	})
	c.Run()
}

运行结果:

先运行到 A 中间件， A 调用 c.Next()此时继续调用下一个中间件， 所以先打印 A1、到 B1， 最后到业务函数， 输出”业务函数”. 最后业务函数执行完毕，退回上一层 B 中间件，B 中间件执行完毕，回到 A。此时 A 里面的 c.Next()已经全部执行完毕，最后输出 A1-end， 最后响应 HTTP 请求。

所以如果要想做最后的响应拦截，那么第一个中间件要通过 c.Next()的方式，在 c.Next()之后添加最终响应逻辑。 例如可以统一更改 status 状态码、或者添加 header、或者删除 header 等等。

以上关于详解Gin的中间件执行流程与用法的文章就介绍到这了，更多相关内容请搜索码云笔记以前的文章或继续浏览下面的相关文章，希望大家以后多多支持码云笔记。