中间件

  • 分类使用方式
  1. // 1.全局中间件
  2. router.Use(gin.Logger())
  3. router.Use(gin.Recovery())
  5. // 2.单路由的中间件，可以加任意多个
  6. router.GET("/benchmark", MyMiddelware(), benchEndpoint)
  8. // 3.群组路由的中间件
  9. authorized := router.Group("/", MyMiddelware())
  10. // 或者这样用：
  11. authorized := router.Group("/")
  12. authorized.Use(MyMiddelware())
  13. {
  14.     authorized.POST("/login", loginEndpoint)
  15. }

  • 自定义中间件
  1. //定义
  2. func Logger() gin.HandlerFunc {
  3.     return func(c *gin.Context) {
  4.         t := time.Now()
  6.         // 在gin上下文中定义变量
  7.         c.Set("example", "12345")
  9.         // 请求前
  11.         c.Next()//处理请求
  13.         // 请求后
  14.         latency := time.Since(t)
  15.         log.Print(latency)
  17.         // access the status we are sending
  18.         status := c.Writer.Status()
  19.         log.Println(status)
  20.     }
  21. }
  22. //使用
  23. func main() {
  24.     r := gin.New()
  25.     r.Use(Logger())
  27.     r.GET("/test", func(c *gin.Context) {
  28.         //获取gin上下文中的变量
  29.         example := c.MustGet("example").(string)
  31.         // 会打印: "12345"
  32.         log.Println(example)
  33.     })
  35.     // 监听运行于 0.0.0.0:8080
  36.     r.Run(":8080")
  37. }

  • 中间件参数

  • 内置中间件
    1.简单认证BasicAuth

  1. // 模拟私有数据
  2. var secrets = gin.H{
  3.     "foo":    gin.H{"email": "foo@bar.com", "phone": "123433"},
  4.     "austin": gin.H{"email": "austin@example.com", "phone": "666"},
  5.     "lena":   gin.H{"email": "lena@guapa.com", "phone": "523443"},
  6. }
  8. func main() {
  9.     r := gin.Default()
  11.     // 使用 gin.BasicAuth 中间件，设置授权用户
  12.     authorized := r.Group("/admin", gin.BasicAuth(gin.Accounts{
  13.         "foo":    "bar",
  14.         "austin": "1234",
  15.         "lena":   "hello2",
  16.         "manu":   "4321",
  17.     }))
  19.     // 定义路由
  20.     authorized.GET("/secrets", func(c *gin.Context) {
  21.         // 获取提交的用户名（AuthUserKey）
  22.         user := c.MustGet(gin.AuthUserKey).(string)
  23.         if secret, ok := secrets[user]; ok {
  24.             c.JSON(http.StatusOK, gin.H{"user": user, "secret": secret})
  25.         } else {
  26.             c.JSON(http.StatusOK, gin.H{"user": user, "secret": "NO SECRET :("})
  27.         }
  28.     })
  30.     // Listen and serve on 0.0.0.0:8080
  31.     r.Run(":8080")
  32. }