fix for #341 -- only test for qq, I don't have chance to test other websites.

config/config.go

@@ -25,6 +25,8 @@ var (
 	OutputName string
 	// ExtractedData print extracted data
 	ExtractedData bool
+	// ChunkSize size of MB
+	ChunkSizeMB int
 	// UseAria2RPC Use Aria2 RPC to download
 	UseAria2RPC bool
 	// Aria2Token Aria2 RPC Token

downloader/downloader.go

@@ -67,7 +67,7 @@ func writeFile(
 
 // Save save url file
 func Save(
-	urlData URL, refer, fileName string, bar *pb.ProgressBar,
+	urlData URL, refer, fileName string, bar *pb.ProgressBar, chunkSizeMB int,
 ) error {
 	var err error
 	filePath, err := utils.FilePath(fileName, urlData.Ext, false)
@@ -111,9 +111,9 @@ func Save(
 	if fileError != nil {
 		return fileError
 	}
-	if strings.Contains(urlData.URL, "googlevideo") {
+	if strings.Contains(urlData.URL, "googlevideo") || strings.Contains(urlData.URL, "qq.com") {
 		var start, end, chunkSize int64
-		chunkSize = 10 * 1024 * 1024
+		chunkSize = int64(chunkSizeMB) * 1024 * 1024
 		remainingSize := urlData.Size
 		if tempFileSize > 0 {
 			start = tempFileSize
@@ -169,7 +169,7 @@ func Save(
 }
 
 // Download download urls
-func Download(v Data, refer string) error {
+func Download(v Data, refer string, chunkSizeMB int) error {
 	v.genSortedStreams()
 	if config.ExtractedData {
 		jsonData, _ := json.MarshalIndent(v, "", "    ")
@@ -253,7 +253,7 @@ func Download(v Data, refer string) error {
 	bar.Start()
 	if len(data.URLs) == 1 {
 		// only one fragment
-		err := Save(data.URLs[0], refer, title, bar)
+		err := Save(data.URLs[0], refer, title, bar, chunkSizeMB)
 		if err != nil {
 			return err
 		}
@@ -275,7 +275,7 @@ func Download(v Data, refer string) error {
 		wgp.Add()
 		go func(url URL, refer, fileName string, bar *pb.ProgressBar) {
 			defer wgp.Done()
-			err := Save(url, refer, fileName, bar)
+			err := Save(url, refer, fileName, bar, chunkSizeMB)
 			if err != nil {
 				errs = append(errs, err)
 			}

downloader/downloader_test.go

@@ -92,7 +92,7 @@ func TestDownload(t *testing.T) {
 		},
 	}
 	for _, testCase := range testCases {
-		err := Download(testCase.data, "")
+		err := Download(testCase.data, "", 10)
 		if err != nil {
 			t.Error(err)
 		}

main.go

@@ -49,6 +49,7 @@ func init() {
 	flag.StringVar(&config.OutputPath, "o", "", "Specify the output path")
 	flag.StringVar(&config.OutputName, "O", "", "Specify the output file name")
 	flag.BoolVar(&config.ExtractedData, "j", false, "Print extracted data")
+	flag.IntVar(&config.ChunkSizeMB, "cs", 5, "Print chunkSize size of MB")
 	flag.BoolVar(&config.UseAria2RPC, "aria2", false, "Use Aria2 RPC to download")
 	flag.StringVar(&config.Aria2Token, "aria2token", "", "Aria2 RPC Token")
 	flag.StringVar(&config.Aria2Addr, "aria2addr", "localhost:6800", "Aria2 Address")
@@ -84,7 +85,7 @@ func printError(url string, err error) {
 	)
 }
 
-func download(videoURL string) {
+func download(videoURL string, chuckSizeMB int) {
 	var (
 		domain string
 		err    error
@@ -160,7 +161,7 @@ func download(videoURL string) {
 			printError(item.URL, item.Err)
 			continue
 		}
-		err = downloader.Download(item, videoURL)
+		err = downloader.Download(item, videoURL, chuckSizeMB)
 		if err != nil {
 			printError(item.URL, err)
 		}
@@ -214,6 +215,6 @@ func main() {
 		}
 	}
 	for _, videoURL := range args {
-		download(strings.TrimSpace(videoURL))
+		download(strings.TrimSpace(videoURL), config.ChunkSizeMB)
 	}
 }