Go http.Request.GetBody 怎样为重试保留请求体

如果一个 POST 请求第一次发送后需要再次发送，真正要解决的不是“把同一个 Body 再传一次”，而是为第二次请求准备一个全新的读取器。http.Request.GetBody 就是这个可选的请求体工厂：它返回一份新的 io.ReadCloser，但不会自动替换当前的 req.Body。

官方地址：https://pkg.go.dev/net/http#Request.GetBody

先分清楚：Body 负责本次读取，GetBody 负责生成下一份

客户端发送请求时，Transport 会读取 req.Body。读取完成后，这个流通常已经到达末尾，并且可能已经被关闭。GetBody 的类型是一个无参数函数，它的返回值必须是新的 io.ReadCloser，用途是让客户端在需要再次读取请求体时重新取得一份副本。

这也是 307、308 重定向和自定义重试容易混淆的地方。Go 官方文档说明，307 或 308 在保留原方法和请求体时要求 Request.GetBody 已定义；但对一个普通的业务重试循环，是否再次调用 Do、何时调用以及请求是否允许重放，仍由业务代码决定。

为什么 NewRequest 能自动设置 GetBody

http.NewRequest 并不会对所有 io.Reader 都做缓存。对于标准库能识别的内存型 reader，它既知道当前长度，也知道怎样从原始数据重新构造 reader，因此可以安全地生成 GetBody。常见的三个类型是 *bytes.Buffer、*bytes.Reader 和 *strings.Reader。

package main

import (
    "bytes"
    "fmt"
    "net/http"
)

func newReplayableRequest(payload []byte) (*http.Request, error) {
    // bytes.Reader 保留内存中的原始字节，NewRequest 可以据此设置 GetBody。
    req, err := http.NewRequest(http.MethodPost, "https://api.example.com/items", bytes.NewReader(payload))
    if err != nil {
        return nil, err
    }
    if req.GetBody == nil {
        return nil, fmt.Errorf("request body cannot be replayed")
    }
    // Body 仍是第一次发送要读取的流；GetBody 不会自动替换它。
    return req, nil
}

这里检查 GetBody 只是确认请求具备重新取得 body 的能力，不代表已经执行了重试。正文没有把示例当成真实网络结果，读者仍需根据自己的请求地址、认证信息和业务条件决定是否再次发送。

再次发送前：先取新 Body，再调用 Client.Do

一个常见错误是第一次 Do 返回后直接再次调用 client.Do(req)。如果原来的 Body 已经读到 EOF 或被关闭，第二次请求就可能得到空请求体或读取错误。正确的局部动作是调用 GetBody，把新流放回 req.Body，然后再进入下一次发送。

package main

import (
    "fmt"
    "net/http"
)

func resetRequestBody(req *http.Request) error {
    // GetBody 为空表示当前请求没有准备可重复读取的请求体来源。
    if req.GetBody == nil {
        return fmt.Errorf("request body is not replayable")
    }
    nextBody, err := req.GetBody()
    if err != nil {
        return fmt.Errorf("create request body: %w", err)
    }
    // 重新发送前必须绑定新流；只调用 GetBody 而不赋给 Body 没有作用。
    req.Body = nextBody
    return nil
}

如果重试循环最终不再发送，也应关闭已经创建但未交给 Transport 的 body。实际项目还要保持响应体的关闭和错误处理；本文只聚焦“请求体怎样恢复为可读取状态”这一层。

流式 reader 没有副本时，怎样手动补齐 GetBody

如果传给 NewRequest 的是自定义流式 reader，标准库通常不知道怎样从头生成第二份数据。此时可以在业务上保留一份不可变的字节切片，用闭包为每次调用创建新的 reader；如果数据量很大，则应改用文件、临时存储或其他明确支持重复读取的来源，而不是把大数据无界地复制到内存。

package main

import (
    "bytes"
    "io"
    "net/http"
)

func requestWithManualGetBody(payload []byte) (*http.Request, error) {
    // 拷贝一份输入，避免调用方后续修改底层数组导致重试内容变化。
    saved := append([]byte(nil), payload...)
    req, err := http.NewRequest(http.MethodPost, "https://api.example.com/items", bytes.NewReader(saved))
    if err != nil {
        return nil, err
    }
    req.GetBody = func() (io.ReadCloser, error) {
        // 每次都从保存的只读来源创建全新的 Body，互不共享读取位置。
        return io.NopCloser(bytes.NewReader(saved)), nil
    }
    return req, nil
}

手动设置时仍要保留初始的 req.Body，因为 GetBody 只为后续副本服务。若原始数据只能从一次性网络流读取，且没有落盘或缓存副本，就应明确把请求标记为不可重放，避免用看似完整的重试掩盖数据缺失。

常见问题

GetBody 能不能直接代替 Body？

不能。第一次发送仍由 Body 提供数据；GetBody 只是在需要重新读取时生成新流，调用方要把它赋给 req.Body。

为什么 bytes.Reader 可以自动获得 GetBody？

因为它保留了内存数据和当前范围，NewRequest 能据此构造新的 reader；普通自定义 io.Reader 不一定有同样的信息。

有了 GetBody 就一定应该重试吗？

不一定。GetBody 只解决请求体能否重新读取的问题，是否重试还要看请求的业务语义、服务端是否允许重复提交和当前错误是否适合重试。