go - 在 GO 中的 Gorilla 处理程序中的 LoggingHandlers 实现中自定义日志格式

Question

我想在 LoggingHandler 中的 gorilla 处理程序实现中自定义日志格式.基本上它默认提供通用日志格式。我想根据请求 header 进行自定义。假设我将 tenantId 值作为一个请求 header 传递。然后我想将它添加到 os.StdOut 中。

r := mux.NewRouter()
r.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("This is a catch-all route"))
})
loggedRouter := handlers.LoggingHandler(os.Stdout, r)

127.0.0.1 - sandun [10/Oct/2018:13:55:36 -0700] "GET /api/request HTTP/1.0" 200 2326

然后是请求头

teanantId : 50

预期输出

127.0.0.1 - sandun [10/Oct/2018:13:55:36 -0700] "GET /api/request HTTP/1.0" 200 2326 50

我想在日志输出的最后添加tenantid。

我如何在不改变的情况下做到这一点 gorilla handlers liabry .

更多细节:我可以通过更改 gorilla 处理程序库的内部代码来做到这一点。

// buildCommonLogLine builds a log entry for req in Apache Common Log Format.
// ts is the timestamp with which the entry should be logged.
// status and size are used to provide the response HTTP status and size.
func buildCommonLogLine(req *http.Request, url url.URL, ts time.Time, status int, size int) []byte {
    tenantId, err := strconv.Atoi(req.Header.Get("tenantid"))
    if err != nil {
        tenantId = 0
    }
    username := "-"
    if url.User != nil {
        if name := url.User.Username(); name != "" {
            username = name
        }
    }

    host, _, err := net.SplitHostPort(req.RemoteAddr)

    if err != nil {
        host = req.RemoteAddr
    }

    uri := req.RequestURI

    // Requests using the CONNECT method over HTTP/2.0 must use
    // the authority field (aka r.Host) to identify the target.
    // Refer: https://httpwg.github.io/specs/rfc7540.html#CONNECT
    if req.ProtoMajor == 2 && req.Method == "CONNECT" {
        uri = req.Host
    }
    if uri == "" {
        uri = url.RequestURI()
    }

    buf := make([]byte, 0, 3*(len(host)+len(username)+len(req.Method)+len(uri)+len(req.Proto)+50)/2)
    buf = append(buf, host...)
    buf = append(buf, " - "...)
    buf = append(buf, username...)
    buf = append(buf, " ["...)
    buf = append(buf, ts.Format("02/Jan/2006:15:04:05 -0700")...)
    buf = append(buf, `] "`...)
    buf = append(buf, req.Method...)
    buf = append(buf, " "...)
    buf = appendQuoted(buf, uri)
    buf = append(buf, " "...)
    buf = append(buf, req.Proto...)
    buf = append(buf, `" `...)
    buf = append(buf, strconv.Itoa(status)...)
    buf = append(buf, " "...)
    buf = append(buf, strconv.Itoa(size)...)
    buf = append(buf, " "...)
    buf = append(buf, strconv.Itoa(tenantId)...)
    return buf
}

但我认为这不是很好的解决方案。我期待这个社区能提供好的解决方案。

感谢您的帮助。

Answer 1

gorilla 处理程序库使用 Apache Common Log Format .更改输出格式会使您的日志消息对除您以外的所有人绝对不清楚。

惯用的方法:创建您自己的中间件处理程序，写入 teanantId header 的输出(或任何其他 io.Writer)值。

第二种(我认为是错误的)方式:fork 存储库，改变行为并使用它。