Golang实现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引用的依赖包全部完整的写入到生成的文件中的")) //剩下的所有静态文件 staticFiles, _ := ioutil.ReadDir("static/report/") for _, tempfile := range staticFiles { if tempfile.Name() == "jquery.min.js" { continue } tempfilename := "static/report/" + tempfile.Name() tempfileP, _ := os.Open(tempfilename) info, _ := os.Stat(tempfilename) curFileByte := make([]byte, info.Size()) _, err := tempfileP.Read(curFileByte) if err != nil { utils.LogErrorln(err.Error()) } if isJs, _ := regexp.MatchString(`\.js$`, tempfilename); isJs { allStaticFiles = append(allStaticFiles, []byte("")) } else if isCss, _ := regexp.MatchString(`\.css$`, tempfilename); isCss { allStaticFiles = append(allStaticFiles, []byte("")) } tempfileP.Close() } // 转成http.File格式进行返回 mf := &MyFile{ Reader: bytes.NewReader(html), mif: myFileInfo{ name: "report.html", data: html, }, } var f http.File = mf return f }

OK！目前为止，后端的文件生成->打包->压缩都已经做好啦，我们把他们串起来：

func (c *Controller)Download() {
    var b bytes.Buffer
    gw := gzip.NewWriter(&b)

    tw := tar.NewWriter(gw)

    // 生成动态report,并添加进压缩包
    // 调用上文中的testWrite方法
    dataFile := testWrite(responseByRules, strTaskId)
    info, _ := dataFile.Stat()
    header, _ := tar.FileInfoHeader(info, "")
    header.Name = "report_" + strTaskId + "/" + 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())
    }

    tw.Close()
    gw.Close()
    rw := c.Ctx.ResponseWriter
    rw.Header().Set("Content-Disposition", "attachment; filename="+"report_"+strTaskId+".tar.gz")
    rw.Header().Set("Content-Description", "File Transfer")
    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)
    http.ServeContent(rw, c.Ctx.Request, "report_"+strTaskId+".tar.gz", time.Now(), bytes.NewReader(b.Bytes()))
}

后端部分已经全部实现了，前端部分如何接收呢，本例中我做了一个按钮嵌套标签来进行请求:

这样，当前端页面中点击下载报表按钮之后，会自动启动下载，下载我们后端传回的report.tar.gz文件。