Golang實現(xiàn)Md5校驗的代碼示例

更新時間：2024年07月28日 14:09:46 作者：貓頭鷹數(shù)據(jù)分析

最近項目中有個需求,就是地圖文件下發(fā)后,接收方需要文件的md5值,和接收到的文件做比對,以免文件不完整,引起bug,于是測試了下本地文件和遠程文件的md5計算,所以本文給大家介紹了Golang實現(xiàn)Md5校驗,需要的朋友可以參考下

前言

最近項目中有個需求，就是地圖文件下發(fā)后，接收方需要文件的md5值，和接收到的文件做比對，以免文件不完整，引起bug，于是測試了下本地文件和遠程文件的md5計算。

1、本地文件

要獲取指定本地文件的MD5值，你可以使用crypto/md5包來計算文件的MD5散列值。以下是一個示例代碼，演示了如何打開一個文件并計算其MD5值：

package main
 
import (
	"crypto/md5"
	"encoding/hex"
	"fmt"
	"io"
	"net/http"
	"os"
)
 
func main() {
	Md5ByLocalFile()
}
 
func Md5ByLocalFile() {
	// // 指定文件路徑
	filePath := "d:/code/000fa28f-c114-49fe-9699-8c7f8b2eb222.png"
	// filePath := "https://minio.dev.inrobot.cloud/map/000fa28f-c114-49fe-9699-8c7f8b2eb222.png"
	// 打開文件
	file, err := os.Open(filePath)
	if err != nil {
		fmt.Println("Failed to open file:", err)
		return
	}
	defer file.Close()
 
	// 創(chuàng)建一個md5哈希對象
	hasher := md5.New()
 
	// 將文件內(nèi)容讀入哈希對象
	if _, err := io.Copy(hasher, file); err != nil {
		fmt.Println("Failed to read file:", err)
		return
	}
 
	// 計算并獲取哈希值
	hashBytes := hasher.Sum(nil)
 
	// 將哈希值轉換為十六進制字符串
	hashString := hex.EncodeToString(hashBytes)
 
	// 輸出MD5哈希值
	fmt.Println("MD5 hash of the file:", hashString)
}

輸出結果如下：

PS D:\gostudy2022\TimeScheduler\md5> go run .\main.go
LocalFile MD5 hash of the file: b4735b024f3552b1277671303149719b

2、遠程文件

遠程文件其實就是網(wǎng)絡中可訪問的資源文件，要獲取指定網(wǎng)絡地址的文件的MD5值，你需要先下載文件的內(nèi)容到內(nèi)存或磁盤上，然后再計算其MD5值。這里我展示一個示例，該示例使用net/http包來下載文件，并使用crypto/md5包來計算MD5值。以下是完整的示例代碼：

package main
 
import (
	"crypto/md5"
	"encoding/hex"
	"fmt"
	"io"
	"net/http"
	"os"
)
 
func main() {
	Md5ByRemoteFile()
}
 
func Md5ByRemoteFile() {
	// 指定文件的URL
	url := "https://minio.dev.inrobot.cloud/map/000fa28f-c114-49fe-9699-8c7f8b2eb222.png"
 
	// 發(fā)起HTTP GET請求
	resp, err := http.Get(url)
	if err != nil {
		fmt.Println("Failed to fetch file:", err)
		return
	}
	defer resp.Body.Close()
 
	// 檢查響應狀態(tài)碼
	if resp.StatusCode != http.StatusOK {
		fmt.Printf("Failed to fetch file: HTTP status code %d\n", resp.StatusCode)
		return
	}
 
	// 創(chuàng)建一個md5哈希對象
	hasher := md5.New()
 
	// 將文件內(nèi)容讀入哈希對象
	if _, err := io.Copy(hasher, resp.Body); err != nil {
		fmt.Println("Failed to read file:", err)
		return
	}
 
	// 計算并獲取哈希值
	hashBytes := hasher.Sum(nil)
 
	// 將哈希值轉換為十六進制字符串
	hashString := hex.EncodeToString(hashBytes)
 
	// 輸出MD5哈希值
	fmt.Println("MD5 hash of the file:", hashString)
}

這里遠程文件和上例的文件相同，輸出結果如下：

PS D:\gostudy2022\TimeScheduler\md5> go run .\main.go
RemoteFile MD5 hash of the file: b4735b024f3552b1277671303149719b

可以看到，本地文件和遠程文件的md5值是相同的，從而在項目中，我們可以根據(jù)md5值是否相同，判斷文件是不是同一個文件，有沒有被損壞或篡改。

拓展：Golang 計算MD5值

代碼

//MD5 
func Md5SmallFile(path string) (string, error) {
   file, err := os.Open(path)
   if err != nil {
      return "", err
   }
   defer file.Close()

   h := md5.New()
   _, err = io.Copy(h, file)
   if err != nil {
      return "", err
   }

   return hex.EncodeToString(h.Sum(nil)), nil
}


func Md5BigFile(path string) (string, error) {
   var fileChunk uint64 = 10485760
   file, err := os.Open(path)
   if err != nil {
      return "", err
   }
   defer file.Close()
   // calculate the file size
   info, _ := file.Stat()
   fileSize := info.Size()
   blocks := uint64(math.Ceil(float64(fileSize) / float64(fileChunk)))
   h := md5.New()

   for i := uint64(0); i < blocks; i++ {
      blockSize := int(math.Min(float64(fileChunk), float64(fileSize-int64(i*fileChunk))))
      buf := make([]byte, blockSize)

      _, err := file.Read(buf)
      if err != nil {
         return "", err
      }
      _, err = io.WriteString(h, string(buf)) // append into the hash
      if err != nil {
         return "", err
      }
   }
   
   return hex.EncodeToString(h.Sum(nil)), nil
}


func MD5String(v string) string {
	d := []byte(v)
	m := md5.New()
	m.Write(d)
	return hex.EncodeToString(m.Sum(nil))
}

md5.New()返回一個 hash.Hash 對象

hash.Hash對象定義了一個Sum接口，其原型為：