将响应头添加到每个处理程序而不重复同一行

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

“喜欢组合而不是继承” - 4人帮

继承根本就没有被设计到 Golang 中。;)

如果您正在寻找有关原因部分的详细解释，我相信这已经在 SO 上得到了回答，因此我只想向您指出：在 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))

Answer 3

的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))