如何在 Go 中解析文本文件中的 HTTP 请求和响应？

在 Go 中解析来自文本文件的 HTTP 请求和响应

使用存储在文本文件中的 HTTP 管道流时，解析请求和响应对于进一步加工至关重要。在本文中，我们将探讨如何使用 http.ReadRequest 和 http.ReadResponse 函数在 Go 中有效地实现此目的。

问题

给定一个文本文件，其中包含一系列 HTTP 请求和响应，目标是将这些组件解析为数据结构，特别是 Connection 的切片

解决方案

利用上述函数提供了一种简单的方法来解析文本文件中的请求和响应：

import (
    "bufio"
    "bytes"
    "fmt"
    "io"
    "io/ioutil"
    "log"
    "net/http"
    "net/http/httputil"
    "os"
)

type Connection struct {
    Request  *http.Request
    Response *http.Response
}

func ReadHTTPFromFile(r io.Reader) ([]Connection, error) {
    buf := bufio.NewReader(r)
    stream := make([]Connection, 0)

    for {
        req, err := http.ReadRequest(buf)
        if err == io.EOF {
            break
        }
        if err != nil {
            return stream, err
        }

        resp, err := http.ReadResponse(buf, req)
        if err != nil {
            return stream, err
        }

        // Save response body
        b := new(bytes.Buffer)
        io.Copy(b, resp.Body)
        resp.Body.Close()
        resp.Body = ioutil.NopCloser(b)

        stream = append(stream, Connection{Request: req, Response: resp})
    }
    return stream, nil
}

使用方法

要使用此功能，只需阅读输入文本即可将文件写入阅读器并调用 ReadHTTPFromFile。该函数将返回 Connection 实例的切片，其中每个连接代表一个请求及其关联的响应。

f, err := os.Open("/tmp/test.http")
if err != nil {
    log.Fatal(err)
}
defer f.Close()
stream, err := ReadHTTPFromFile(f)
if err != nil {
    log.Fatalln(err)
}

附加说明

• http.ReadRequest并且 http.ReadResponse 可以在同一个缓冲读取器上重复调用，直到到达文件末尾。
• 确保兼容性使用您的输入文件，验证它是否包含具有准确值的 Content-Length 标头。
• 如果您的输入文件包含截断的响应（例如提供的示例中），您可以考虑使用 httputil.DumpResponse 来确保更准确的响应响应内容的综合表示。

以上是如何在 Go 中解析文本文件中的 HTTP 请求和响应？的详细内容。更多信息请关注PHP中文网其他相关文章！