快捷導(dǎo)航

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

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

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

前言

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

1、本地文件

要獲取指定本地文件的MD5值，你可以使用crypto/md5包來計(jì)算文件的MD5散列值。以下是一個(gè)示例代碼，演示了如何打開一個(gè)文件并計(jì)算其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)建一個(gè)md5哈希對(duì)象
	hasher := md5.New()
 
	// 將文件內(nèi)容讀入哈希對(duì)象
	if _, err := io.Copy(hasher, file); err != nil {
		fmt.Println("Failed to read file:", err)
		return
	}
 
	// 計(jì)算并獲取哈希值
	hashBytes := hasher.Sum(nil)
 
	// 將哈希值轉(zhuǎn)換為十六進(jìn)制字符串
	hashString := hex.EncodeToString(hashBytes)
 
	// 輸出MD5哈希值
	fmt.Println("MD5 hash of the file:", hashString)
}

輸出結(jié)果如下：

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

2、遠(yuǎn)程文件

遠(yuǎn)程文件其實(shí)就是網(wǎng)絡(luò)中可訪問的資源文件，要獲取指定網(wǎng)絡(luò)地址的文件的MD5值，你需要先下載文件的內(nèi)容到內(nèi)存或磁盤上，然后再計(jì)算其MD5值。這里我展示一個(gè)示例，該示例使用net/http包來下載文件，并使用crypto/md5包來計(jì)算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請(qǐng)求
	resp, err := http.Get(url)
	if err != nil {
		fmt.Println("Failed to fetch file:", err)
		return
	}
	defer resp.Body.Close()
 
	// 檢查響應(yīng)狀態(tài)碼
	if resp.StatusCode != http.StatusOK {
		fmt.Printf("Failed to fetch file: HTTP status code %d\n", resp.StatusCode)
		return
	}
 
	// 創(chuàng)建一個(gè)md5哈希對(duì)象
	hasher := md5.New()
 
	// 將文件內(nèi)容讀入哈希對(duì)象
	if _, err := io.Copy(hasher, resp.Body); err != nil {
		fmt.Println("Failed to read file:", err)
		return
	}
 
	// 計(jì)算并獲取哈希值
	hashBytes := hasher.Sum(nil)
 
	// 將哈希值轉(zhuǎn)換為十六進(jìn)制字符串
	hashString := hex.EncodeToString(hashBytes)
 
	// 輸出MD5哈希值
	fmt.Println("MD5 hash of the file:", hashString)
}

這里遠(yuǎn)程文件和上例的文件相同，輸出結(jié)果如下：

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

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

拓展：Golang 計(jì)算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()返回一個(gè) hash.Hash 對(duì)象

hash.Hash對(duì)象定義了一個(gè)Sum接口，其原型為：