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

一、背景

我们在使用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请求.