Go gorilla 记录每个请求的持续时间和状态代码

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《Go gorilla 记录每个请求的持续时间和状态代码》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我有一个使用 gorilla/mux v1.7.3 和 gorilla/context v1.1.1 在 go1.13 中编写的 rest api。

我想记录每个请求的持续时间和状态代码。我已经有了一个基本的工作版本，但我想知道是否有一个更干净的选项。

我有大约 20 个 http 调用。我们以其中一个为例：

client.handlefunc("/register", register.handle).methods("post")

register.handle 的签名为：

func handle(w http.responsewriter, r *http.request) { ... }

因此，为了记录每个请求持续时间和状态代码，我只是用“日志处理程序”包装了方法处理程序 register.handle：

client.handlefunc("/register", log.handlerequestwithlog(register.handle)).methods("post")

当 log.handlerequestwithlog 只是对函数执行进行计时，并使用返回的状态代码（使用 azure 指标进行日志记录）进行记录时：

func handlerequestwithlog(h func(http.responsewriter, *http.request) int) http.handlerfunc {
    return func(writer http.responsewriter, request *http.request) {
        starttime := time.now()
        statuscode := h(writer, request)
        duration := time.now().sub(starttime)
        trace := appinsights.newrequesttelemetry(request.method, request.url.path, duration, string(rune(statuscode)))
        trace.timestamp = time.now()
        client.track(trace)
    }
}

因此，为了提取状态代码，我需要每个 http 方法都返回状态代码。所以我将签名更改为：

func Handle(w http.ResponseWriter, r *http.Request) int { .. }

我想知道是否有比为每个请求返回 http 状态代码更好的选择。另外，如果我想从请求中获得更多信息，会发生什么情况？我也会返回这个值吗？

解决方案

您可以定义一个实现 http.responsewriter 的 logresponsewriter 来跟踪响应数据（状态、正文等），而不是更改 http 处理程序签名。

1. 让我们定义 logresponsewriter

type logresponsewriter struct {
    http.responsewriter
    statuscode int
    buf        bytes.buffer
}

func newlogresponsewriter(w http.responsewriter) *logresponsewriter {
    return &logresponsewriter{responsewriter: w}
}

func (w *logresponsewriter) writeheader(code int) {
    w.statuscode = code
    w.responsewriter.writeheader(code)
}

func (w *logresponsewriter) write(body []byte) (int, error) {
    w.buf.write(body)
    return w.responsewriter.write(body)
}

1. 定义logmiddleware（这里，我使用std log和mux.MiddlewareFunc）

type logmiddleware struct {
    logger *log.logger
}

func newlogmiddleware(logger *log.logger) *logmiddleware {
    return &logmiddleware{logger: logger}
}

func (m *logmiddleware) func() mux.middlewarefunc {
    return func(next http.handler) http.handler {
        return http.handlerfunc(func(w http.responsewriter, r *http.request) {
            starttime := time.now()

            logrespwriter := newlogresponsewriter(w)
            next.servehttp(logrespwriter, r)

            m.logger.printf(
                "duration=%s status=%d body=%s",
                time.since(starttime).string(),
                logrespwriter.statuscode,
                logrespwriter.buf.string())
        })
    }
}

1. 在 main.go 中启动服务器。在这里，我们可以在路由器中使用 logmiddleware。

func main() {
    logger := log.new(os.stdout, "", log.lstdflags)

    router := mux.newrouter()
    router.handlefunc("/", func(w http.responsewriter, r *http.request) {
        time.sleep(1 * time.second)

        w.writeheader(http.statusok)
        w.header().set("content-type", "application/json")
        w.write([]byte(`{"msg": "hello world"}`))
    })

    logmiddleware := newlogmiddleware(logger)
    router.use(logmiddleware.func())

    logger.fatalln(http.listenandserve(":8080", router))
}

访问http://localhost:8080时的日志结果：

2020/10/08 00:51:38 duration=1.003927394s status=200 body={"msg": "hello world"}

1. 跟进：如果我想从请求中获取更多信息，会发生什么情况？

在日志中间件中，您可以读取请求标头和正文（来自 r *http.request），然后使用持续时间和响应数据记录它们。请注意，您应该使用 reassign the request body，以便您的实际处理程序可以读取请求正文。

这是对 https://stackoverflow.com/a/64249969/1630083（从下面）的回答。

我使用的 websocket 遇到了这个问题：

http: panic serving 172.21.0.10:44614: interface conversion: *apiserver.logresponsewriter is not http.hijacker: missing method hijack

但是代码来自：http://www.inanzzz.com/index.php/post/q7bs/access-log-middleware-in-golang

// Overrides http.Flusher type.
func (w *LogResponseWriter) Flush() {
    if fl, ok := w.ResponseWriter.(http.Flusher); ok {
        if w.statusCode == 0 {
            w.WriteHeader(http.StatusOK)
        }
        fl.Flush()
    }
}

// Overrides http.Hijacker type.
func (w *LogResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
    hj, ok := w.ResponseWriter.(http.Hijacker)
    if !ok {
        return nil, nil, fmt.Errorf("the hijacker interface is not supported")
    }
    return hj.Hijack()
}

我现在可以使用了，谢谢！

以上就是《Go gorilla 记录每个请求的持续时间和状态代码》的详细内容，更多关于的资料请关注golang学习网公众号！