如何在 Go 中利用自定义的 RoundTripper 函数拦截底层网络请求-Golang-PHP中文网

roundtripper 是 http 客户端执行网络调用的核心接口，需通过替换 transport 字段完全替代默认实现并委托给 http.defaulttransport 以保留连接复用等能力；常见用法是实现日志记录、header 修改等逻辑，注意 body 读取后需重建、context 取消必须透传。

如何在 go 中利用自定义的 roundtripper 函数拦截底层网络请求

RoundTripper 是什么，为什么不能直接替换 Transport.RoundTrip

Go 的 http.Client 发起请求时，真正执行网络调用的是其 Transport 字段持有的 RoundTripper 实例。默认是 http.DefaultTransport，它实现了 RoundTripper 接口——核心就是 RoundTrip(*http.Request) (*http.Response, error) 方法。

你不能“拦截”默认实现，因为它是封装好的结构体；但你可以**提供自己的实现**，只要满足接口签名。关键点：不是“修改”原 RoundTripper，而是**完全替代它**，并把原始逻辑委托给底层（比如 http.DefaultTransport）来保留 DNS、TLS、连接复用等能力。

写一个基础 RoundTripper 包裹器：日志 + 委托

最常见需求是打日志或改 Header。你需要在自己的 RoundTrip 方法里做前置/后置处理，再调用底层 transport 的 RoundTrip。注意：必须复用同一个底层 transport 实例（不能每次 new），否则会丢失连接池和 keep-alive。

type loggingRoundTripper struct {
    rt http.RoundTripper
}

func (l *loggingRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
    log.Printf("→ %s %s", req.Method, req.URL.String())
    resp, err := l.rt.RoundTrip(req)
    if err != nil {
        log.Printf("← error: %v", err)
    } else {
        log.Printf("← %d %s", resp.StatusCode, resp.Status)
    }
    return resp, err
}

// 使用方式：
client := &http.Client{
    Transport: &loggingRoundTripper{rt: http.DefaultTransport},
}

别漏掉 rt 字段初始化，否则 panic
不要在 RoundTrip 里 new http.Transport —— 这会丢掉连接复用
如果需要并发安全，确保底层 rt 本身是线程安全的（http.DefaultTransport 是）

修改请求或响应 Body：必须考虑 io.ReadCloser 生命周期

Body 是 io.ReadCloser，只能读一次。想读取并重放，必须用 io.NopCloser + bytes.NewReader 重建，且要小心内存占用（大 Body 需流式处理）。

使用Go语言搭建家庭相册系统-相关课件

下载

func (m *bodyModifier) RoundTrip(req *http.Request) (*http.Response, error) {
    // 读取原始 body
    bodyBytes, _ := io.ReadAll(req.Body)
    req.Body.Close()

    // 修改 body 内容（例如注入 trace ID）
    modified := bytes.ReplaceAll(bodyBytes, []byte("X-Trace-ID"), []byte("X-Trace-ID: abc123"))

    // 重新赋值 body
    req.Body = io.NopCloser(bytes.NewReader(modified))

    return m.rt.RoundTrip(req)
}