golang #代码片段 #文件服务 #下载 #goframe

文件的断点续传,需要使用到 HTTP 协议的 206 状态码 实现文件的断点续传,具体的原理在网上可以找到一堆。网上找的所有 golang 代码都缺少一个步骤 flush 即将文件的内容写入后应主动将内容发送到客户端 即代码 39行 内容。

在研究了 java 的实现后,发现所有的 golang 实现都没有用 flush 不知道是 golang 本身提供的 net/http 里的 writer 已经实现了写入即 flush 的功能。

我们使用的 goframe 框架进行开发,在断点续传时需要做如下处理。PS:已使用迅雷和

参考的网站有 https://www.cnblogs.com/peteremperor/p/14081050.html https://blog.csdn.net/zgh670042085/article/details/84416227

  1. func download(r *ghttp.Request) {
  2. f,_ := gfile.Open("D:\\Windows.iso")
  3. info, _:= f.Stat()
  4. r.Response.Header().Add("Accept-Ranges", "bytes")
  5. r.Response.Header().Add("Content-Disposition", "attachment; filename="+info.Name())
  6. var start, end int64
  7. //fmt.Println(request.Header,"\n")
  8. if rangeBytes := r.Header.Get("Range"); rangeBytes != "" {
  9. if strings.Contains(rangeBytes, "bytes=") && strings.Contains(rangeBytes, "-") {
  10. fmt.Sscanf(rangeBytes, "bytes=%d-%d", &start, &end)
  11. if end == 0 {
  12. end = info.Size() - 1
  13. }
  14. r.Response.Header().Add("Content-Length", strconv.FormatInt(end-start+1, 10))
  15. r.Response.Header().Add("Content-Range", fmt.Sprintf("bytes %v-%v/%v", start, end, info.Size()))
  16. r.Response.WriteHeader(http.StatusPartialContent)
  17. } else {
  18. r.Response.WriteHeader(http.StatusBadRequest)
  19. return
  20. }
  21. } else {
  22. r.Response.Header().Add("Content-Length", strconv.FormatInt(info.Size(), 10))
  23. start = 0
  24. end = info.Size() - 1
  25. }
  26. _, _ = f.Seek(start, 0)
  27. n := 1024 * 1024
  28. buf := make([]byte, n)
  29. for {
  30. if end-start+1 < int64(n) {
  31. n = int(end - start + 1)
  32. }
  33. _, err := f.Read(buf[:n])
  34. if err != nil {
  35. }
  36. err = nil
  37. r.Response.Write(buf[:n])
  38. r.Response.Flush()
  39. start += int64(n)
  40. if start >= end+1 {
  41. return
  42. }
  43. }
  44. }