Go 项目实战：实现一个提供压缩文件下载功能的 http server
Posted Go语言中文网
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Go 项目实战：实现一个提供压缩文件下载功能的 http server相关的知识，希望对你有一定的参考价值。
最近遇到了一个下载静态html报表的需求，需要以提供压缩包的形式完成下载功能，实现的过程中发现相关文档非常杂，故总结一下自己的实现。
开发环境：
系统环境：MacOS + Chrome 框架：beego 压缩功能：tar + gzip 目标压缩文件：自带数据和全部包的静态html文件
首先先提一下http server文件下载的实现，其实就是在后端返回前端的数据包中，将数据头设置为下载文件的格式，这样前端收到返回的响应时，会直接触发下载功能(就像时平时我们在chrome中点击下载那样)
数据头设置格式如下：
func (c *Controller)Download() {
//...文件信息的产生逻辑
//
//rw为responseWriter
rw := c.Ctx.ResponseWriter
//规定下载后的文件名
rw.Header().Set("Content-Disposition", "attachment; filename="+"(文件名字)")
rw.Header().Set("Content-Description", "File Transfer")
//标明传输文件类型
//如果是其他类型，请参照：https://www.runoob.com/http/http-content-type.html
rw.Header().Set("Content-Type", "application/octet-stream")
rw.Header().Set("Content-Transfer-Encoding", "binary")
rw.Header().Set("Expires", "0")
rw.Header().Set("Cache-Control", "must-revalidate")
rw.Header().Set("Pragma", "public")
rw.WriteHeader(http.StatusOK)
//文件的传输是用byte slice类型，本例子中：b是一个bytes.Buffer，则需调用b.Bytes()
http.ServeContent(rw, c.Ctx.Request, "(文件名字)", time.Now(), bytes.NewReader(b.Bytes()))
}
这样，beego后端就会将在头部标记为下载文件的数据包发送给前端，前端收到后会自动启动下载功能。
然而这只是最后一步的情况，如何将我们的文件先进行压缩再发送给前端提供下载呢？
如果需要下载的不只一个文件，需要用tar打包，再用gzip进行压缩，实现如下：
//最内层用bytes.Buffer来进行文件的存储
var b bytes.Buffer
//嵌套tar包的writter和gzip包的writer
gw := gzip.NewWriter(&b)
tw := tar.NewWriter(gw)
dataFile := //...文件的产生逻辑，dataFile为File类型
info, _ := dataFile.Stat()
header, _ := tar.FileInfoHeader(info, "")
//下载后当前文件的路径设置
header.Name = "report" + "/" + header.Name
err := tw.WriteHeader(header)
if err != nil {
utils.LogErrorln(err.Error())
return
}
_, err = io.Copy(tw, dataFile)
if err != nil {
utils.LogErrorln(err.Error())
}
//...可以继续添加文件
//tar writer 和 gzip writer的关闭顺序一定不能反
tw.Close()
gw.Close()
最后和中间步骤完成了，我们只剩File文件的产生逻辑了，由于是静态html文件，我们需要把所有html引用的依赖包全部完整的写入到生成的文件中的
原文链接：https://segmentfault.com/a/1190000038867527
推荐阅读
以上是关于Go 项目实战：实现一个提供压缩文件下载功能的 http server的主要内容，如果未能解决你的问题，请参考以下文章