如何使用 proxy.ModifyResponse？

Question

我已经实现了最基本的反向代理来拉一个页面，然后向正文添加一些内容。不幸的是，我添加到 html 的尝试没有生效。下面的代码只显示了原始页面，但没有我在响应中添加的“猴子”。需要哪些额外的调用才能使其工作？我最终想用它来替换自定义 css 的 css。

package main

import (
    "log"
    "net/http"
    "net/http/httputil"
    "net/url"
    "fmt"
    "github.com/PuerkitoBio/goquery"
    "bytes"
)

type Director func(*http.Request)

func (f Director) Then(g Director) Director {
    return func(req *http.Request) {
        f(req)
        g(req)
    }
}

func hostDirector(host string) Director {
    return func(req *http.Request) {
        req.Host = host
    }
}

func UpdateResponse(r *http.Response) error {
    doc, err := goquery.NewDocumentFromReader(r.Body)
    if err != nil{
        //log.New("Research")
        log.Fatal("Bad doc %v", err)
        return err
    }
    html, err := goquery.OuterHtml(doc.First())
    if err != nil{
        log.Fatal("Bad html %v", err)
        return err
    }
    fmt.Printf("Body %v", html)
    r.Write(bytes.NewBufferString("monkeys"+html))

    return nil
}

func main() {
    url, _ := url.Parse("http://cnn.com/")
    proxy := httputil.NewSingleHostReverseProxy(url)

    d := proxy.Director
    // sequence the default director with our host director
    proxy.Director = Director(d).Then(hostDirector(url.Hostname()))
    proxy.ModifyResponse = UpdateResponse
    http.Handle("/", proxy)
    log.Fatal(http.ListenAndServe(":9090", nil))
}

Answer 1

您的ModifyResponse代码没有修改响应。在您的代码中，您有：

r.Write(bytes.NewBufferString("monkeys"+html))

似乎您正在写回复，但不是。r是*http.Response其具有Write方法：

func (r *Response) Write(w io.Writer) error

该文件说：

Write 以 HTTP/1.x 服务器响应格式将 r 写入 w，包括状态行、标题、正文和可选的预告片。

由于bytes.Buffer也实现了 a io.Writer，代码编译，但不是从缓冲区写入响应，而是将响应写入缓冲区，这是绝对不受欢迎的。

事实是，http.Response根本不提供修改其Body. 很明显，这Bodey是一个io.ReadCloser. 你可以做的是io.ReadCloser用身体创造一个新的。请记住更改中的Content-Length字段，Header否则可能会导致问题。

func UpdateResponse(r *http.Response) error {
    b, _ := ioutil.ReadAll(r.Body)
    buf := bytes.NewBufferString("Monkey")
    buf.Write(b)
    r.Body = ioutil.NopCloser(buf)
    r.Header["Content-Length"] = []string{fmt.Sprint(buf.Len())}
    return nil
}

游乐场：https : //play.golang.org/p/_Tyvo6GVN3x