1回答

bobby

2020-12-04

已采纳

package main

import (
   "fmt"
   "net/http"
   "time"

   "github.com/gin-gonic/gin"
)

func MyLogger() gin.HandlerFunc {
   return func(c *gin.Context) {
      t := time.Now()
      c.Set("example", "123456")
      //让原本改执行的逻辑继续执行
      c.Next()

      end := time.Since(t)
      fmt.Printf("耗时:%V\n", end)
      status := c.Writer.Status()
      fmt.Println("状态", status)
   }
}

func Hook404() gin.HandlerFunc {
   return func(c *gin.Context) {

      c.Next()
      status := c.Writer.Status()
      if status == 404 {
         c.JSON(http.StatusOK, gin.H{
            "msg": "页面找不到",
         })
      }
   }
}

func main() {
   router := gin.Default()
   //使用logger和recovery中间件 全局所有
   router.Use(Hook404())

   router.GET("/ping", func(c *gin.Context) {
      c.JSON(http.StatusOK, gin.H{
         "message": "pong",
      })
   })

   router.Run(":8083")
}

可以通过组件来拦截返回状态码

1

3

bobby

回复

难猿

如果你这里能正常运行就可以

2020-12-05

共3条回复