03-Gin中间件-摩杜云开发者社区

一：全局中间件

1.1: 简单的中间件案例

实现的功能：在每个请求之前，追加一个随机值。
随机值中间件的实现

package middleware

import (
  "github.com/gin-gonic/gin"
  "math/rand"
  "time"
)

func init() {
  rand.Seed(time.Now().UnixNano())
}

var letterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")

func RandStringRunes(n int) string {
  b := make([]rune, n)
  for i := range b {
    b[i] = letterRunes[rand.Intn(len(letterRunes))]
  }
  return string(b)
}

// RandomMiddleWare MiddleWare 随机值中间件，每个请求进来， 我们都产生一个随机值， 存放在上下文中。
func RandomMiddleWare() gin.HandlerFunc {
  return func(c *gin.Context) {
    // 随机值中间件会在请求上下文中， 设置一个随机值
    c.Set("randomValue", RandStringRunes(20))
  }
}

将中间件注册到引擎中

package routers

import (
    "github.com/gin-gonic/gin"
    "learn_gin/middleware"
)

type Option func(*gin.Engine)

var options []Option

// Include 将APP中的路由放入options切片中
func Include(opts ...Option) {
    options = append(options, opts...)
}

// Init 初始化引擎， 并将路由注册到引擎中
func Init() *gin.Engine {
    engine := gin.Default()
    // 注册中间件
    engine.Use(middleware.RandomMiddleWare())
    for _, opt := range options {
        opt(engine)
    }
    return engine
}

1.2: 控制请求之后再运行中间件的后半部分

Next()函数控制
代码案例：

// RandomMiddleWare MiddleWare 随机值中间件，每个请求进来， 我们都产生一个随机值， 存放在上下文中。
func RandomMiddleWare() gin.HandlerFunc {
    return func(c *gin.Context) {
        // 随机值中间件会在请求上下文中， 设置一个随机值
        c.Set("randomValue", RandStringRunes(20))
        // 阻断
        c.Next()
        fmt.Printf("返回请求后再执行\n")
    }
}

运行效果：

该请求的随机值是PVaxecLBhcWXDbIkaxHU
返回请求后再执行
[GIN] 2022/09/08 - 18:05:56 | 200 |     218.628µs |       127.0.0.1 | GET      "/animal/login?name=root&password=123456"

二：局部中间件

1: 局部中间件不注册到引擎中
2: 局部中间件再路由中添加即可，这边只有某些路由会走这个中间件。

func Routers(e *gin.Engine) {
    e.GET("/animal/login", middleware.RandomMiddleWare(), testArgs)
}

三：常用的中间件

参考官网推荐

https://github.com/gin-gonic/contrib/blob/master/README.md

目录

一：全局中间件

1.1: 简单的中间件案例

1.2: 控制请求之后再运行中间件的后半部分

二：局部中间件

三： 常用的中间件

三：常用的中间件