在go语言中实现将文件流传输到浏览器或写入http.ResponseWriter示例

Go  /  管理员 发布于 1年前   647

在go语言中怎么实现将PDF或MP3等文件直接流式传输/下载到用户的网络浏览器（客户端）？

解决方案：

使用 bytes.NewBuffer() 函数将文件转换为缓冲区，然后写入 http.ResponseWriter。

代码片段摘自之前的文章：

在go语言中使用GoPDF包把html生成PDF文件示例

https://www.zongscan.com/demo333/96261.html

示例代码：

 func PDF(w http.ResponseWriter, r *http.Request) {
          ...
          // 抓取生成的 receipt.pdf 文件并将其流到浏览器中
          streamPDFbytes, err := ioutil.ReadFile("./receipt.pdf")
          
          if err != nil {
                  fmt.Println(err)
                  os.Exit(1)
          }
          
          b := bytes.NewBuffer(streamPDFbytes)
          
          // 直接流向客户端（浏览器）
          w.Header().Set("Content-type", "application/pdf")
          
          if _, err := b.WriteTo(w); err != nil { 
                  fmt.Fprintf(w, "%s", err)
          }
          
          w.Write([]byte("生成PDF"))
  }

有兴趣的可以一试！