Go 怎么用 io.TeeReader 同时读取和计算摘要
来源:17golang原创
时间:2026-09-08 15:10:00 476浏览 收藏
要同时读取请求体并计算 SHA-256 摘要,可以把原始 io.Reader 包成 io.TeeReader,再让后续逻辑只读取这个包装后的 Reader。每次读取到的字节会同步写入哈希对象;等流被完整消费后,再调用 Sum(nil) 取得摘要。它不会缓存整份正文,也不会让你第二次从原始请求体读取。
io.TeeReader把读取到的字节写给一个io.Writer,写入完成前本次读取不会结束。- 哈希摘要要在下游读完包装后的 Reader 后再调用
hash.Sum(nil)。 - 如果下游绕过包装 Reader,或只读取了一部分,摘要也只对应实际经过 TeeReader 的字节。
用 TeeReader 把读取和摘要写入接在一起
最小实现是创建哈希对象,再把请求体交给 io.TeeReader。包装器本身仍然是一个 Reader:下游怎么读取,它就怎么把同一批字节写入哈希对象。
package main
import (
"crypto/sha256"
"fmt"
"io"
"strings"
)
func main() {
source := strings.NewReader("order=42&amount=99")
digest := sha256.New()
stream := io.TeeReader(source, digest)
// 消费包装后的 Reader,读取到的字节才会进入摘要对象。
written, err := io.Copy(io.Discard, stream)
if err != nil {
panic(err)
}
// 读完后再取结果;Sum 不会清空或重置当前哈希状态。
fmt.Printf("bytes=%d digest=%x\\n", written, digest.Sum(nil))
}
io.Copy 返回的字节数可以帮助记录实际消费量;只有当它返回成功时,才适合把当前摘要当作完整正文的摘要。若下游还要解析正文,就把 stream 交给解析器,而不是重新使用 source。

三个边界决定摘要是否可信
官方文档给出的关键约束有三个:TeeReader 没有内部缓冲;写入目标的 Write 必须完成,本次读取才会完成;写入目标返回的错误会被包装成读取错误。因此它适合把读取链和摘要链绑定起来,但不适合把摘要计算当成后台异步任务。
| 检查点 | 正确做法 | 常见误区 |
|---|---|---|
| 读取对象 | 后续组件继续读取 stream | 绕过包装器直接读取原始 Reader |
| 完成时机 | 确认 io.Copy 成功后调用 Sum(nil) | 刚读一小段就把摘要当成全文结果 |
| 错误处理 | 把 Copy 返回的读错和写错一起处理 | 只检查下游解析错误,忽略摘要写入失败 |
在 HTTP 处理链里尤其要注意请求体的所有权:如果中间件先读取了原始 req.Body,后面的包装器看不到已经被消费的字节;如果包装器先交给 JSON 解码器,摘要会随着解码器的读取逐步更新。需要验签时,应明确由哪一层完整消费流,并让同一层在成功读完后保存摘要。

把 TeeReader 接进 HTTP 请求体时怎么排查
先确认包装器确实被下游使用,再记录 io.Copy 或解码器报告的字节数和错误。摘要为空通常不是 SHA-256 算法失效,而是还没有读取,或者读取路径绕过了 TeeReader。摘要长度固定但内容不对时,优先检查是否只读取了部分请求体、是否有提前返回,以及是否在取摘要前调用了 Reset。
- 摘要用于验签时,先保留原始字节边界,不要在读取后重新格式化 JSON。
- 哈希对象实现的是
io.Writer,正常写入一般不会报错;自定义 Writer 则必须处理它的返回错误。 - 如果需要失败后重新读取正文,应该提前缓存受控大小的字节,不能指望 TeeReader 自动回放。
常见问题
TeeReader 会把请求体复制到内存吗?
不会。它没有内部缓冲,只把每次读取的字节同步写给目标 Writer;是否缓存取决于你额外使用的下游组件。
为什么调用 Sum(nil) 得到的摘要不完整?
通常是下游还没读完,或者某段读取绕过了 TeeReader。先确认完整消费成功,再取摘要。
写入摘要失败会返回什么错误?
TeeReader 会把目标 Writer 的写入错误报告为读取错误,因此要检查包装 Reader 的读取或 Copy 返回值。
-
860 收藏
-
843 收藏
-
826 收藏
-
809 收藏
-
792 收藏
-
372 收藏
-
201 收藏
-
376 收藏
-
495 收藏
-
434 收藏
-
309 收藏
-
100 收藏
-
328 收藏
-
Golang · Go教程 | 3小时前 | Go教程 · database/sql · 数据库元数据 · SQL NULL · Go database/sql rows ColumnTypes ColumnType.Length128 收藏
-
319 收藏
-
272 收藏
-
327 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 立即学习 543次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 立即学习 516次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 立即学习 500次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 立即学习 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 立即学习 485次学习