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
        }
    }

}

 

posted @ 2020-12-03 17:39  Mr.peter  阅读(1212)  评论(0编辑  收藏  举报