golang打印完整报文

在golang中，打印完整报文是一个非常常见且重要的需求。无论是在调试阶段还是在生产环境中，了解报文的内容对于排查问题、优化性能以及分析系统行为都至关重要。在本文中，我们将探讨如何使用golang实现打印完整报文的功能。

什么是完整报文？

完整报文指的是包含请求头和请求体、或者响应头和响应体的所有内容的字符串。它是请求或响应中的文本表示，可以用于显示、存储或传输。在HTTP通信中，完整报文由报文首部和报文主体组成。

使用io.Copy

golang中的标准库提供了 io.Copy 方法，该方法可以将一个 Reader 中的内容复制到一个 Writer 中。我们可以借助 io.Copy 方法来读取请求或响应的内容，并将其打印到终端或写入文件。

假设我们有一个用于发送 HTTP 请求的函数 SendRequest，它的定义如下：

func SendRequest() {
    // ....
}

我们可以修改 SendRequest 函数以打印完整报文：

func SendRequest() {
    // 创建请求
    req, err := http.NewRequest("GET", "http://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }
    
    // 发送请求
    client := &http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        log.Fatal(err)
    }
    
    // 打印响应报文
    var buf bytes.Buffer
    io.Copy(&buf, resp.Body)
    fmt.Println(buf.String())
    
    resp.Body.Close()
}

上述代码中，我们创建了一个缓冲区（buffer）变量 buf，然后使用 io.Copy 方法将响应体的内容复制到该缓冲区中。最后，我们可以通过调用 buf.String() 方法，将缓冲区中的内容转换为字符串并打印出来。

使用httputil.DumpResponse

golang 的 net/http 包中提供了 httputil.DumpResponse 方法，可以更方便地打印响应的完整报文。这个方法会返回一个字节数组，其中包含了完整报文的内容。

修改 SendRequest 函数如下：

import "net/http/httputil"

func SendRequest() {
    // 创建请求
    req, err := http.NewRequest("GET", "http://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }
    
    // 发送请求
    client := &http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        log.Fatal(err)
    }
    
    // 打印响应报文
    dump, err := httputil.DumpResponse(resp, true)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(string(dump))
    
    resp.Body.Close()
}

上述代码中，我们使用 httputil.DumpResponse 方法将响应的完整报文转换为字符串并打印出来。第二个参数指定是否要转换为请求或响应的结构体形式。

打印请求报文

类似地，我们可以使用 httputil.DumpRequest 也来打印请求的完整报文：

import "net/http/httputil"

func SendRequest() {
    // 创建请求
    req, err := http.NewRequest("GET", "http://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }
    
    // 打印请求报文
    dump, err := httputil.DumpRequestOut(req, true)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(string(dump))
}

上述代码中，我们使用 httputil.DumpRequestOut 方法将请求的完整报文转换为字符串并打印出来。

总之，在golang中打印完整报文是一项非常实用的功能。通过使用 io.Copy 或 httputil.DumpResponse 方法，我们可以在调试或分析阶段轻松地获取请求或响应的完整报文。这对于我们解决问题和优化系统性能都非常有帮助。