用Golang语言 实现断点续传

HTTP Range 请求

HTTP Range请求允许服务器向客户端发送HTTP消息体的一部分数据。Partial Requests在使用比较大的网络媒体文件或者在下载文件时提供暂停和恢复功能时很有用。
这也是下载时实现HTTP断点续传的一个关键。

HTTP 206 (Partial Content)

如果服务器能返回HTTP 206请求,我们就知道它能够支持Range request.

# curl -I https://www.baidu.com/ -H "Range: bytes=0-"

HTTP/1.1 206 Partial Content
Cache-Control: private, no-cache, no-store, proxy-revalidate, no-transform
Connection: Keep-Alive
Content-Length: 277
Content-Range: bytes 0-276/277
Content-Type: text/html
Date: Mon, 25 Dec 2017 15:29:35 GMT
Etag: "575e1f71-115"
Last-Modified: Mon, 13 Jun 2016 02:50:25 GMT
Pragma: no-cache
Server: bfe/1.0.8.18
  • go代码
        req, err := http.NewRequest(http.MethodHead, f.URL, nil)
    if err != nil {
        log.Fatalf("Invalid url for downloading: %s, error: %v", f.URL, err)
    }
    req.Header.Set("Range", "bytes=0-")
    
    client := &http.Client{
        Timeout: time.Second * 5,
    }
    resp, err := client.Do(req)
    if err != nil {
        log.Fatal(err)
    }
    defer resp.Body.Close()
    
        switch resp.StatusCode {
    case http.StatusPartialContent:
        rangeSatisfiable = true
        log.Println("Partial Content is supported.")
    case http.StatusOK, http.StatusRequestedRangeNotSatisfiable:
        log.Println(f.URL, "does not support for range request.")
    default:
        log.Fatal("Got unexpected status code", resp.StatusCode)
        return
    }

Range 请求头

我们可以使用Range请求头,告诉服务器应该返回给客户端的内容部分。

# 通常<unit>为 bytes
Range: <unit>=<range-start>-
Range: <unit>=<range-start>-<range-end>
Range: <unit>=<range-start>-<range-end>, <range-start>-<range-end>
Range: <unit>=<range-start>-<range-end>, <range-start>-<range-end>, <range-start>-<range-end>
  • 代码实现
        req, err := http.NewRequest(http.MethodGet, f.URL, nil)
    if err != nil {
        return
    }
        // s.String()实现:  fmt.Sprintf("bytes=%d-%d", h.StartPos, h.EndPos)
    req.Header.Set("Range", s.String())
    
    client := &http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        return
    }
    defer resp.Body.Close()
    
    data, err := ioutil.ReadAll(resp.Body)

Seek and Write

最后类似于Java中的RandAccessFile, 我们可以使用io.WriteAt (推荐,使用接口更好,也容易测试)或者直接用文件句柄写n, err := f.FileHandler.WriteAt(data, int64(s.StartPos))

发表评论