Golang實現斷點續傳

• https://github.com/artificerpi/oget

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

具體實現請訪問oget　，歡迎交流和分享。