
在 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)
}附加说明
以上是如何在 Go 中解析文本文件中的 HTTP 请求和响应?的详细内容。更多信息请关注PHP中文网其他相关文章!