我正在与接受范围和Goroutines并行下载一个大的.zip文件。该应用程序发送了多个请求,以使用其Range标头从URL下载zip文件的10MB块。
将请求分成单独的Goroutines分成不同的范围,并将获得的数据写入临时文件。这些文件被命名为1、2、3 ...
package main
import (
"bufio"
"fmt"
"io"
"io/ioutil"
"log"
"net/http"
"os"
"strconv"
"sync"
)
var wg sync.WaitGroup
func main() {
url := "https://path/to/large/zip/file/zipfile.zip"
res, _ := http.Head(url)
maps := res.Header
length, _ := strconv.Atoi(maps["Content-Length"][0]) // Get the content length from the header request
chunks := (length / (1024 * 1024 * 10)) + 1
// startByte and endByte determines the positions of the chunk that should be downloaded
var startByte = 0
var endByte = (1024 * 1024 * 10) - 1
//body := make([][]byte, chunks)
body := make([]io.ReadCloser, chunks)
for i := 0; i < chunks; i++ {
wg.Add(1)
go func(min int, max int, i int) {
client := &http.Client {}
req, _ := http.NewRequest("GET", url, nil)
rangeHeader := "bytes=" + strconv.Itoa(min) +"-" + strconv.Itoa(max)
fmt.Println(rangeHeader)
req.Header.Add("Range", rangeHeader)
resp,_ := client.Do(req)
defer resp.Body.Close()
reader, _ := ioutil.ReadAll(resp.Body)
body[i] = resp.Body
ioutil.WriteFile(strconv.Itoa(i), reader, 777) // Write to the file i as a byte array
wg.Done()
}(startByte, endByte, i)
startByte = endByte + 1
endByte += 1024 * 1024 * 10
}
wg.Wait()
filepath := "zipfile.zip"
// Create the file
_, err := os.Create(filepath)
if err != nil {
return
}
file, _ := os.OpenFile(filepath, os.O_APPEND|os.O_WRONLY, os.ModeAppend)
if err != nil {
log.Fatal(err)
}
for j := 0; j < chunks; j++ {
newFileChunk, err := os.Open(strconv.Itoa(j))
if err != nil {
log.Fatal(err)
}
defer newFileChunk.Close()
chunkInfo, err := newFileChunk.Stat()
if err != nil {
log.Fatal(err)
}
var chunkSize int64 = chunkInfo.Size()
chunkBufferBytes := make([]byte, chunkSize)
// read into chunkBufferBytes
reader := bufio.NewReader(newFileChunk)
_, err = reader.Read(chunkBufferBytes)
file.Write(chunkBufferBytes)
file.Sync() //flush to disk
chunkBufferBytes = nil // reset or empty our buffer
}
//Verify file size
filestats, err := file.Stat()
if err != nil {
log.Fatal(err)
return
}
actualFilesize := filestats.Size()
if actualFilesize != int64(length) {
log.Fatal("Actual Size: ", actualFilesize, " Expected: ", length)
return
}
file.Close()
}
下载所有文件后,我尝试将它们重新组合为一个.zip文件。但是,将文件放在一起后,由于似乎已损坏,我无法解压缩最终文件。
我想知道我在做什么错,或者是否有更好的方法来解决这个问题。预先感谢。
编辑:以下是记录到控制台的内容
bytes=0-10485759
bytes=10485760-20971519
2018/12/04 11:21:28 Actual Size: 16877828 Expected: 16877827
答案 0 :(得分:4)
问题出在您的范围请求中
线条
resp,_ := client.Do(req)
defer resp.Body.Close()
由go vet
报告,因为未检查错误。如果您在最后一个区块中检查响应代码,它是416-使用的范围不正确,请对此进行更改
resp, err := client.Do(req)
if err != nil {
panic(err)
}
if resp.StatusCode == 416 {
fmt.Println("incorrect range")
}
defer resp.Body.Close()
我还将循环变量更改为for i := 0; i < chunks-1; i++ {
并在执行例行程序后更改了该部分
startByte = endByte + 1
endByte += 1024 * 1024 * 10
if startByte >= length {
break
}
for endByte >= length {
endByte = endByte - 1
}
并以类似方式更改j循环变量
这些更改似乎对我有用,但是我没有合适的测试数据可以真正检查