欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Go语言下载网络图片或文件的方法示例

程序员文章站 2023-10-11 14:07:38
最近闲来无事, 于是就简单学习了下go语言的基本的用法。由于实践才是最快的学习方法,所以这里就以下载网络图片或文件入手来学习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)
}

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。