将 http.Response 转换为字节数组

Question

我正在尝试开发一个 tcp 代理，在这个 tcp 代理中我必须同时处理 http 和 tcp 请求。

目前，对于传入的请求，我检测它是 http 还是 tcp 请求，如果是 http，则将其解析为 http.Request：

func (s *TcpProxy) OnMessage(c *connection.Connection, ctx interface{}, data []byte) interface{} {
    reader := bytes.NewReader(data)
    newReader := bufio.NewReader(reader)
    req, err := http.ReadRequest(newReader)
    // This is an http request
}

现在我可以方便地操作请求，因为我可以使用从该接口公开的方法，然后最终我会将响应从我的代理发送回接收到入站请求的服务。

func (s *TcpProxy) OnMessage(c *connection.Connection, ctx interface{}, data []byte) interface{} {
    reader := bytes.NewReader(data)
    newReader := bufio.NewReader(reader)
    req, err := http.ReadRequest(newReader)
    // Manipulate http request
    // ...
    // Proxy the request
    proxyReq, err := http.NewRequest(req.Method, proxyUrl, req.Body)

    // Capture the duration while making a request to the destination service.
    res, err := httpClient.Do(proxyReq)
    
    buf := res.ToBuffer() // <= How can I achieve this
   
    c.Send(buf)

    c.Close()
    return nil
}

但是我找不到将响应转换回字节或字符串数​​组的方法，是我遗漏了什么吗？

Answer 1

http.Request 对象有一个Write 方法：

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

Write 以有线格式写入 HTTP/1.1 请求，即标头和正文。

您可以使用它来将字节写入缓冲区对象。例如：

package main

import (
    "bytes"
    "fmt"
    "net/http"
)

func main() {
    var buf bytes.Buffer

    req, err := http.NewRequest("GET", "http://google.com", nil)
    if err != nil {
        panic(err)
    }

    client := &http.Client{}
    res, err := client.Do(req)
    if err != nil {
        panic(err)
    }
    defer res.Body.Close()

    if err := res.Write(&buf); err != nil {
        panic(err)
    }

    // ...do whatever you want with the buffer here...
    fmt.Println(buf.String())
}

一个 Buffer 对象有一个 Bytes 方法，如果这是你想要的，它将返回一个字节数组。