go從指定的URL下載圖片并保存到本地的代碼實現(xiàn)

更新時間：2024年08月28日 09:18:03 作者：ZL_567

這段代碼定義了一個名為 downloadImage 的函數(shù),其目的是從指定的URL下載圖片并保存到本地文件系統(tǒng),本文是對代碼功能的詳細描述,對大家的學習或工作有一定的幫助,需要的朋友可以參考下

1.函數(shù)定義

該函數(shù)接受一個 imageName 參數(shù)，圖片的文件名。

2.構(gòu)建圖片URL

使用 fmt.Sprintf 函數(shù)構(gòu)建圖片的URL，這個URL指向一個特定的網(wǎng)絡路徑，用于下載圖片。

url := fmt.Sprintf("https://xxxxxxx.com.cn/pic/l/sevp_nsmc_wxbl_fy4b_etcc_achn_lno_py_%s", imageName)

3.檢查圖片是否存在：

發(fā)起一個HEAD請求到構(gòu)建的URL，檢查圖片是否存在。

如果圖片不存在（HTTP 404），記錄一條日志并返回，不進行下載操作。

resp, err := http.Head(url)
if err != nil {
	log.Println(err)
	return // 發(fā)生錯誤時，跳過當前循環(huán)
}
defer resp.Body.Close()

if resp.StatusCode == http.StatusNotFound {
	log.Printf("圖片 %s 不存在于服務器\n", imageName)
	return // 圖片不存在時，跳過當前循環(huán)
}

4.下載圖片

如果圖片存在，發(fā)起一個GET請求，從URL下載圖片。

response, err := http.Get(url)
if err != nil {
	log.Println(err)
	return
}

5.創(chuàng)建本地文件

根據(jù)圖片名稱和預設的路徑創(chuàng)建一個本地文件夾，用于保存下載的圖片。

filepath := fmt.Sprintf("./uploads/image/%s", imageName)
file, err := os.Create(filepath)
if err != nil {
	log.Println(err)
	return
}

6.保存圖片到本地

下載的圖片內(nèi)容寫入到本地文件中。

_, err = io.Copy(file, response.Body)
if err != nil {
	log.Println(err)
	return
}

7.完整代碼

在整個過程中，對任何可能發(fā)生的錯誤進行捕獲和記錄。
使用 defer 語句確保在函數(shù)返回之前關(guān)閉打開的文件和網(wǎng)絡連接，釋放資源。

func downloadImage(imageName string) {
	url := fmt.Sprintf("https://xxxxxxx.com.cn/pic/l/sevp_nsmc_wxbl_fy4b_etcc_achn_lno_py_%s", imageName)
	// 如果不存在，檢查 HTTP 狀態(tài)碼
	resp, err := http.Head(url)
	if err != nil {
		log.Println(err)
		return // 發(fā)生錯誤時，跳過當前循環(huán)
	}
	defer resp.Body.Close()

	if resp.StatusCode == http.StatusNotFound {
		log.Printf("圖片 %s 不存在于服務器\n", imageName)
		return // 圖片不存在時，跳過當前循環(huán)
	}
	response, err := http.Get(url)
	if err != nil {
		log.Println(err)
		return
	}
	defer response.Body.Close()

	filepath := fmt.Sprintf("./uploads/image/%s", imageName)
	file, err := os.Create(filepath)
	if err != nil {
		log.Println(err)
		return
	}
	defer file.Close()

	_, err = io.Copy(file, response.Body)
	if err != nil {
		log.Println(err)
		return
	}
}