快捷導(dǎo)航

Go語言下載網(wǎng)絡(luò)圖片或文件的方法示例

更新時間：2018年12月23日 16:45:04 作者：Asche

這篇文章主要介紹了Go語言下載網(wǎng)絡(luò)圖片或文件的方法示例，文中通過示例代碼介紹的非常詳細，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

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

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

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

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

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

然后獲得get請求響應(yīng)的reader對象

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

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

imgPath := "C:\\Users\\Asche\\go\\src\\GoSpiderTest\\"
  // 根據(jù)圖片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)
  // 輸出文件字節(jié)大小
  fmt.Printf("Total length: %d", written)

或則手動讀取和寫入

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

    if len < 0 || err != nil{
      return
    }
    // 注意這里byte數(shù)組后的[0:len]，不然可能會導(dǎo)致寫入多余的數(shù)據(jù)
    _, _ = 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請求響應(yīng)的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)
}

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: