golang 进度下载文件

小塔罗 · · 2433 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

//一个没找到工作的下午独自撸起来的代码
//很不美观临时作品
package main

import (
   "net/http"
   "fmt"
   "os"
   "errors"
   "io"
   "strconv"
)

func main() {

   downloadFile("http://localhost:8089/ThunderSpeed1.0.35.366.exe", func(length, downLen int64) {
      fmt.Println(length, downLen, float32(downLen)/float32(length))
   })

}

func downloadFile(url string, fb func(length, downLen int64)) error {
   var (
      fsize   int64
      buf     = make([]byte, 32*1024)
      written int64
   )
   //创建一个http client
   client := new(http.Client)
   //get方法获取资源
   resp, err := client.Get(url)
   if err != nil {
      return err
   }
   //读取服务器返回的文件大小
   fsize, err = strconv.ParseInt(resp.Header.Get("Content-Length"), 10, 32)
   if err != nil {
      fmt.Println(err)
   }
   //创建文件
   file, err := os.Create("test.exe")
   if err != nil {
      return err
   }
   defer file.Close()
   if resp.Body == nil {
      return errors.New("body is null")
   }
   defer resp.Body.Close()
   //下面是 io.copyBuffer() 的简化版本
   for {
      //读取bytes
      nr, er := resp.Body.Read(buf)
      if nr > 0 {
         //写入bytes
         nw, ew := file.Write(buf[0:nr])
         //数据长度大于0
         if nw > 0 {
            written += int64(nw)
         }
         //写入出错
         if ew != nil {
            err = ew
            break
         }
         //读取是数据长度不等于写入的数据长度
         if nr != nw {
            err = io.ErrShortWrite
            break
         }
      }
      if er != nil {
         if er != io.EOF {
            err = er
         }
         break
      }
      //没有错误了快使用 callback

      fb(fsize, written)
   }
   return err
}

有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:小塔罗

查看原文:golang 进度下载文件

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

2433 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传