Gin：中间件

VSCode 搭建Gin
Gin：路由配置和获取参数
Gin：路由抽离与分组
Gin：自定义控制器及控制器继承
Gin：中间件
Gin：文件上传
Gin：Cookie及二级域名共享Cookie
Gin：Session

Gin 框架允许开发者在处理请求的过程中，加入钩子（Hook）函数。这个钩子函数就叫中间件，中间件适合处理一些公共的业务逻辑，比如登录认证、权限校验、数据分页、记录日志、耗时统计等。简单来说就是在匹配路由前和匹配路由完成后执行的一系列操作。

1. 添加中间件

• 这里先引入路由中间件和全局中间件，同时在中间件和控制器之间进行数据传递
• Gin 中的中间件必须是一个 gin.HandlerFunc 类型，配置路由的时候可以传递多个 func 回调函数

package main

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

type User struct {
	Id   int64  `json:"id,string"`
	Name string `json:"name"`
}

var key = "name"
var value = "fisher"

//全局中间件
func gobleMiddlewareOne(c *gin.Context) {
	//在中间件和控制器中传递数据
	c.Set(key, value)
	fmt.Println("1-全局中间件One")
	// 调用该请求的剩余处理程序
	c.Next()
	fmt.Println("2-全局中间件One")
}
func gobleMiddlewareTwo(c *gin.Context) {
	fmt.Println("1-全局中间件Two")
	// 调用该请求的剩余处理程序
	c.Next()
	fmt.Println("2-全局中间件Two")
}

//路由中间件
func initMiddlewareOne(c *gin.Context) {
	fmt.Println("1-路由中间件One")
	start := time.Now().UnixMicro()
	c.Next()
	fmt.Println("1-路由中间件One")
	end := time.Now().UnixMicro()
	fmt.Println("3-程序执行时间是: ", end-start)

}
func initMiddlewareTwo(c *gin.Context) {
	fmt.Println("1-路由中间件Two")
	c.Next()
	fmt.Println("2-路由中间件Two")
}

func main() {

	user := User{math.MaxInt64, "fisher"}
	engine := gin.Default()
	engine.Use(gobleMiddlewareOne, gobleMiddlewareTwo)
	engine.GET("/index", initMiddlewareOne, initMiddlewareTwo, func(c *gin.Context) {
		//name是空接口类型interface{}，需要转换
		name, exists := c.Get(key)
		//类型断言
		v, ok := name.(string)
		fmt.Println("首页返回数据", v, exists)
		time.Sleep(time.Second)
		if ok {
			c.JSON(http.StatusOK, gin.H{
				"code": 0,
				"msg":  "goland_go",
				"data": user,
			})
		} else {
			c.JSON(http.StatusBadRequest, gin.H{
				"code": 1,
				"msg":  "error",
			})
		}

	})
	fmt.Printf("%#v", user)
	engine.Run()
}

2. 启动服务

• http://localhost:8080/index

• 查看打印，可以看出中间件的执行顺序

1-全局中间件One
1-全局中间件Two         
1-路由中间件One         
1-路由中间件Two         
首页返回数据 fisher true
2-路由中间件Two
1-路由中间件One
3-程序执行时间是:  1000335
2-全局中间件Two
2-全局中间件One

3. 路由分组中间件

• 如果对路由进行了分组，可以使用下面的方式引入中间件，路由分组可以参考之前的文章

func UserRoutersInit(router *gin.Engine) {

	userRouter := router.Group("/user")
	// 路由分组中间件，下面两种方式都可以
	// userRouter := router.Group("/user",middlewares.InitMiddleware)
	userRouter.Use(middlewares.InitMiddleware)

	userRouter.GET("/addUser", user.UserController{}.AddUser)
	userRouter.GET("/editUser", user.UserController{}.EditUser)
	userRouter.GET("/queryUser", user.UserController{}.QueryUser)

}

4. 使用 goroutine

• 当在中间件或 handler 中启动新的 goroutine 时，不能使用原始的上下文（c *gin.Context），
  必须使用其只读副本（c.Copy()）

engine.GET("/copy", func(c *gin.Context) {
	cCp := c.Copy()
	go func() {
		time.Sleep(5 * time.Second)
		// 这里使用你创建的副本
		fmt.Printf("%v Request path: %v", time.Now().Format("2006-01-02 15:04:05"), cCp.Request.URL.Path)
	}()
	c.String(200, "hello world!")
})

• 启动服务查看打印