怎么在Golang中利用http server實現(xiàn)一個壓縮文件下載功能？相信很多沒有經(jīng)驗的人對此束手無策，為此本文總結(jié)了問題出現(xiàn)的原因和解決方法，通過這篇文章希望你能解決這個問題。

網(wǎng)站建設(shè)哪家好，找創(chuàng)新互聯(lián)！專注于網(wǎng)頁設(shè)計、網(wǎng)站建設(shè)、微信開發(fā)、微信小程序開發(fā)、集團企業(yè)網(wǎng)站建設(shè)等服務(wù)項目。為回饋新老客戶創(chuàng)新互聯(lián)還提供了甘德免費建站歡迎大家使用！

開發(fā)環(huán)境：

系統(tǒng)環(huán)境：MacOS + Chrome
框架：beego
壓縮功能：tar + gzip
目標壓縮文件：自帶數(shù)據(jù)和全部包的靜態(tài)html文件

首先先提一下http server文件下載的實現(xiàn)，其實就是在后端返回前端的數(shù)據(jù)包中，將數(shù)據(jù)頭設(shè)置為下載文件的格式，這樣前端收到返回的響應(yīng)時，會直接觸發(fā)下載功能(就像時平時我們在chrome中點擊下載那樣)
數(shù)據(jù)頭設(shè)置格式如下：

func (c *Controller)Download() {
  //...文件信息的產(chǎn)生邏輯
  //
  //rw為responseWriter
  rw := c.Ctx.ResponseWriter
  //規(guī)定下載后的文件名
  rw.Header().Set("Content-Disposition", "attachment; filename="+"(文件名字)")
  rw.Header().Set("Content-Description", "File Transfer")
  //標明傳輸文件類型
  //如果是其他類型，請參照：/tupian/20230522/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，則需調(diào)用b.Bytes()
  http.ServeContent(rw, c.Ctx.Request, "(文件名字)", time.Now(), bytes.NewReader(b.Bytes()))
}

這樣，beego后端就會將在頭部標記為下載文件的數(shù)據(jù)包發(fā)送給前端，前端收到后會自動啟動下載功能。

然而這只是最后一步的情況，如何將我們的文件先進行壓縮再發(fā)送給前端提供下載呢？

如果需要下載的不只一個文件，需要用tar打包，再用gzip進行壓縮，實現(xiàn)如下：

  //最內(nèi)層用bytes.Buffer來進行文件的存儲
  var b bytes.Buffer
  //嵌套tar包的writter和gzip包的writer
  gw := gzip.NewWriter(&b)
  tw := tar.NewWriter(gw)


  dataFile := //...文件的產(chǎn)生邏輯，dataFile為File類型
  info, _ := dataFile.Stat()
  header, _ := tar.FileInfoHeader(info, "")
  //下載后當前文件的路徑設(shè)置
  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())
  }
  //...可以繼續(xù)添加文件
  //tar writer 和 gzip writer的關(guān)閉順序一定不能反
  tw.Close()
  gw.Close()

最后和中間步驟完成了，我們只剩File文件的產(chǎn)生邏輯了，由于是靜態(tài)html文件，我們需要把所有html引用的依賴包全部完整的寫入到生成的文件中的"))   //剩下的所有靜態(tài)文件   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()   }      // 轉(zhuǎn)成http.File格式進行返回   mf := &MyFile{     Reader: bytes.NewReader(html),     mif: myFileInfo{       name: "report.html",       data: html,     },   }   var f http.File = mf   return f }

OK！ 目前為止，后端的文件生成->打包->壓縮都已經(jīng)做好啦，我們把他們串起來：

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

  tw := tar.NewWriter(gw)

  // 生成動態(tài)report,并添加進壓縮包
  // 調(diào)用上文中的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()))
}

后端部分已經(jīng)全部實現(xiàn)了，前端部分如何接收呢，本例中我做了一個按鈕嵌套標簽來進行請求:

   下載報表

這樣，當前端頁面中點擊下載報表按鈕之后，會自動啟動下載，下載我們后端傳回的report.tar.gz文件。

看完上述內(nèi)容，你們掌握怎么在Golang中利用http server實現(xiàn)一個壓縮文件下載功能的方法了嗎？如果還想學到更多技能或想了解更多相關(guān)內(nèi)容，歡迎關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道，感謝各位的閱讀！

本文標題：怎么在Golang中利用httpserver實現(xiàn)一個壓縮文件下載功能-創(chuàng)新互聯(lián)
網(wǎng)站鏈接：http://weahome.cn/article/depsch.html