Go hash.Hash对同一输入计算多种摘要的组合方法

需要对同一份输入同时计算 SHA-256、SHA-512 等多个摘要时，不要让一个 hash.Hash 在算法之间来回切换。更稳妥的组合方式是：每个算法创建一个独立实例，再用 io.MultiWriter 把同一批字节分发给它们，最后分别调用 Sum(nil) 取结果。

官方文档：https://pkg.go.dev/hash

用 io.MultiWriter 把同一份输入写入多个 hash.Hash

hash.Hash 本质上是一个带有 Write、Sum、Reset、Size 和 BlockSize 的接口。组合多个摘要时，先把每个实现放入独立变量，再把它们转换成 []io.Writer 交给 io.MultiWriter。这样输入只需要写一次，状态仍然分开。

package main

import (
    "crypto/sha256"
    "crypto/sha512"
    "encoding/hex"
    "fmt"
    "hash"
    "io"
)

type digestSpec struct {
    name string
    hash hash.Hash
}

func main() {
    input := []byte("order-2026-09") // 同一份输入由所有摘要实例共同消费
    sha256Hash := sha256.New()
    sha512Hash := sha512.New()
    sha384Hash := sha512.New384()

    writers := []io.Writer{sha256Hash, sha512Hash, sha384Hash}
    fanout := io.MultiWriter(writers...)
    if _, err := fanout.Write(input); err != nil {
        // MultiWriter 遇到下游错误会停止，生产代码应保留错误上下文
        panic(fmt.Errorf("write digest input: %w", err))
    }

    results := []digestSpec{
        {name: "sha256", hash: sha256Hash},
        {name: "sha512", hash: sha512Hash},
        {name: "sha384", hash: sha384Hash},
    }
    for _, item := range results {
        // Sum(nil) 取当前摘要；它不会自动清空 item.hash 的内部状态
        fmt.Printf("%s=%s\\n", item.name, hex.EncodeToString(item.hash.Sum(nil)))
    }
}

这里的关键不是把三种算法“合并成一个摘要”，而是复用同一个输入分发点。每个 hash.Hash 都从空状态开始，并独立保存自己的中间状态和输出长度。图中的连线表示静态关系，不表示一次真实执行截图。

结果命名、Sum 参数和复用边界

多个摘要最终都会变成 []byte，如果只保留字节切片，调用方很容易忘记它对应哪一种算法。可以用名称和实例组成轻量的结果描述，统一在循环中读取，必要时再用十六进制编码成可记录的字符串。

因此，组合计算通常应先收集所有结果，再决定是否复用实例。若确实要复用，可把清理动作放在消费完摘要之后：

func reuseHash(h hash.Hash, next []byte) []byte {
    // 调用方必须确认旧摘要已经消费完，再清空状态
    h.Reset()
    if _, err := h.Write(next); err != nil {
        // 保留错误返回，避免把不完整摘要当成成功结果
        return nil
    }
    return h.Sum(nil)
}

实际项目里不要把带状态的实例放进多个并发请求共享的全局变量；Reset 只能解决顺序复用，不能替代互斥、对象池借还协议或每次创建新实例。

摘要长度和组合实现的检查清单

Size() 适合在输出层做长度检查。例如 SHA-256 返回 32 字节，SHA-384 返回 48 字节，SHA-512 返回 64 字节；这些长度来自具体实现，不应由通用接口猜测。需要落库或传输时，先约定二进制还是十六进制文本格式，避免同一个字段一会儿存原始字节、一会儿存编码字符串。

• 输入是否只写入一次分发器，且每个摘要实例只绑定一个算法？
• 结果是否带有稳定的算法名称，并在编码前后保持长度约定？
• 是否在读取完所有 Sum 结果后再 Reset，并避免跨协程共享可变状态？
• 是否把“通用 hash.Hash 接口”与“具体算法的安全用途”分开评估？

相关问题

调用 Sum(nil) 后还需要 Reset 吗？

如果实例不再使用，可以直接丢弃；如果要顺序复用，仍需在消费完当前结果后显式调用 Reset。

能否把同一个 hash.Hash 同时交给多个 goroutine？

不要默认这样做。接口本身没有给出并发安全保证；并发任务应各自创建实例，或由调用方提供清晰的同步和生命周期管理。

为什么不直接调用每个算法的 Sum 函数？

一次性、短输入可以直接调用便捷函数；当输入来自流或需要统一管理算法列表时，独立 hash.Hash 加 io.MultiWriter 更容易复用同一写入路径。