最近遇到了一個下載靜態html報表的需求，需要以提供壓縮包的形式完成下載功能，實現的過程中發現相關文檔非常雜，故總結一下自己的實現。
開發環境：
系統環境：MacOS + Chrome
框架：beego
壓縮功能：tar + gzip
目標壓縮文件：自帶數據和全部包的靜態html文件
首先先提一下http server文件下載的實現，其實就是在后端返回前端的數據包中，將數據頭設置為下載文件的格式，這樣前端收到返回的響應時，會直接觸發下載功能(就像時平時我們在chrome中點擊下載那樣)
數據頭設置格式如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
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進行壓縮，實現如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
//最內層用bytes.Buffer來進行文件的存儲var b bytes.Buffer//嵌套tar包的writter和gzip包的writergw := gzip.NewWriter(&b)tw := tar.NewWriter(gw)dataFile := //...文件的產生邏輯，dataFile為File類型info, _ := dataFile.Stat()header, _ := tar.FileInfoHeader(info, "")//下載后當前文件的路徑設置header.Name = "report" + "/" + header.Nameerr := 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轉文件的函數可以用，于是我們需要自己實現。
實現如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
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 allfunc (mif myFileInfo) ModTime() time.Time { return time.Time{} } // Return whatever you wantfunc (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 dofunc (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} |
依賴包和數據的寫入邏輯：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
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！ 目前為止，后端的文件生成->打包->壓縮都已經做好啦，我們把他們串起來：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
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>標簽來進行請求:
|
1
2
3
|
<a href="/traffic/download_indicator?task_id={{$.taskId}}&task_type={{$.taskType}}&status={{$.status}}&agent_addr={{$.agentAddr}}&glaucus_addr={{$.glaucusAddr}}" rel="external nofollow" > <button style="font-family: 'SimHei';font-size: 14px;font-weight: bold;color: #0d6aad;text-decoration: underline;margin-left: 40px;" type="button" class="btn btn-link">下載報表</button></a> |
這樣，當前端頁面中點擊下載報表按鈕之后，會自動啟動下載，下載我們后端傳回的report.tar.gz文件。
到此這篇關于Golang實現http server提供壓縮文件下載功能的文章就介紹到這了,更多相關Golang http server 壓縮下載內容請搜索服務器之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持服務器之家！
原文鏈接：https://segmentfault.com/a/1190000038867527
