Go-RESTful實現(xiàn)下載功能思路詳解

更新時間：2022年10月11日 10:15:25 作者：伯納烏純白

這篇文章主要介紹了Go-RESTful實現(xiàn)下載功能,文件下載包括文件系統(tǒng)IO和網絡IO，本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下

Go-RESTful實現(xiàn)下載功能

下載實現(xiàn)思路

下圖為實現(xiàn)一個文件下載所需要考慮的因素：

文件系統(tǒng)IO：

● 文件流的讀寫，其中又包括分文件類型讀寫、文件直接拷貝、借助緩沖區(qū)進行IO操作。一般采用直接對二進制文件進行讀寫，也有特殊情況如zip壓縮包

網絡IO：

● 文件傳輸?shù)木W絡協(xié)議，是通過http還是tcp進行傳輸？一般是通過http層面進行讀寫，
● 網絡IO中文件主體放在header還是body中？文件直接通過操作系統(tǒng)IO還是通過網絡IO形成附件供下載
● 網絡傳輸?shù)腃ontent-Type，是否符合框架的標準？appclication/octet-stream和其他如applicaiton/zip的區(qū)別
下載的實現(xiàn)流程

服務建立

業(yè)務背景是需要啟動一個服務，使得用戶可以通過訪問這個服務對某一資源的接口地址進行訪問后下載，因此需要先建立Web Service

ws := new(restful.WebService)
	ws.Path("/download").Consumes(restful.MIME_JSON,restful.MIME_OCTET).
		Produces(restful.MIME_JSON, restful.MIME_OCTET)

這里需要注意的是，Consumers和Produces中務必指定MIME類型，否則會按JSON或者XML處理（具體邏輯可以查看相關源碼）。支持的MIME類型如下：

MIME_XML   = "application/xml"          // Accept or Content-Type used in Consumes() and/or Produces()
	MIME_JSON  = "application/json"         // Accept or Content-Type used in Consumes() and/or Produces()
	MIME_OCTET = "application/octet-stream" // If Content-Type is not present in request, use the default

配置路由

建立一個路由如下，同時引入函數(shù)的handler：

ws.Route(ws.GET("/img").To(download.DownLoadRequest).Doc("Add user").
		Returns(http.StatusOK, "下載成功", ""))
	//注冊webservice
	restful.Add(ws)
	log.Fatal(http.ListenAndServe(":8080", nil)) // 啟動監(jiān)聽

這里需要注意的有幾點：

restful是接口風格，并不是直接的http方法，因此restful.request并不和http.request等價。要接收http.request的數(shù)據(jù)應當是對restful.request的Writer進行操作。
一定要引入Content-Disposition，這樣才會使得Get到的二進制文件直接以附件的形式加載出來
Content-Type一定要是restful支持的MIME類型

func DownLoadRequest(request *restful.Request, response *restful.Response) {
	// 建立客戶端去Get請求一個資源，此處以一張圖片為例子
	client := http.Client{}
	defer client.CloseIdleConnections()
	res, err := client.Get("https://img-home.csdnimg.cn/images/20201124032511.png")
	if err != nil {
		response.WriteError(http.StatusInternalServerError, errors.New("下載失敗"))
	}
	//此處是關鍵
	response.ResponseWriter.Header().Set(restful.HEADER_ContentType, restful.MIME_OCTET)
	response.ResponseWriter.Header().Set("Content-Disposition", "attachment;filename=20201124032511.png")
	// 將客戶端請求的結果序列化出來
	// 不要忘了關閉Body
	defer res.Body.Close()
	b, err := ioutil.ReadAll(res.Body)
	if err != nil {
		response.WriteError(http.StatusInternalServerError, errors.New("下載失敗"))
	}
	_, err = response.ResponseWriter.Write(b)
	if err != nil {
		response.WriteError(http.StatusInternalServerError, errors.New("文件讀取失敗"))
	}
}

用Postman請求一下這個服務：