0x00
下载文件时暂停后可以继续接着下载,在线看视频时可以随意拖动进度条,这些都是断点续传所实现的应用。
http1.1支持Range属性从而实现断点续传,客户端在暂停时记录了已经下载的文件范围,当继续下载时就向服务器发送文件剩余的范围(Range),服务器则根据客户端请求的范围(Range)返回相应文件的部分数据,而不用将整个文件返回给客户端。
多线程下载器就是利用这个属性进行多线程下载,下载器将服务器返回的文件数据范围进行分割,然后根据分割后的范围向服务器索取数据。
0x01
首先要告诉客户端支持Range断点续传,然后客户端才能发起Range请求。
这需要在服务器返回http的head以下内容:
writer.Header().Add("Accept-ranges", "bytes") //告诉客户端支持Range,并且是bytes级的断点续传
writer.Header().Add("Content-Length", 文件大小) //文件的总体大小
writer.Header().Add("Content-Disposition", "attachment; filename=文件名称“)
若客户端请求Ranges,服务器就返回Content-Range
,Content-Range
的内容是0~(文件大小-1)/文件大小
例如一个大小为1024的文件,Content-Range
内容为0-1023/1024
//
writer.Header().Add("Content-Range", "bytes 0-1023/1024")
//断点
writer.Header().Add("Content-Range", "bytes 200-1023/1024")
服务器要从客户端的请求的Range判断是否是有效的范围。
如果是有效的范围就要将相应状态码设为206 (Partial Content),否则就设为416 (Request Range Not Satisfiable)
0x02
代码实现:
func sendFile(writer http.ResponseWriter, request *http.Request, f *os.File) {
defer f.Close()
info, err := f.Stat()
if err != nil {
log.Println("sendFile1", err.Error())
http.NotFound(writer, request)
return
}
writer.Header().Add("Accept-Ranges", "bytes")
writer.Header().Add("Content-Disposition", "attachment; filename="+info.Name())
var start, end int64
//fmt.Println(request.Header,"\n")
if r := request.Header.Get("Range"); r != "" {
if strings.Contains(r, "bytes=") && strings.Contains(r, "-") {
fmt.Sscanf(r, "bytes=%d-%d", &start, &end)
if end == 0 {
end = info.Size() - 1
}
if start > end || start < 0 || end < 0 || end >= info.Size() {
writer.WriteHeader(http.StatusRequestedRangeNotSatisfiable)
log.Println("sendFile2 start:", start, "end:", end, "size:", info.Size())
return
}
writer.Header().Add("Content-Length", strconv.FormatInt(end-start+1, 10))
writer.Header().Add("Content-Range", fmt.Sprintf("bytes %v-%v/%v", start, end, info.Size()))
writer.WriteHeader(http.StatusPartialContent)
} else {
writer.WriteHeader(http.StatusBadRequest)
return
}
} else {
writer.Header().Add("Content-Length", strconv.FormatInt(info.Size(), 10))
start = 0
end = info.Size() - 1
}
_, err = f.Seek(start, 0)
if err != nil {
log.Println("sendFile3", err.Error())
writer.WriteHeader(http.StatusInternalServerError)
return
}
n := 512
buf := make([]byte, n)
for {
if end-start+1 < int64(n) {
n = int(end - start + 1)
}
_, err := f.Read(buf[:n])
if err != nil {
log.Println("1:", err)
if err != io.EOF {
log.Println("error:", err)
}
return
}
err = nil
_, err = writer.Write(buf[:n])
if err != nil {
//log.Println(err, start, end, info.Size(), n)
return
}
start += int64(n)
if start >= end+1 {
return
}
}
}