快捷導(dǎo)航

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

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

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

1.函數(shù)定義

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

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

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

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

3.檢查圖片是否存在：

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

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

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

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

4.下載圖片

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

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

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

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

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.完整代碼

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

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ā)生錯(cuò)誤時(shí)，跳過當(dāng)前循環(huán)
	}
	defer resp.Body.Close()

	if resp.StatusCode == http.StatusNotFound {
		log.Printf("圖片 %s 不存在于服務(wù)器\n", imageName)
		return // 圖片不存在時(shí)，跳過當(dāng)前循環(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
	}
}

總結(jié)

總結(jié)來說，這段代碼的主要功能是嘗試從指定的URL下載圖片，并在下載前檢查圖片是否存在。如果存在，則下載并保存到本地；如果不存在，則記錄錯(cuò)誤并跳過下載。同時(shí)，它還處理了可能出現(xiàn)的錯(cuò)誤，并確保在函數(shù)結(jié)束時(shí)正確地關(guān)閉了所有資源。

以上就是go從指定的URL下載圖片并保存到本地的代碼實(shí)現(xiàn)的詳細(xì)內(nèi)容，更多關(guān)于go從指定URL下載圖片并保存的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: