如何使用Go和http.Transport实现HTTP请求的日志记录？

如何使用Go和http.Transport实现HTTP请求的日志记录？

在使用Go语言进行HTTP请求时，我们经常会遇到需要记录请求的详细信息的情况，例如记录请求的URL、请求方法、请求头、请求体等。这些信息对于调试和排查问题非常有帮助。本文将介绍如何使用Go和http.Transport实现HTTP请求的日志记录。

Go语言中，我们可以使用http包进行HTTP请求，而http.Transport则负责发送和接收HTTP请求和响应。通过自定义实现http.Transport的RoundTrip方法，我们可以在发送请求前和收到响应后进行请求日志的记录。

以下是一个示例代码：

package main

import (
    "log"
    "net/http"
    "net/http/httputil"
    "os"
    "time"
)

// LoggingTransport 实现了http.RoundTripper接口
type LoggingTransport struct {
    Transport http.RoundTripper
    Logger    *log.Logger
}

// RoundTrip 实现了http.RoundTripper接口的RoundTrip方法
func (t *LoggingTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    startTime := time.Now()

    // 打印请求信息
    dump, err := httputil.DumpRequestOut(req, true)
    if err != nil {
        return nil, err
    }
    t.Logger.Println(string(dump))

    // 发送请求
    resp, err := t.Transport.RoundTrip(req)
    if err != nil {
        return nil, err
    }

    // 打印响应信息
    dump, err = httputil.DumpResponse(resp, true)
    if err != nil {
        return nil, err
    }
    t.Logger.Println(string(dump))

    // 计算请求耗时并打印
    duration := time.Since(startTime)
    t.Logger.Printf("Request took %s", duration)

    return resp, nil
}

func main() {
    // 创建自定义Transport
    transport := &LoggingTransport{
        Transport: http.DefaultTransport,
        Logger:    log.New(os.Stdout, "", log.LstdFlags),
    }

    // 创建自定义的http.Client
    client := &http.Client{
        Transport: transport,
    }

    // 创建自定义的http.Request
    req, err := http.NewRequest("GET", "http://www.example.com", nil)
    if err != nil {
        log.Fatal(err)
    }

    // 发送请求
    resp, err := client.Do(req)
    if err != nil {
        log.Fatal(err)
    }
    defer resp.Body.Close()

    // 处理响应
    // ...

}

在上面的代码中，我们首先定义了一个LoggingTransport类型，它实现了http.RoundTripper接口的RoundTrip方法。在这个方法中，我们首先获取当前时间作为请求开始时间，然后使用httputil包的DumpRequestOut方法将请求信息转换为字节数组并写入日志文件，接着发送请求，然后再使用DumpResponse方法将响应信息转换为字节数组并写入日志文件，最后计算请求耗时并打印。

在main函数中，我们创建了自定义的Transport和Client，分别将其传递给http.Client的Transport字段和http.NewRequest函数的最后一个参数，然后发送请求并处理响应。

通过这种方式，我们可以方便地实现HTTP请求的日志记录。可以根据实际需求，将日志信息输出到控制台、写入文件或者发送到日志收集系统中。

总结：本文介绍了如何使用Go和http.Transport实现HTTP请求的日志记录。我们通过自定义实现http.Transport的RoundTrip方法，在发送请求前和收到响应后分别记录请求的详细信息，从而方便调试和排查问题。希望本文对你有所帮助！