登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go教程

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

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删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>