从外部 Go 中间件控制 HTTP 标头

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《从外部 Go 中间件控制 HTTP 标头》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到golang等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

问题内容

假设我在 Go 中有一个中间件，我想Server用我自己的值覆盖任何现有的标头。

// Server attaches a Server header to the response.
func Server(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("/", authHandler)
http.ListenAndServe(":8000", Server(http.DefaultServeMux, "myapp"))

不幸的是，如果 authHandler 或 DefaultServeMux 调用处理的任何内容w.Header().Add("Server", "foo")（例如httputil.ReverseProxy.ServeHTTP），我最终会在响应中得到两个 Server 标头。

$ http localhost:5962/v1/hello_world
HTTP/1.1 200 OK
Content-Length: 11
Content-Type: text/plain
Date: Tue, 12 Jul 2016 04:54:04 GMT
Server: inner-middleware
Server: myapp

我真正想要的是这样的：

// Server attaches a Server header to the response.
func Server(h http.Handler, serverName string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        h.ServeHTTP(w, r)
        w.Header().Set("Server", serverName)
    })
}

但是，ServeHTTP 的语义不允许这样做：

ServeHTTP 应该将回复头和数据写入 ResponseWriter 然后返回。返回请求完成的信号；在 ServeHTTP 调用完成之后或同时使用 ResponseWriter 或从 Request.Body 读取是无效的。

我已经看到了一些相当丑陋的黑客攻击，例如httputil.ReverseProxy.ServeHTTP中的代码，它复制标题并重写响应代码，或者使用 httptest.ResponseRecorder，它将整个正文读入字节缓冲区。

我还可以颠倒传统的中间件顺序，以某种方式将我的服务器中间件放在最后，或者让它成为最里面的中间件。

我有什么简单的方法吗？

正确答案

您可以定义一个自定义类型，它包装一个 ResponseWriter 并在写入所有标头之前插入 Server 标头，但代价是额外的间接层。这是一个例子：

type serverWriter struct {
    w           http.ResponseWriter
    name        string
    wroteHeader bool
}

func (s serverWriter) WriteHeader(code int) {
    if s.wroteHeader == false {
        s.w.Header().Set("Server", s.name)
        s.wroteHeader = true
    }
    s.w.WriteHeader(code)
}

func (s serverWriter) Write(b []byte) (int, error) {
    return s.w.Write(b)
}

func (s serverWriter) Header() http.Header {
    return s.w.Header()
}

// Server attaches a Server header to the response.
func Server(h http.Handler, serverName string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        sw := serverWriter{
            w:           w,
            name:        serverName,
            wroteHeader: false,
        }
        h.ServeHTTP(sw, r)
    })
}

我在这里写了更多关于这个的内容。https://kev.inburke.com/kevin/how-to-write-go- middleware/

今天关于《从外部 Go 中间件控制 HTTP 标头》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于golang的内容请关注golang学习网公众号！