色婆ijqwj 发表于 2018-9-20 11:27:07

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请求头,告诉服务器应该返回给客户端的内容部分。
  

# 通常为 bytes  
Range: =-
  
Range: =-
  
Range: =-, -
  
Range: =-, -, -
  


[*]代码实现
  

    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 ,欢迎交流和分享。


页: [1]
查看完整版本: Golang实现断点续传