在 go 中为多个处理程序设置 http 标头

Question

我正在尝试为多个处理程序设置一个 http 标头。我的第一个想法是制作一个自定义写入函数，该函数将在编写响应之前设置标题，如底部的代码示例。但是，当我传递一个指向 http.ResponseWriter 的指针并尝试从我的函数访问它时，它告诉我“类型 *http.ResponseWriter 没有 Header 方法”。为多个处理程序设置标头的最佳方法是什么，以及为什么指针不能按我希望的方式工作？func HelloServer(w http.ResponseWriter, req *http.Request) {    type Message struct {        Name string        Body string        Time int64    }    m := Message{"Alice", "Hello", 1294706395881547000}    b, _ := json.Marshal(m)    WriteJSON(&w, b)}func WriteJSON(wr *http.ResponseWriter, rawJSON []byte) {    *wr.Header().Set("Content-Type", "application/json")    io.WriteString(*wr, string(rawJSON))}func main() {    http.HandleFunc("/json", HelloServer)    err := http.ListenAndServe(":9000", nil)    if err != nil {    log.Fatal("ListenAndServer: ", err)    }}

Answer 1

我不确定多处理程序的事情，但我知道为什么你写的代码失败了。关键是这一行：

*wr.Header().Set("Content-Type", "application/json")

由于运算符优先级，被解释为：

*(wr.Header().Set("Content-Type", "application/json"))

由于wr有型*http.ResponseWriter，这是一个指针，以和接口，而不是接口本身，这是不行的。我假设您知道这一点，这就是您这样做的原因*wr。我假设您要向编译器暗示的是：

(*wr).Header().Set("Content-Type", "application/json")

如果我没记错的话，那应该可以编译并正常运行。

Answer 2

您不需要使用，*wr因为它已经引用了一个指针。

wr.Header().Set("Content-Type", "application/json") 应该足够了。

如果你想为每个请求设置“全局”标头，你可以创建一个满足的函数http.HandleFunc（go.auth 有一个很好的例子），然后像这样包装你的处理程序：

http.HandleFunc("/hello", Defaults(helloHandler))

另请查看net/http文档，其中包含更多示例。