xiaoqidun/ihash
1
0
Fork 0
mirror of https://github.com/xiaoqidun/ihash.git synced 2025-04-21 18:01:24 +08:00
Code Issues Projects Releases Wiki Activity

feat(校验速度): 去除进度条加快校验速度

Browse Source
This commit is contained in:
xiaoqidun
2020-10-21 17:20:35 +08:00
parent 7c636ded82
commit 9192cd340f
291 changed files with 896 additions and 211184 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
ihash.go
View File

@ -10,6 +10,7 @@ import (
"fmt"
"golang.org/x/crypto/sha3"
"hash"
"io"
"io/ioutil"
"os"
"path/filepath"
@ -111,8 +112,7 @@ func StrHash(hashType string, hashData []byte) string {
}
func FileHash(hashType string, hashFile string) {
msg := "progress:%.2f%% filename:%s\r"
success := "%x %s\n"
success := "%x %s"
fileOpen, err := os.Open(hashFile)
if err != nil {
return
@ -124,8 +124,6 @@ func FileHash(hashType string, hashFile string) {
if err != nil || fileStat.IsDir() {
return
}
fileSize := fileStat.Size()
readSize := 0
readBytes := make([]byte, 8192)
var hashHandle interface{}
switch hashType {
@ -146,32 +144,17 @@ func FileHash(hashType string, hashFile string) {
case "sha3sum512":
hashHandle = sha3.New512()
}
printLength := 0
for {
n, err := fileOpen.Read(readBytes)
if err != nil {
if err == io.EOF && n > 0 {
hashHandle.(hash.Hash).Write(readBytes[:n])
}
break
}
readSize += n
hashHandle.(hash.Hash).Write(readBytes[:n])
if 0 == readSize%67108864 {
printMsg := fmt.Sprintf(msg, float64(readSize)/float64(fileSize)*100, fileStat.Name())
printMsgLength := len(printMsg)
if printLength > printMsgLength {
printMsg += strings.Repeat(" ", printLength-printMsgLength)
}
fmt.Print(printMsg)
_ = os.Stdout.Sync()
printLength = len(printMsg)
}
}
printMsg := fmt.Sprintf(success, hashHandle.(hash.Hash).Sum(nil), hashFile)
printMsgLength := len(printMsg)
if printLength > printMsgLength {
printMsg += strings.Repeat(" ", printLength-printMsgLength)
}
fmt.Print(printMsg)
_ = os.Stdout.Sync()
fmt.Println(fmt.Sprintf(success, hashHandle.(hash.Hash).Sum(nil), hashFile))
}
func GetFileName() string {