我们需要自定义一个日志中间件,按照我们需要的参数进行记录。
自定义 Logger()
middleware/logger/logger.go
package logger import ( "bytes" "encoding/json" "fmt" "github.com/gin-gonic/gin" "go-gin-api/app/config" "go-gin-api/app/util" "log" "os" ) type bodyLogWriter struct { gin.ResponseWriter body *bytes.Buffer } func (w bodyLogWriter) Write(b []byte) (int, error) { w.body.Write(b) return w.ResponseWriter.Write(b) } func (w bodyLogWriter) WriteString(s string) (int, error) { w.body.WriteString(s) return w.ResponseWriter.WriteString(s) } func SetUp() gin.HandlerFunc { return func(c *gin.Context) { bodyLogWriter := &bodyLogWriter{body: bytes.NewBufferString(""), ResponseWriter: c.Writer} c.Writer = bodyLogWriter //开始时间 startTime := util.GetCurrentMilliTime() //处理请求 c.Next() responseBody := bodyLogWriter.body.String() var responseCode int var responseMsg string var responseData interface{} if responseBody != "" { response := util.Response{} err := json.Unmarshal([]byte(responseBody), &response) if err == nil { responseCode = response.Code responseMsg = response.Message responseData = response.Data } } //结束时间 endTime := util.GetCurrentMilliTime() if c.Request.Method == "POST" { c.Request.ParseForm() } //日志格式 accessLogMap := make(map[string]interface{}) accessLogMap["request_time"] = startTime accessLogMap["request_method"] = c.Request.Method accessLogMap["request_uri"] = c.Request.RequestURI accessLogMap["request_proto"] = c.Request.Proto accessLogMap["request_ua"] = c.Request.UserAgent() accessLogMap["request_referer"] = c.Request.Referer() accessLogMap["request_post_data"] = c.Request.PostForm.Encode() accessLogMap["request_client_ip"] = c.ClientIP() accessLogMap["response_time"] = endTime accessLogMap["response_code"] = responseCode accessLogMap["response_msg"] = responseMsg accessLogMap["response_data"] = responseData accessLogMap["cost_time"] = fmt.Sprintf("%vms", endTime - startTime) accessLogJson, _ := util.JsonEncode(accessLogMap) if f, err := os.OpenFile(config.AppAccessLogName, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666); err != nil { log.Println(err) } else { f.WriteString(accessLogJson + "\n") } } }
运行后多请求几次,日志输出在文件中:
{"cost_time":"0ms","request_client_ip":"::1","request_method":"GET","request_post_data":"","request_proto":"HTTP/1.1","request_referer":"","request_time":1567172568233,"request_ua":"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.142 Safari/537.36","request_uri":"/ping","response_code":1,"response_data":null,"response_msg":"pong","response_time":1567172568233} {"cost_time":"0ms","request_client_ip":"::1","request_method":"GET","request_post_data":"","request_proto":"HTTP/1.1","request_referer":"","request_time":1567172569158,"request_ua":"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.142 Safari/537.36","request_uri":"/ping","response_code":1,"response_data":null,"response_msg":"pong","response_time":1567172569158} {"cost_time":"0ms","request_client_ip":"::1","request_method":"POST","request_post_data":"name=admin","request_proto":"HTTP/1.1","request_referer":"","request_time":1567172629565,"request_ua":"PostmanRuntime/7.6.0","request_uri":"/product","response_code":-1,"response_data":null,"response_msg":"Key: 'ProductAdd.Name' Error:Field validation for 'Name' failed on the 'NameValid' tag","response_time":1567172629565}
OK,咱们想要的所有参数全都记录了!
抛出几个问题吧:
1、有没有开源的日志记录工具?
当然有,其中 logrus 是用的最多的,这个工具功能强大,原来我也分享过,可以看下原来的文章《使用 logrus 进行日志收集》。
2、为什么将日志记录到文本中?
因为,日志平台可以使用的是 ELK。
使用 Logstash 进行收集文本文件,使用 Elasticsearch 引擎进行搜索分析,最终在 Kibana 平台展示出来。
3、当大量请求过来时,写入文件会不会出问题?
可能会,这块可以使用异步,咱们可以用下 go 的 chan,具体实现看代码吧,我就不贴了。
本文来自:51CTO博客
感谢作者:wx6087c7391d3cd
查看原文:[系列] - go-gin-api 路由中间件 - 日志记录(三)
相关阅读 >>
ctx.request().body only fetch once
更多相关阅读请进入《Go》频道 >>
Go语言101
一个与时俱进的Go编程知识库。