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

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