【Golang】实现文件服务器断点续传

Author： Rehtt
发布时间：February 5, 2020
7982 views
One comment
2878 words
Categories： Golang Golang笔记

# 0x00
下载文件时暂停后可以继续接着下载，在线看视频时可以随意拖动进度条，这些都是断点续传所实现的应用。
http1.1支持**Range**属性从而实现断点续传，客户端在暂停时记录了已经下载的文件范围，当继续下载时就向服务器发送文件剩余的范围(Range)，服务器则根据客户端请求的范围(Range)返回相应文件的部分数据，而不用将整个文件返回给客户端。

多线程下载器就是利用这个属性进行多线程下载，下载器将服务器返回的文件数据范围进行分割，然后根据分割后的范围向服务器索取数据。

# 0x01
首先要告诉客户端支持Range断点续传，然后客户端才能发起Range请求。
这需要在服务器返回http的head以下内容：
```golang
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`
```golang
//
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
代码实现：
```golang
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
		}
	}

}
```

Last modification：February 24, 2020