golang批量下載文件

package main

import (
    "bufio"
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "os"
    "path/filepath"
    "strings"
    "sync"
)

var (
    downloadDestFolder = "/Users/zhangsheng/Go/WorkPlace/src/xcrm"
    urlFilePath        = "/Users/zhangsheng/Go/WorkPlace/src/xcrm/file.txt"
)

func init() {
    log.SetFlags(log.Lshortfile)
    _ = os.MkdirAll(downloadDestFolder, 0777)
}

func main() {
    fi, err := os.Open(urlFilePath)
    if err != nil {
        fmt.Printf("Error: %s\n", err)
        return
    }
    defer fi.Close()
    br := bufio.NewReader(fi)
    var w sync.WaitGroup
    for {
        line, _, err := br.ReadLine()
        if err != nil {
            log.Println("read url complete")
            break
        }
        list := strings.Split(string(line), ",")
        w.Add(1)
        go download(list[1], list[0]+".xlsx", &w)
    }
    w.Wait()

}

func download(url string, filename string, w *sync.WaitGroup) {
    res, err := http.Get(url)
    if err != nil {
        log.Printf("http.Get -> %v", err)
        return
    }
    data, err := ioutil.ReadAll(res.Body)
    if err != nil {
        log.Printf("ioutil.ReadAll -> %s", err.Error())
        return
    }
    defer res.Body.Close()
    if err = ioutil.WriteFile(downloadDestFolder+string(filepath.Separator)+filename, data, 0777); err != nil {
        log.Println("Error Saving:", filename, err)
    } else {
        log.Println("Saved:", filename)
    }
    w.Done()
}

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容