戈朗 : Can I remove response headers coming from ReverseProxy?

Question

我正在使用 httputil.ReverseProxy 将 Amazon s3 文件代理给我的客户。我想隐藏所有来自 Amazon 的 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
}}
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,
}