您好,登录后才能下订单哦!
怎么在Golang中利用http server实现一个压缩文件下载功能?相信很多没有经验的人对此束手无策,为此本文总结了问题出现的原因和解决方法,通过这篇文章希望你能解决这个问题。
开发环境:
系统环境: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引用的依赖包全部完整的写入到生成的文件中的<script>和<style>标签下。此外,在本例子中,报告部分还需要一些静态的json数据来填充表格和图像,这部分数据是以map存储在内存中的。当然可以先保存成文件再进行上面一步的打包压缩,但是这样会产生并发的问题,因此我们需要先将所有的依赖包文件和数据写入一个byte.Buffer中,最后将这个byte.Buffer转回File格式。
Golang中并没有写好的byte.Buffer转文件的函数可以用,于是我们需要自己实现。
实现如下:
type myFileInfo struct { name string data []byte } func (mif myFileInfo) Name() string { return mif.name } func (mif myFileInfo) Size() int64 { return int64(len(mif.data)) } func (mif myFileInfo) Mode() os.FileMode { return 0444 } // Read for all func (mif myFileInfo) ModTime() time.Time { return time.Time{} } // Return whatever you want func (mif myFileInfo) IsDir() bool { return false } func (mif myFileInfo) Sys() interface{} { return nil } type MyFile struct { *bytes.Reader mif myFileInfo } func (mf *MyFile) Close() error { return nil } // Noop, nothing to do func (mf *MyFile) Readdir(count int) ([]os.FileInfo, error) { return nil, nil // We are not a directory but a single file } func (mf *MyFile) Stat() (os.FileInfo, error) { return mf.mif, nil }
依赖包和数据的写入逻辑:
func testWrite(data map[string]interface{}, taskId string) http.File { //最后生成的html,打开html模版 tempfileP, _ := os.Open("views/traffic/generatePage.html") info, _ := tempfileP.Stat() html := make([]byte, info.Size()) _, err := tempfileP.Read(html) // 将data数据写入html var b bytes.Buffer // 创建Json编码器 encoder := json.NewEncoder(&b) err = encoder.Encode(data) if err != nil { utils.LogErrorln(err.Error()) } // 将json数据添加到html模版中 // 方式为在html模版中插入一个特殊的替换字段,本例中为{Data_Json_Source} html = bytes.Replace(html, []byte("{Data_Json_Source}"), b.Bytes(), 1) // 将静态文件添加进html // 如果是.css,则前后增加<style></style>标签 // 如果是.js,则前后增加<script><script>标签 allStaticFiles := make([][]byte, 0) // jquery 需要最先进行添加 tempfilename := "static/report/jquery.min.js" tempfileP, _ = os.Open(tempfilename) info, _ = os.Stat(tempfilename) curFileByte := make([]byte, info.Size()) _, err = tempfileP.Read(curFileByte) allStaticFiles = append(allStaticFiles, []byte("<script>")) allStaticFiles = append(allStaticFiles, curFileByte) allStaticFiles = append(allStaticFiles, []byte("</script>")) //剩下的所有静态文件 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("<script>")) allStaticFiles = append(allStaticFiles, curFileByte) allStaticFiles = append(allStaticFiles, []byte("</script>")) } else if isCss, _ := regexp.MatchString(`\.css$`, tempfilename); isCss { allStaticFiles = append(allStaticFiles, []byte("<style>")) allStaticFiles = append(allStaticFiles, curFileByte) allStaticFiles = append(allStaticFiles, []byte("</style>")) } 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())) }
后端部分已经全部实现了,前端部分如何接收呢,本例中我做了一个按钮嵌套<a>标签来进行请求:
<a href="/traffic/download_indicator?task_id={{$.taskId}}&task_type={{$.taskType}}&status={{$.status}}&agent_addr={{$.agentAddr}}&glaucus_addr={{$.glaucusAddr}}" rel="external nofollow" > <button type="button" class="btn btn-link">下载报表</button> </a>
这样,当前端页面中点击下载报表按钮之后,会自动启动下载,下载我们后端传回的report.tar.gz文件。
看完上述内容,你们掌握怎么在Golang中利用http server实现一个压缩文件下载功能的方法了吗?如果还想学到更多技能或想了解更多相关内容,欢迎关注亿速云行业资讯频道,感谢各位的阅读!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。