perf: 一点小优化

LagrangeDev · Aug 27, 2024 · 7bc594c · 7bc594c
1 parent 47cb0ec
commit 7bc594c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/utils/crypto/hash.go b/utils/crypto/hash.go
@@ -62,11 +62,10 @@ func ComputeMd5AndSha1AndLength(r io.ReadSeeker) ([]byte, []byte, uint64) {
 	defer func(r io.ReadSeeker, offset int64, whence int) {
 		_, _ = r.Seek(offset, whence)
 	}(r, 0, io.SeekStart)
+	_, _ = r.Seek(0, io.SeekStart)
 	md5r := md5.New()
 	sha1r := sha1.New()
-	_, _ = r.Seek(0, io.SeekStart)
-	length, _ := io.Copy(md5r, r)
-	_, _ = r.Seek(0, io.SeekStart)
-	_, _ = io.Copy(sha1r, r)
+	mr := io.MultiWriter(md5r, sha1r)
+	length, _ := io.Copy(mr, r)
 	return md5r.Sum(nil), sha1r.Sum(nil), uint64(length)
 }