http - 如何检查ResponseWriter是否已编写

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



当然，我可以从指示天气的函数中返回 bool(boolean) 值，例如通过重定向或其他方式写到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是否已写入？

最佳答案

唯一的方法是使用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
    }
}

我还把a simple package交给我来处理。如果愿意，也可以随意使用它。