当前位置:首页 > 文章列表 > Golang > Go教程 > Go 怎么用 io.TeeReader 同时读取和计算摘要

Go 怎么用 io.TeeReader 同时读取和计算摘要

来源:17golang原创 2026-09-08 15:10:00 0浏览 收藏

要同时读取请求体并计算 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

Go io.TeeReader 将请求体 Reader 分给下游读取和 hash.Hash 写入的静态结构图
图1:看清输入流、TeeReader、下游消费和摘要结果之间的静态连接。

三个边界决定摘要是否可信

官方文档给出的关键约束有三个:TeeReader 没有内部缓冲;写入目标的 Write 必须完成,本次读取才会完成;写入目标返回的错误会被包装成读取错误。因此它适合把读取链和摘要链绑定起来,但不适合把摘要计算当成后台异步任务。

检查点正确做法常见误区
读取对象后续组件继续读取 stream绕过包装器直接读取原始 Reader
完成时机确认 io.Copy 成功后调用 Sum(nil)刚读一小段就把摘要当成全文结果
错误处理把 Copy 返回的读错和写错一起处理只检查下游解析错误,忽略摘要写入失败

在 HTTP 处理链里尤其要注意请求体的所有权:如果中间件先读取了原始 req.Body,后面的包装器看不到已经被消费的字节;如果包装器先交给 JSON 解码器,摘要会随着解码器的读取逐步更新。需要验签时,应明确由哪一层完整消费流,并让同一层在成功读完后保存摘要。

Go io.TeeReader 中读取边界、Writer.Write 写入边界和 hash.Sum 摘要边界关系图
图2:对照读取、写入和取摘要三个边界,定位摘要为空或不完整的原因。

把 TeeReader 接进 HTTP 请求体时怎么排查

先确认包装器确实被下游使用,再记录 io.Copy 或解码器报告的字节数和错误。摘要为空通常不是 SHA-256 算法失效,而是还没有读取,或者读取路径绕过了 TeeReader。摘要长度固定但内容不对时,优先检查是否只读取了部分请求体、是否有提前返回,以及是否在取摘要前调用了 Reset

  • 摘要用于验签时,先保留原始字节边界,不要在读取后重新格式化 JSON。
  • 哈希对象实现的是 io.Writer,正常写入一般不会报错;自定义 Writer 则必须处理它的返回错误。
  • 如果需要失败后重新读取正文,应该提前缓存受控大小的字节,不能指望 TeeReader 自动回放。

常见问题

TeeReader 会把请求体复制到内存吗?

不会。它没有内部缓冲,只把每次读取的字节同步写给目标 Writer;是否缓存取决于你额外使用的下游组件。

为什么调用 Sum(nil) 得到的摘要不完整?

通常是下游还没读完,或者某段读取绕过了 TeeReader。先确认完整消费成功,再取摘要。

写入摘要失败会返回什么错误?

TeeReader 会把目标 Writer 的写入错误报告为读取错误,因此要检查包装 Reader 的读取或 Copy 返回值。

版本声明
本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
Redis ZRANGE 返回相同分数成员时怎么保证顺序Redis ZRANGE 返回相同分数成员时怎么保证顺序
上一篇
Redis ZRANGE 返回相同分数成员时怎么保证顺序
VS Code 怎么为特定语言设置独立的 Format On Save
下一篇
VS Code 怎么为特定语言设置独立的 Format On Save
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    543次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    516次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    500次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    485次学习
查看更多
AI推荐
  • H2O EvalGPT:开源LLM大模型评估与排行榜工具
    H2O EvalGPT
    H2O EvalGPT是H2O.ai推出的开源LLM评估平台,提供详细的大模型性能排行榜、行业特定基准测试及A/B测试功能,助您快速选择最适合项目的高性能大语言模型。
    26次使用
  • SuperCLUE中文大模型评测基准:功能、能力维度与应用指南
    SuperCLUE
    SuperCLUE是权威的中文大语言模型综合评测基准,涵盖语言理解、知识应用、AI Agent智能体及安全性等12项核心能力。通过多轮对话与客观测试,定期发布榜单与技术报告,为模型研发、优化及行业选型提供科学依据。
    179次使用
  • C-Eval中文评测基准:大语言模型多学科能力评估指南
    C-Eval
    深入了解C-Eval中文评估套件,涵盖52个学科与4级难度。本文详解其功能特点、Zero-shot/Few-shot使用方法及代码示例,助您全面评测LLM中文理解与泛化能力。
    118次使用
  • AI Prompt Library:免费AI提示词库,助力ChatGPT高效创作与营销
    AI Prompt Library
    探索AI Prompt Library免费资源库,涵盖营销、写作及多场景AI提示词。兼容ChatGPT、Claude等工具,一键复制优化输出,提升工作效率。
    41次使用
  • Generrated:DALL·E 2/3 AI绘画提示词灵感库与图像对比平台
    Generrated
    Generrated汇集9300+张DALL·E生成图像及对应提示词,支持查看完整图集、对比DALL·E 2与3版本差异,是AI绘图新手学习Prompt设计与获取创作灵感的实用工具。
    23次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码