Go語言下載網絡圖片或文件

最近閑來無事， 於是就簡單學習了下Go語言的基本的用法。由於實踐才是最快的學習方法，所以這里就以下載網絡圖片或文件入手來學習Go語言

文件下載到本地，通常的思路就是先獲得網絡文件的 輸入流 以及本地文件的 輸出流 ，再讀取輸入流到輸出流中，因此自然也要獲取相應的Reader和Writer。

下面首先是使用GoLang的http.Get()方法(使用client的do方法原理類似)

	// 圖片以花瓣網的圖片為例
	imgUrl := "http://hbimg.b0.upaiyun.com/32f065b3afb3fb36b75a5cbc90051b1050e1e6b6e199-Ml6q9F_fw320"

	res, err := http.Get(imgUrl)
	if err != nil {
		fmt.Println("A error occurred!")
		return
	}
	// defer后的為延時操作，通常用來釋放相關變量
	defer res.Body.Close()

然后獲得get請求響應的reader對象

	// 獲得get請求響應的reader對象
	reader := bufio.NewReaderSize(res.Body, 32 * 1024)

上面獲取到了輸入流的reader對象， 下面來獲取本地文件的輸出流的writer對象

	imgPath := "C:\\Users\\Asche\\go\\src\\GoSpiderTest\\"
	// 根據圖片url獲取其文件名
	fileName := path.Base(imgUrl)


        file, err := os.Create(imgPath + fileName)
	if err != nil {
		panic(err)
	}
	// 獲得文件的writer對象
	writer := bufio.NewWriter(file)

好了， reader和writer對象都得到了，然后在讀取寫入。
如果嫌麻煩的話，可以直接copy：

	written, _ := io.Copy(writer, reader)
	// 輸出文件字節大小
	fmt.Printf("Total length: %d", written)

或則手動讀取和寫入

	bytes := make([]byte, 32 * 1024)
	for {
		len, err := reader.Read(bytes)

		if len < 0 || err != nil{
			return
		}
		// 注意這里byte數組后的[0:len]，不然可能會導致寫入多余的數據
		_, _ = writer.Write(bytes[0:len])
		fmt.Printf("%d ", len)
	}

ok，下載完成。

下面貼上完整的代碼(讀取寫入采用的方法一)

package main

import (
	"bufio"
	"fmt"
	"io"
	"net/http"
	"os"
	"path"
)

func main() {
	imgPath := "C:\\Users\\Asche\\go\\src\\GoSpiderTest\\"
	imgUrl := "http://hbimg.b0.upaiyun.com/32f065b3afb3fb36b75a5cbc90051b1050e1e6b6e199-Ml6q9F_fw320"

	fileName := path.Base(imgUrl)


	res, err := http.Get(imgUrl)
	if err != nil {
		fmt.Println("A error occurred!")
		return
	}
	defer res.Body.Close()
	// 獲得get請求響應的reader對象
	reader := bufio.NewReaderSize(res.Body, 32 * 1024)


	file, err := os.Create(imgPath + fileName)
	if err != nil {
		panic(err)
	}
	// 獲得文件的writer對象
	writer := bufio.NewWriter(file)

	written, _ := io.Copy(writer, reader)
	fmt.Printf("Total length: %d", written)
}