三十三、Gin的中间件

news2024/12/28 18:32:24

定义：

一、处理请求前执行

1、模拟请求前需要鉴权

2、使用use方法

3、实验

二、处理请求后执行

1、在具体方法中添加输出方便观察整个处理请求过程生命周期

2、在next方法后增加输出即可在处理完成请求后执行内容

3、最终执行结果

定义：

在Gin框架中，中间件是用于处理请求和响应的功能模块，可以在处理请求之前或之后处理一些额外的操作

一、处理请求前执行

1、模拟请求前需要鉴权

通过获取key为session_id的值，没有获取到则为鉴权失败，返回403错误

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

const SessionKey = "session_id"

type SessionAuth struct {
}

func (s *SessionAuth) Auth(ctx *gin.Context) {
	sessionID := ctx.GetHeader(SessionKey)
	if sessionID == "" {
		//如果没有获取到sessionID则执行终端，并返回403错误码
		ctx.AbortWithStatusJSON(http.StatusForbidden, "session_id is empty")
	}
	//不为空的情况下继续执行
	ctx.Next()
}

2、使用use方法

在此group下的请求必须先访问session的auth方法

func CmsRouter(r *gin.Engine) {
	session := &SessionAuth{}
	root := r.Group(rootPath).Use(session.Auth)
	//初始化
	cmsApp := services.NewCmsApp()
	//路由绑定
	root.GET("/hello", cmsApp.Hello)
}

3、实验

启动项目访问路径后返回

"session_id is empty"

二、处理请求后执行

1、在具体方法中添加输出方便观察整个处理请求过程生命周期

func (c *CmsApp) Hello(ctx *gin.Context) {
	fmt.Println("方法开始")
	ctx.JSON(http.StatusOK, gin.H{
		"message": "hello",
	})
	fmt.Println("方法结束")
}

2、在next方法后增加输出即可在处理完成请求后执行内容

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

const SessionKey = "session_id"

type SessionAuth struct {
}

func (s *SessionAuth) Auth(ctx *gin.Context) {
	sessionID := ctx.GetHeader(SessionKey)
	if sessionID == "" {
		//如果没有获取到sessionID则执行终端，并返回403错误码
		ctx.AbortWithStatusJSON(http.StatusForbidden, "session_id is empty")
	}
	//不为空的情况下继续执行
	ctx.Next()
	//执行后
	fmt.Println("调用后执行-success")
}