如何检查 ResponseWriter 是否已写入

Question

使用 Golang 的net/http包，如何检查是否ResponseWriter已写入？我收到以下错误消息：

http：多个 response.WriteHeader 调用

当然，我可以从我的函数中返回布尔值等，指示我ResponseWriter使用重定向或其他方式写入的天气，我尝试过，但我肯定能够在ResponseWriter使用简单的方法写入之前检查是否已写入。

我正在寻找一个类似于以下内容的函数，我可以在写入之前使用它ResponseWriter：

if w.Header().Get("Status-Code") == "" {
    http.Redirect(w, r, "/", http.StatusSeeOther)
} else {
    fmt.Println("Response Writer has already been written to.")
}

上面的代码似乎不起作用...任何人都知道如何检查是否ResponseWriter已写入？

Answer 1

做到这一点的唯一方法是使用以下自定义实现http.ResponseWriter：

type doneWriter struct {
    http.ResponseWriter
    done bool
}

func (w *doneWriter) WriteHeader(status int) {
    w.done = true
    w.ResponseWriter.WriteHeader(status)
}

func (w *doneWriter) Write(b []byte) (int, error) {
    w.done = true
    return w.ResponseWriter.Write(b)
}

func myMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        dw := &doneWriter{ResponseWriter: w}
        next.ServeHTTP(dw, r)
        if dw.done {
            // Something already wrote a response
            return
        }
        // Nothing else wrote a response
        w.WriteHeader(http.StatusOK)
        // Whatever you want here
    }
}

我还整理了一个简单的包来为我处理这个问题。如果您愿意，也可以随意使用它。