type CustomResponseWriter struct {
gin.ResponseWriter
body *bytes.Buffer
}
func (w CustomResponseWriter) Write(b []byte) (int, error) {
w.body.Write(b)
return w.ResponseWriter.Write(b)
}
func (w CustomResponseWriter) WriteString(s string) (int, error) {
w.body.WriteString(s)
return w.ResponseWriter.WriteString(s)
}
func AccessLogHandler() gin.HandlerFunc {
return func(c *gin.Context) {
blw := &CustomResponseWriter{body: bytes.NewBufferString(""), ResponseWriter: c.Writer}
c.Writer = blw
c.Next()
fmt.Sprintf("url=%s, status=%d, resp=%s", c.Request.URL, c.Writer.Status(), blw.body.String())
}
}
//初始化中间件
func InitMiddleware(r *gin.Engine) {
// 生成请求唯一id
r.Use(setUUIDContext())
//access log
r.Use(AccessLogHandler())
// 跨域请求处理
r.Use(Cors())
// 异常保护
r.Use(Recover)
}
启动main函数后看到Console中的内容
url=/microservice/api/v1/verify_code_get, status=200, resp={"code":0,"msg":"success"}
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。