将响应标头添加到每个处理程序，而不重复相同的行

Question

我正在写一个小网站，每个页面，我都在其标题中添加一个服务器名称：

func httpSignUp(rw http.ResponseWriter, req *http.Request) {
    rw.Header().Set("Server", SERVER_NAME)
}

我想知道是否有一种方法可以设置http.ResponseWriter的默认服务器名称，所以我不必一遍又一遍地使用相同的行？

Answer 1

创建一个包装器来设置标头：

func wrap(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (
         w.Header().Set("Server", SERVER_NAME)
         h.ServeHTTP(w, r)
    })
}

包裹个人处理程序

http.Handle("/path", wrap(aHandler)(
http.Handle("/another/path", wrap(anotherHandler))

或传递给ListenAndServe的根处理程序：

log.Fatal(http.ListenAndServe(addr, wrap(rootHandler))

Answer 2

http.ResponseWriter是interface，而不是struct。所以，你不能直接扩展它。你需要扩展内部struct。但这不是解决这个问题的惯用方法。

您可以采取的一种方法是使用middleware。 middleware只是一段代码，它将在主请求处理程序之前执行，并可用于执行一些常见任务。

例如，使用middleware方法编写相同的东西：

func injectServerHeader(handler http.Handler, serverName string) http.Handler {
    ourFunc := func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Server", serverName)
        handler.ServeHTTP(w, r)
    }
    return http.HandlerFunc(ourFunc)
}

这样，您可以使用中间件包装实际的http处理程序。因此，您不必一遍又一遍地编写相同的代码。

例如：

http.Handle("/some-path", injectServerHeader(aHandler))

Answer 3

“更喜欢继承的构成” - Gang of 4

首先，Golang并没有将继承设计为Golang。 ;）

如果你正在寻找有关为什么部分的详细解释，我相信这已经得到了回答，因此我只想指出它：Embedding instead of inheritance in Go。

那么，您实际上可以通过适配器设计模式实现相同的结果，这使您能够从内置库扩展功能，并且对我来说，它比继承更灵活。

func adapter(h http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Server", SERVER_NAME)
    h.ServeHTTP(w, r)
  })
}

或者传入serverName作为参数：

func adapter(h http.Handler, serverName string) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Server", serverName)
    h.ServeHTTP(w, r)
  })
}

---

最后，您可以灵活地选择要“继承”的处理程序：

http.Handle("/path", adapter(your_handler))

或者，如果它适用于每个处理程序，只需“继承”到根处理程序：

http.ListenAndServe(port, adapter(root_Handler))