Golang 断点续传
下载文件时暂停后可以继续接着下载,在线看视频时可以随意拖动进度条,这些都是断点续传所实现的应用。
http1.1支持Range属性从而实现断点续传,客户端在暂停时记录了已经下载的文件范围,当继续下载时就向服务器发送文件剩余的范围Range,服务器则根据客户端请求的范围Range返回相应文件的部分数据,而不用将整个文件返回给客户端。
多线程下载器就是利用这个属性进行多线程下载,下载器将服务器返回的文件数据范围进行分割,然后根据分割后的范围向服务器索取数据。
1.首先要告诉客户端支持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 PartialContent,否则就设为416
2.代码实现
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 } } }