V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
The Go Programming Language
http://golang.org/
Go Playground
Go Projects
Revel Web Framework
fanwei
V2EX  ›  Go 编程语言

如何拦截并修改 response 的内容啊?

  •  
  •   fanwei · 2016-09-03 22:36:45 +08:00 · 1607 次点击
    这是一个创建于 2789 天前的主题,其中的信息可能已经有所发展或是发生改变。

    如下所示,简单做个代理

    func main() {
    	remote, _ := url.Parse("http://www.xxxx.com")
    	proxy := httputil.NewSingleHostReverseProxy(remote)
    	log.Fatal( http.ListenAndServe(":9999", proxy))
    }
    

    我想把代理后返回的一些内容替换掉.但是怎么都成功不了. 下面这样怎么没反映? go 的 http 太"精髓"了.才学几天,摸不着门道. 哪位同学能指个路吗?谢谢.

    func (t *transport) RoundTrip(req *http.Request) (res *http.Response, err error) {
    	res, err = t.RoundTripper.RoundTrip(req)
    
    	b, err := ioutil.ReadAll(res.Body)
    	if err != nil {
    		return nil, err
    	}
    	defer res.Body.Close()
    	b = bytes.Replace(b, []byte("Keywords"), []byte("xxxxxxxx"), -1)
    	body := ioutil.NopCloser(bytes.NewReader(b))
    	res.Body = body
    	res.ContentLength = int64(len(b))
    	res.Header.Set("Content-Length", strconv.Itoa(len(b)))
    	return res, nil
    }
    
    3 条回复    2016-09-04 12:15:39 +08:00
    sakeven
        2
    sakeven  
       2016-09-04 00:03:35 +08:00
    我猜是因为 res 的内容被压缩了,所以没被替换。
    不显式的在 req 里声明压缩响应,是能成功的。
    加一句这个试试 req.Header.Del("Accept-Encoding")
    fanwei
        3
    fanwei  
    OP
       2016-09-04 12:15:39 +08:00
    @sakeven
    ....................低级错误,后来发现确实是这原因.
    谢谢
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   我们的愿景   ·   实用小工具   ·   1335 人在线   最高记录 6543   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 25ms · UTC 23:38 · PVG 07:38 · LAX 16:38 · JFK 19:38
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.