go语言中使用TeeReader计算文件hash值示例代码

Go  /  管理员 发布于 3年前   1214

上一篇我们实现使用TeeReader计算文件下载速度功能，这篇来实现使用TeeReader计算读取速度功能

https://www.zongscan.com/demo333/95754.html

示例代码

package main

import (
	"crypto/sha256"
	"fmt"
	"io"
	"log"
	"net/http"
	"os"
)

func main() {
	out, err := os.Create("test.tmp")
	if err != nil {
		log.Fatal(err)
	}
	defer out.Close()
	resp, err := http.Get("https://golang.google.cn/dl/go1.18.2.src.tar.gz")

	if err != nil {
		log.Fatal(err)
	}

	defer resp.Body.Close()
	h := sha256.New()
	tee := io.TeeReader(resp.Body, h)
	io.Copy(out, tee)
	fmt.Printf("%x", h.Sum(nil))
}

输出结果：

2c44d03ea2c34092137ab919ba602f2c261a038d08eb468528a3f3a28e5667e2

总结

对比TeeReader和io.ReadAll()读取文件后再sha256占用的内存来看

TeeReader占用的内存更少。