黑窗口下带进度条的http下载

发布时间:2020-06-14 03:07:37 作者:codyguo
来源:网络 阅读:1188
package main

import (
    "flag"
    "fmt"
    "io"
    "log"
    "net/http"
    "os"
    "strconv"
    "strings"
    "time"

    "github.com/cheggaaa/pb"
)

var (
    url = flag.String("url", "", "The download file URL.")
)

func Usage() {
    fmt.Printf("Usage of %s:\n  -url=\"http://www.xxx.com/file.exe\": The download file URL.\n", os.Args[0])
    os.Exit(1)
}

func main() {
    flag.Parse()

    if os.Args == nil || *url == "" {
        Usage()
    }

    // 解析/后的文件名字
    urlMap := strings.Split(*url, "/")
    fileName := urlMap[len(urlMap)-1]

    // 解析带? = fileName 的文件名字
    if strings.Contains(fileName, "=") {
        splitName := strings.Split(fileName, "=")
        fileName = splitName[len(splitName)-1]
    }

    resp, err := http.Get(*url)
    if err != nil {
        log.Fatal(err)
    }

    // 判断get url的状态码, StatusOK = 200
    if resp.StatusCode == http.StatusOK {
        log.Printf("[INFO] 正在下载: [%s]", fileName)
        fmt.Print("\n")

        downFile, err := os.Create(fileName)
        if err != nil {
            log.Fatal(err)
        }
        // 不要忘记关闭打开的文件.
        defer downFile.Close()

        // 获取下载文件的大小
        i, _ := strconv.Atoi(resp.Header.Get("Content-Length"))
        sourceSiz := int64(i)
        source := resp.Body

        // 创建一个进度条
        bar := pb.New(int(sourceSiz)).SetUnits(pb.U_BYTES).SetRefreshRate(time.Millisecond * 10)

        // 显示下载速度
        bar.ShowSpeed = true

        // 显示剩余时间
        bar.ShowTimeLeft = true

        // 显示完成时间
        bar.ShowFinalTime = true

        bar.SetMaxWidth(80)

        bar.Start()

        writer := io.MultiWriter(downFile, bar)
        io.Copy(writer, source)
        bar.Finish()

        fmt.Print("\n")
        log.Printf("[INFO] [%s]下载成功.", fileName)
    } else {
        fmt.Print("\n")
        log.Printf("[ERROR] [%s]下载失败,%s.", fileName, resp.Status)
    }

}


推荐阅读:
  1. Postman发送带cookie的http请求
  2. 下载器-http 进度条 文件读写

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

进度条 http下载 黑窗口

上一篇:set和dict理论和操作

下一篇:日志监控_ElasticStack-0002.Logstash编码插件及实际生产案例应用?

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》