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

Go crypto/sha256.New 怎样对流式输入持续写入并读取摘要

来源:17golang原创

时间:2026-11-11 14:22:32 172浏览 收藏

在文件上传、对象存储或消息流处理中,输入往往不是一整块字节切片,而是一个持续产生数据的 io.Reader。这时可以先用 sha256.New() 创建一个实现了 hash.Hash 的状态对象,再把每段数据写进去,最后调用 Sum(nil) 取出当前 SHA-256 摘要。关键点是:Sum 只读取当前状态,不会清空它。

官方地址:https://pkg.go.dev/crypto/sha256#New

最稳妥的组合是“一个输入流对应一个 hash.Hash”,用 io.Copy 负责持续写入,用返回的错误判断输入是否完整,确认写入成功后再调用 Sum。

先把流接入 sha256.New 返回的 hash.Hash

sha256.New 返回的是 hash.Hash 接口,而不是一次性计算结果。它内部保存了已经写入的数据状态,因此输入可以分成多段,分段边界不会改变最终摘要。

io.Reader、io.Copy、sha256.New 与 hash.Hash 的流式摘要结构说明图
图1:流式摘要结构说明图,展示输入流与 hash.Hash 的静态关系,不是截图或运行证据。
package main

import (
    "crypto/sha256"
    "fmt"
    "io"
    "os"
)

func digestFile(path string) ([]byte, error) {
    // 打开输入文件;调用方负责根据返回的错误决定是否继续。
    file, err := os.Open(path)
    if err != nil {
        return nil, err
    }
    // 函数返回时关闭文件,避免流式读取留下文件描述符。
    defer file.Close()

    // New 创建持续保存 SHA-256 状态的 hash.Hash。
    hasher := sha256.New()
    // io.Copy 反复读取 file,并把每段数据写入 hasher。
    if _, err := io.Copy(hasher, file); err != nil {
        return nil, err
    }
    // Sum(nil) 返回当前摘要;它不会重置 hasher。
    return hasher.Sum(nil), nil
}

func main() {
    // 示例只负责展示调用方式,不把错误当作摘要结果。
    sum, err := digestFile("data.bin")
    if err != nil {
        fmt.Println("读取失败:", err)
        return
    }
    fmt.Printf("SHA-256: %x\\n", sum)
}

这里没有手动维护缓冲区,也没有把整个文件读进内存。io.Copy 返回的字节数可以用于记录处理量,真正决定摘要是否可用的是它返回的错误:读取中断时应直接返回错误,不要发布一个不完整的摘要。

Sum 读取摘要时为什么不会清空状态

hash.Hash 的 Sum(b) 语义是把当前摘要追加到 b 后返回结果。它不会改变底层哈希状态,所以可以在流尚未结束时读取一次阶段性摘要,也可以在最终读取后继续写入更多数据。

hash.Hash 的 Write、Sum、Reset 与摘要字节切片关系说明图
图2:Sum 与状态对象关系说明图,突出读取摘要不等于重置 hash,不是截图或运行证据。
package main

import (
    "crypto/sha256"
    "fmt"
)

func main() {
    hasher := sha256.New()

    // 第一段写入后可以读取阶段性摘要,Sum 不会清空状态。
    _, _ = hasher.Write([]byte("part-a"))
    first := hasher.Sum(nil)

    // 继续写入会在原有状态上累计,而不是从空状态开始。
    _, _ = hasher.Write([]byte("part-b"))
    final := hasher.Sum(nil)

    // Reset 才会把 hasher 恢复到初始状态,适合复用同一个对象。
    hasher.Reset()
    _, _ = hasher.Write([]byte("fresh"))
    fresh := hasher.Sum(nil)

    fmt.Printf("阶段=%x\\n最终=%x\\n重置后=%x\\n", first, final, fresh)
}

如果需要复用一个已有前缀的输出缓冲区,可以传入 buf[:0],这样摘要会追加到该切片的起始位置;不要把返回的摘要切片误认为是可继续写入的哈希状态。

流式写入最容易忽略的三个边界

场景正确处理原因
输入很大让 io.Copy 连接 Reader 和 hash.Hash避免一次性分配完整输入
读取返回错误先处理错误,再决定是否保存摘要摘要可能只覆盖了部分数据
需要重新计算调用 Reset 或新建 hasherSum 本身不会清空累计状态

还要注意 sha256.New 只负责创建 SHA-256 哈希对象,并不替你定义数据边界。做文件摘要时,边界通常是文件结束;做消息摘要时,必须由上层明确一条消息何时结束,不能因为某次读取恰好返回较少字节就提前调用最终摘要。

常见问题:sha256.New 和 sha256.Sum256 怎么选

已有完整 []byte 且不需要分段写入时,sha256.Sum256(data) 更直接,它返回固定长度数组。输入来自文件、网络或其他 io.Reader 时,用 sha256.New 配合 Write 或 io.Copy 更自然,也能在结束前读取阶段性状态。

常见问题:调用 Sum 后还要不要 Reset

如果摘要对象后面还会接着写同一份数据,就不要 Reset;如果要让同一个对象开始计算另一份独立数据,调用 Reset。只调用 Sum 不会开始下一次计算。

小结

把流式 SHA-256 拆成三个动作即可:用 sha256.New 建立状态,用 io.Copy 或多次 Write 累计输入,用 Sum 读取摘要。把读取错误、消息边界和是否 Reset 处理清楚,就能避免“摘要生成了但只覆盖半段数据”以及“第二份数据混入第一份状态”这两类常见问题。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>