1. 程式人生 > >Go語言下載網路圖片或檔案

Go語言下載網路圖片或檔案

最近閒來無事, 於是就簡單學習了下Go語言的基本的用法。由於實踐才是最快的學習方法,所以這裡就以下載網路圖片或檔案入手來學習Go語言

檔案下載到本地,通常的思路就是先獲得網路檔案的 輸入流 以及本地檔案的 輸出流 ,再讀取輸入流到輸出流中,因此自然也要獲取相應的ReaderWriter

下面首先是使用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)
}