Golang: how to read response body of ReverseProxy?

前端 未结 4 758
一整个雨季
一整个雨季 2021-01-31 05:13
package main

import (
    \"net/http\"
    \"net/http/httputil\"
    \"net/url\"
)

func main() {
    target := &url.URL{Scheme: \"http\", Host: \"www.google.com\"}         


        
4条回答
  •  南旧
    南旧 (楼主)
    2021-01-31 05:19

    httputil.ReverseProxy has a Transport field. You can use it to modify the response. For example:

    type transport struct {
        http.RoundTripper
    }
    
    func (t *transport) RoundTrip(req *http.Request) (resp *http.Response, err error) {
        resp, err = t.RoundTripper.RoundTrip(req)
        if err != nil {
            return nil, err
        }
        b, err := ioutil.ReadAll(resp.Body)
        if err != nil {
            return nil, err
        }
        err = resp.Body.Close()
        if err != nil {
            return nil, err
        }
        b = bytes.Replace(b, []byte("server"), []byte("schmerver"), -1)
        body := ioutil.NopCloser(bytes.NewReader(b))
        resp.Body = body
        resp.ContentLength = int64(len(b))
        resp.Header.Set("Content-Length", strconv.Itoa(len(b)))
        return resp, nil
    }
    
    // ...
    proxy := httputil.NewSingleHostReverseProxy(target)
    proxy.Transport = &transport{http.DefaultTransport}
    

    Playground example of the whole thing: http://play.golang.org/p/b0S5CbCMrI.

提交回复
热议问题