Go实现Aop拦截,打印请求及响应

AncestorDragon · · 307 次点击 · 开始浏览    置顶

```go import ( "bytes" "github.com/gin-gonic/gin" log "github.com/sirupsen/logrus" "time" ) type responseBodyWriter struct { gin.ResponseWriter body *bytes.Buffer } func (r responseBodyWriter) Write(b []byte) (int, error) { r.body.Write(b) return r.ResponseWriter.Write(b) } /** 请求之前 */ func LogAopReq() func(c *gin.Context) { return func(c *gin.Context) { //设置日志格式 log.SetFormatter(&log.JSONFormatter{ TimestampFormat: "2006-01-02 15:04:05", }) // 开始时间 startTime := time.Now() w := &responseBodyWriter{body: &bytes.Buffer{}, ResponseWriter: c.Writer} c.Writer = w // 处理请求 c.Next() // 结束时间 endTime := time.Now() // 执行时间 latencyTime := endTime.Sub(startTime) // 请求方式 reqMethod := c.Request.Method // 请求路由 reqUri := c.Request.RequestURI // 状态码 statusCode := c.Writer.Status() // 请求IP clientIP := c.Request.Host log.WithFields(log.Fields{ "status_code": statusCode, "latency_time": latencyTime, "client_ip": clientIP, "req_method": reqMethod, "req_uri": reqUri, "response": w.body.String(), }).Info() } } ```

有疑问加站长微信联系(非本文作者)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:1006366459

307 次点击  
加入收藏 微博
2 回复  |  直到 2021-01-18 18:06:23
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传