Golang：我可以删除来自 ReverseProxy 的响应 headers 吗？

Question

我正在使用 httputil.ReverseProxy 将 Amazon s3 文件代理给我的客户。我想隐藏所有来自 Amazon 的 headers - 是否可以不重新实现反向代理？

proxy := httputil.ReverseProxy{Director: func(r *http.Request) {
    r.Header = http.Header{} // Don't send client's request headers to Amazon.
    r.URL = proxyURL
    r.Host = proxyURL.Host
}}
proxy.ServeHTTP(w, r) // How do I remove w.Headers ?

Answer 1

你可以实施ReverseProxy.Transport

type MyTransport struct{
    header http.Header
}
func (t MyTransport) RoundTrip(r *Request) (*Response, error){
    resp, err := http.DefaultTransport.RoundTrip(r)
    resp.Header = t.header
    return resp, err
}
mytransport := MyTransport{
//construct Header
}
proxy := httputil.ReverseProxy{Director: func(r *http.Request) {
    r.Header = http.Header{} // Don't send client's request headers to Amazon.
    r.URL = proxyURL
    r.Host = proxyURL.Host
  },
  Transport: mytransport,
}

Answer 2

这是我对 remove/replace 所有 http.ReverseProxy 响应的解决方案 headers:

type responseHeadersTransport http.Header

func (t responseHeadersTransport) RoundTrip(r *http.Request) (*http.Response, error) {
    resp, err := http.DefaultTransport.RoundTrip(r)
    if err != nil {
        return nil, err
    }
    resp.Header = http.Header(t)
    return resp, nil
}


func ProxyFile(w http.ResponseWriter, r *http.Request) {
    // ...

    headers := http.Header{}
    headers.Set("Content-Type", file.ContentType)
    headers.Set("Content-Length", fmt.Sprintf("%d", file.Filesize))
    headers.Set("Content-Disposition", fmt.Sprintf("inline; filename=\"%s\"", file.Filename))

    proxy := httputil.ReverseProxy{
        Director: func(r *http.Request) { // Remove request headers.
            r.Header = http.Header{}
            r.URL = proxyURL
            r.Host = proxyURL.Host
        },
        Transport: responseHeadersTransport(headers), // Replace response headers.
    }
    proxy.ServeHTTP(w, r)
}

Golang：我可以删除来自 ReverseProxy 的响应 headers 吗？

Golang: Can I remove response headers coming from ReverseProxy?

reverse-proxy

go

Golang：我可以删除来​​自 ReverseProxy 的响应 headers 吗？

Golang: Can I remove response headers coming from ReverseProxy?

reverse-proxy

go

Golang：我可以删除来自 ReverseProxy 的响应 headers 吗？