记录 Golang 自定义 ServeHTTP 状态的日志

我正在构建自己的网络框架，并且遇到了问题，当我尝试使用响应状态代码记录请求时，我收到错误：无效的内存地址或零指针取消引用

我的 ServeHTTP 函数：

func (r *Router) ServeHTTP(w http.ResponseWriter, req *http.Request) {

    handler := r.getHandler(req.Method, req.URL.Path)
    if handler == nil {
        notFoundHandler(w, req)
        return
    }
    done := make(chan struct{})

    reqStruct := newReqStruct(w, req)

    go func() {
        handler(reqStruct)
        done <- struct{}{}
        close(done)
    }()
    <-done
    logger.logRequest(req.URL.Path, req.Method, req.Response.StatusCode)

}

正确答案

这一行：

logger.logRequest(req.URL.Path, req.Method, req.Response.StatusCode)

req.Response 只会针对 HTTP 重定向进行填充。在这种情况下，它将为零。

如果您想捕获响应代码，请编写一个包装响应编写器并实现 WriteHeader 来捕获响应代码。