本文介绍一种高效方式，为任意 io.reader 添加固定前缀（如协议头），避免将全部数据读入内存计算长度，通过类型断言和接口检测智能优化性能。
本文介绍一种高效方式，为任意 io.reader 添加固定前缀（如协议头），避免将全部数据读入内存计算长度，通过类型断言和接口检测智能优化性能。
在 Go 中，若需为一个 io.Reader 添加前缀（例如包含长度信息的 header），最直观的做法是先 io.Copy 到 bytes.Buffer 中以获取总长度，再拼接 header 和原始数据——但这种“全量缓冲”策略在高并发场景下会显著增加内存压力，尤其当每个请求都涉及数十 KB 数据时。
更优解是：尽可能避免预读全部数据，优先利用底层 reader 自身的长度信息能力。Go 标准库（如 net/http）正是采用类似策略来推断 Content-Length。核心思路是：
- 对已知长度的类型（如
*bytes.Buffer、*bytes.Reader、*strings.Reader）直接调用.Len()； - 对支持随机访问的
io.Seeker（如文件、*os.File、某些网络封装 reader），通过Seek(0, 2)获取末尾位置，再回退到原位，安全得出长度； - 仅当以上都不适用时，才退化为内存缓冲（此时
in已被完全读取，可复用为io.Reader）。
以下是经过生产验证的实现：
func addHeader(in io.Reader) (io.Reader, error) {
var n int
switch v := in.(type) {
case *bytes.Buffer:
n = v.Len()
case *bytes.Reader:
n = v.Len()
case *strings.Reader:
n = v.Len()
case io.Seeker:
// 保存当前位置
cur, err := v.Seek(0, io.SeekCurrent)
if err != nil {
return nil, err
}
// 移动到末尾获取长度
end, err := v.Seek(0, io.SeekEnd)
if err != nil {
return nil, err
}
// 恢复原始位置
_, err = v.Seek(cur, io.SeekStart)
if err != nil {
return nil, err
}
n = int(end - cur)
default:
// 退化方案：缓冲全部内容
var buf bytes.Buffer
if _, err := buf.ReadFrom(in); err != nil {
return nil, err
}
n = buf.Len()
in = &buf // 复用已读取的 buffer 作为后续 reader
}
header := strings.NewReader(fmt.Sprintf("header:%d\n", n))
return io.MultiReader(header, in), nil
}✅ 关键优势：
- 对
*bytes.Reader等零拷贝获取长度，无额外内存分配； - 对
io.Seeker类型（如磁盘文件）仅做三次 seek 调用，不触发实际读取； - 仅对真正“流式不可回溯”的 reader（如 HTTP 请求体、管道 reader）才缓冲——且缓冲后
in被复用，避免二次拷贝。
⚠️ 注意事项：
-
io.Seeker的Seek(0, io.SeekEnd)在部分实现中可能不支持（如http.Request.Body默认是不可 seek 的io.ReadCloser），因此该分支仅在明确可控的输入源下生效； - 若原始
in是有状态的（如已部分读取的 reader），io.Seeker分支可能因 seek 行为破坏其内部状态，请确保调用方了解输入语义； - header 字符串建议以
\n结尾，便于接收端解析；若协议要求二进制 header，可用binary.Write或encoding/binary构造。
此方案在保持接口简洁（返回 io.Reader）的同时，兼顾了通用性与性能，在 100KB 级别数据及中等并发下可显著降低 GC 压力与内存占用，是构建高吞吐流式服务的实用技巧。
