快捷導(dǎo)航

Gin的中間件執(zhí)行流程與用法詳解

更新時(shí)間：2024年04月30日 10:31:08 作者：GEEK?JUMP

我們?cè)谑褂肎in框架進(jìn)行Web開(kāi)發(fā)的時(shí)候,基本上都會(huì)遇到登錄攔截的場(chǎng)景,在Gin當(dāng)中,?中間件和業(yè)務(wù)處理函數(shù)都是一樣的類型,都是一種函數(shù),本文給大家介紹了Gin的中間件執(zhí)行流程與用法,需要的朋友可以參考下

一、背景

我們?cè)谑褂肎in框架進(jìn)行Web開(kāi)發(fā)的時(shí)候，基本上都會(huì)遇到登錄攔截的場(chǎng)景。例如某些接口必須在登錄以后才能訪問(wèn)，根據(jù)登錄用戶的信息以及權(quán)限，拿到屬于自己的數(shù)據(jù), 反之，沒(méi)登錄過(guò)則直接拒絕訪問(wèn)。那么我們?cè)趺醋龅竭@些登錄攔截呢? 做過(guò)Java spring開(kāi)發(fā)或者其它框架開(kāi)發(fā)的同學(xué)知道，這種場(chǎng)景一般都是設(shè)置一個(gè)登錄的攔截器，全局統(tǒng)一在攔截器進(jìn)行登錄權(quán)限校驗(yàn)處理，符合登錄條件則放行請(qǐng)求到業(yè)務(wù)函數(shù)，否則拒絕訪問(wèn)。這樣可以方便我們做統(tǒng)一管理，不需要每個(gè)業(yè)務(wù)函數(shù)再自己重復(fù)寫一套一模一樣的攔截邏輯。

在Gin當(dāng)中，中間件和業(yè)務(wù)處理函數(shù)都是一樣的類型，都是一種函數(shù)，這個(gè)函數(shù)簽名是這樣的 func(c *gin.Context){},Gin的Egine對(duì)象就是把這些函數(shù)func(c *gin.Context){}放到一個(gè)slice切片數(shù)組里面，然后默認(rèn)按照順序執(zhí)行。只要你的函數(shù)func(c *gin.Context){}，我們都可以把這個(gè)函數(shù)作為Gin的中間件函數(shù)。

業(yè)務(wù)函數(shù)這個(gè)handlers切片數(shù)組的最后一個(gè)元素, 那么在業(yè)務(wù)函數(shù)前面我們可以增加很多的函數(shù)，這些函數(shù)我們稱為中間件函數(shù).

我們可以調(diào)用c.Next() 執(zhí)行下一個(gè)中間件函數(shù)/后續(xù)函數(shù)，也可以調(diào)用c.Abort()終止后續(xù)的中間件執(zhí)行。

通過(guò)c.Next(), 查看源碼我們看到這么一個(gè)結(jié)構(gòu):

c.Next()就是將這個(gè)handlers[函數(shù)鏈數(shù)組]往下移動(dòng)一個(gè)元素，也就是執(zhí)行調(diào)用下一個(gè)中間件函數(shù)。

每個(gè)函數(shù)接收到的*Context, 都可以隨時(shí)結(jié)束響應(yīng)HTTP請(qǐng)求，也可以獲取HTTP請(qǐng)求參數(shù)，由此做到中間件攔截處理的作用。

二、執(zhí)行流程圖

1、總流程圖

如果沒(méi)有調(diào)用Abort()、Next()這些函數(shù)，則執(zhí)行是按照中間件使用的順序執(zhí)行。

2、Abort()的使用

調(diào)用Abort(), 終止后續(xù)中間件的執(zhí)行，直接響應(yīng)HTTP內(nèi)容。默認(rèn)Abort()，響應(yīng)狀態(tài)碼是200，并且沒(méi)有任何payload響應(yīng)內(nèi)容。除了Abort(), 還有基于Abort()演變的幾個(gè)函數(shù)，方便我們更改響應(yīng)狀態(tài)碼例如403、內(nèi)容等等。

源測(cè)試代碼:

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()
	// 使用兩個(gè)中間件函數(shù), 按照定義的順序執(zhí)行 A,B
	c.Use(A, B)
	c.GET("/hello", func(c *gin.Context) {
		fmt.Println("業(yè)務(wù)函數(shù)")
		c.JSON(http.StatusOK, gin.H{
			"message": "hello world",
		})
	})
	c.Run()
}

運(yùn)行結(jié)果: 運(yùn)行到A中間件遇到Abort(), 后續(xù)的中間件或業(yè)務(wù)函數(shù)也不會(huì)被執(zhí)行

3、Next()的使用 ;

源測(cè)試代碼:

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()
	// 使用兩個(gè)中間件函數(shù), 按照定義的順序執(zhí)行 A,B
	c.Use(A, B)
	c.GET("/hello", func(c *gin.Context) {
		fmt.Println("業(yè)務(wù)函數(shù)")
		c.JSON(http.StatusOK, gin.H{
			"message": "hello world",
		})
	})
	c.Run()
}

運(yùn)行結(jié)果:

先運(yùn)行到A中間件， A調(diào)用c.Next()此時(shí)繼續(xù)調(diào)用下一個(gè)中間件，所以先打印A1、到B1，最后到業(yè)務(wù)函數(shù)，輸出"業(yè)務(wù)函數(shù)". 最后業(yè)務(wù)函數(shù)執(zhí)行完畢，退回上一層B中間件，B中間件執(zhí)行完畢，回到A. 此時(shí)A里面的c.Next()已經(jīng)全部執(zhí)行完畢，最后輸出A1-end，最后響應(yīng)HTTP請(qǐng)求.