Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
400aab0964 | |||
fc046da8a6 | |||
e37ec06287 |
28
download.go
28
download.go
@@ -6,6 +6,7 @@ import (
|
||||
"log"
|
||||
"os"
|
||||
"regexp"
|
||||
"strings"
|
||||
"sync"
|
||||
//"path/filepath"
|
||||
|
||||
@@ -39,20 +40,25 @@ func (b *Batch) Start() {
|
||||
|
||||
for scanner.Scan() {
|
||||
//Call download
|
||||
log.Println(scanner.Text())
|
||||
//add go routine to wait for
|
||||
if strings.TrimSpace(scanner.Text()) != "" {
|
||||
wg.Add(1)
|
||||
go downloadVideo(b.destDir, scanner.Text(), wg)
|
||||
log.Println(scanner.Text())
|
||||
go downloadVideo(b.destDir, strings.TrimSpace(scanner.Text()), &wg)
|
||||
} else {
|
||||
log.Println("Ignoring blank line")
|
||||
}
|
||||
}
|
||||
|
||||
//wait for all go routines to finish
|
||||
wg.Wait()
|
||||
log.Println("All downloads finished")
|
||||
if err := scanner.Err(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func downloadVideo(destDir, url string, wg sync.WaitGroup) {
|
||||
func downloadVideo(destDir, url string, wg *sync.WaitGroup) {
|
||||
// NewYoutube(debug) if debug parameter will set true we can log of messages
|
||||
//defer Done() so that WaitGroup knows when the routine finishes
|
||||
defer wg.Done()
|
||||
@@ -63,6 +69,20 @@ func downloadVideo(destDir, url string, wg sync.WaitGroup) {
|
||||
filename := fmt.Sprintf("%s/%s.%s", destDir, y.StreamList[0]["title"], fileext)
|
||||
|
||||
log.Println("Start downloading:", filename)
|
||||
y.StartDownload(filename)
|
||||
go func() {
|
||||
var i int64 = 0
|
||||
for i < 100 {
|
||||
i = <-y.DownloadPercent
|
||||
if i%10 == 0 {
|
||||
log.Println(i, "%", y.StreamList[0]["title"])
|
||||
}
|
||||
}
|
||||
}()
|
||||
err := y.StartDownload(filename)
|
||||
if err != nil {
|
||||
log.Println("Failed to download", url)
|
||||
log.Println("Message:", err.Error())
|
||||
} else {
|
||||
log.Println("Finished downloading:", filename)
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user