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) {
    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
    //判断header有没有range
    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 > end || start < 0 || end-1 < 0 || end > info.Size() {
                writer.WriteHeader(http.StatusRequestedRangeNotSatisfiable)
                log.Println("sendFile2")
                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.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
    }
    
    buf := make([]byte, 512)
    for {
        n, err := f.Read(buf)
        if err != nil {
            if err != io.EOF {
                log.Println("error:", err)
            }
            return
        }
        writer.Write(buf[:n])
        if start += int64(n); start >= end+1 {
            return
        }
    }
}
Last modification:February 9th, 2020 at 05:56 pm
如果觉得我的文章对你有用,请随意赞赏