如何使用 proxy.ModifyResponse？

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《如何使用 proxy.ModifyResponse？》，聊聊，希望可以帮助到正在努力赚钱的你。

我已经实现了最基本的反向代理来提取页面，然后向正文添加一些内容。不幸的是，我添加到 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))
}

解决方案

您的 modifyresponse 代码未修改响应。在您的代码中，您有：

r.write(bytes.newbufferstring("monkeys"+html))

您似乎正在写回复，但不是。 r 是 *http.response ，它有一个 write 方法：

func (r *response) write(w io.writer) error

document 说：

write 以 http/1.x 服务器响应格式将 r 写入 w，包括状态行、标头、正文和可选的尾部。

由于 bytes.buffer 还实现了 io.writer，因此代码可以编译，但不是从缓冲区写入响应，而是将响应写入缓冲区，这是绝对不希望的。

事实是，http.response 根本没有提供修改其 body 的方法。很明显，bodey 是一个 io.readcloser。您可以做的是使用正文创建一个新的 io.readcloser 。请记住更改 header 中的 content-length 字段，否则可能会导致问题。

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

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何使用 proxy.ModifyResponse？》文章吧，也可关注golang学习网公众号了解相关技术文章。