Go io.TeeReader 记录上传流量而不改变数据流

上传文件时，如果直接把请求体读入 []byte 再统计，内存和延迟都会跟着文件大小增长。更稳妥的做法是把原始 io.Reader 交给 io.TeeReader，再把已经读出的字节同步写给一个只计数的 io.Writer。下游仍然从 TeeReader 读取同一份数据，统计器不会改写内容。

io.TeeReader 的核心是读取时复制，不是事后回放

它接收一个 Reader 和一个 Writer，返回的仍是 Reader。每次下游调用 Read，TeeReader 先从源 Reader 取得数据，再把这段数据写入 Writer，写入完成后才返回读取结果。Go 官方实现明确没有内部缓冲，因此旁路 Writer 变慢时，上传读取也会变慢；Writer 返回错误时，这个错误会从 Read 传回调用方。

用只计数 Writer 记录真正经过的字节

计数 Writer 不需要保存上传内容，只要按 Write 收到的切片长度累加，并返回完整长度即可。下面的包装器还保留了原请求体的 Close，避免为了统计而把资源生命周期改掉：

type byteCounter struct {
    n int64
}

func (c *byteCounter) Write(p []byte) (int, error) {
    // 统计已经被 TeeReader 读到的字节，不复制上传内容。
    c.n += int64(len(p))
    return len(p), nil
}

type readCloser struct {
    io.Reader
    io.Closer
}

func withUploadCount(body io.ReadCloser) (*byteCounter, io.ReadCloser) {
    counter := new(byteCounter)
    // TeeReader 只负责旁路写入，Close 仍委托给原始请求体。
    wrapped := readCloser{
        Reader: io.TeeReader(body, counter),
        Closer: body,
    }
    return counter, wrapped
}

这个计数器是单个请求内使用的。如果多个 goroutine 会同时消费同一个 Reader，应先重新设计读取模型；Go 的 io 原语默认也不能推断为并发安全。

把包装后的 Reader 接到上传转发链路

统计器准备好后，直接把返回的 Reader 交给 io.Copy。复制成功时，返回值是实际从流中读出的字节数；复制提前失败时，计数器可能已经包含部分数据，不能把它当成完整上传量。

func saveUpload(dst io.Writer, body io.ReadCloser) (int64, error) {
    counter, countedBody := withUploadCount(body)
    defer countedBody.Close()

    // io.Copy 按需读取；每次读取都会先经过计数器的 Write。
    written, err := io.Copy(dst, countedBody)
    if err != nil {
        // 这里同时覆盖源 Reader、目标 Writer 和旁路统计器的错误。
        return counter.n, fmt.Errorf("保存上传流失败，已统计 %d 字节: %w", counter.n, err)
    }
    if written != counter.n {
        // 正常的全量复制中两者应一致，差异提示链路实现有问题。
        return counter.n, fmt.Errorf("统计字节数不一致: copy=%d count=%d", written, counter.n)
    }
    return counter.n, nil
}

生产代码还可以把 counter.n 写入请求日志或进度指标，但不要把完整上传内容写入日志。若要限制体积，应在 TeeReader 外层配合 http.MaxBytesReader 或 io.LimitReader，限制与统计是两个不同职责。

三个容易混淆的流量口径

TeeReader 不是异步镜像器，也不会替下游缓存数据。旁路 Writer 写得慢，读取就会被反压；旁路写失败，上传不能继续当作成功。若业务只想观测而不希望观测系统拖慢主链路，应另行设计有界缓冲、丢弃策略和可观测性指标，不能误以为 TeeReader 自带这些能力。

常见问题

io.TeeReader 会把上传内容完整保存下来吗？

不会。它只把当前 Read 返回的字节写给 Writer，是否保存由 Writer 决定；只计数 Writer 只保留一个数字。

为什么统计值比 Content-Length 小？

请求可能没有读完就因目标错误、限制或业务提前退出。TeeReader 统计的是实际经过读取链路的数据，不是请求头声明值。

能不能用 io.MultiWriter 替代 TeeReader？

两者位置不同。MultiWriter 复制写入端已经拥有的数据，TeeReader 在读取端旁路观察数据；对上传 Reader 做流式统计时，TeeReader 更直接。