1. 局部中间件

  1. package main
  2. import (
  3. "fmt"
  4. "time"
  5. "github.com/gin-gonic/gin"
  6. )
  7. // 定义中间
  8. func MiddleWare() gin.HandlerFunc {
  9. return func(c *gin.Context) {
  10. t := time.Now()
  11. fmt.Println("中间件开始执行了")
  12. // 设置变量到Context的key中,可以通过Get()取
  13. c.Set("request", "中间件")
  14. // 执行函数
  15. c.Next()
  16. // 中间件执行完后续的一些事情
  17. status := c.Writer.Status()
  18. fmt.Println("中间件执行完毕", status)
  19. t2 := time.Since(t)
  20. fmt.Println("time:", t2)
  21. }
  22. }
  23. func main() {
  24. // 1.创建路由
  25. // 默认使用了2个中间件Logger(), Recovery()
  26. r := gin.Default()
  27. //局部中间键使用
  28. r.GET("/ce", MiddleWare(), func(c *gin.Context) {
  29. // 取值
  30. req, _ := c.Get("request")
  31. fmt.Println("request:", req)
  32. // 页面接收
  33. c.JSON(200, gin.H{"request": req})
  34. })
  35. r.Run()
  36. }

效果演示:

局部中间件 - 图1

局部中间件 - 图2