快捷導(dǎo)航

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

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

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

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

下載實(shí)現(xiàn)思路

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

文件系統(tǒng)IO：

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

網(wǎng)絡(luò)IO：

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

服務(wù)建立

業(yè)務(wù)背景是需要啟動(dòng)一個(gè)服務(wù)，使得用戶可以通過(guò)訪問這個(gè)服務(wù)對(duì)某一資源的接口地址進(jìn)行訪問后下載，因此需要先建立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中務(wù)必指定MIME類型，否則會(huì)按JSON或者XML處理（具體邏輯可以查看相關(guān)源碼）。支持的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

配置路由

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

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

這里需要注意的有幾點(diǎn)：

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

func DownLoadRequest(request *restful.Request, response *restful.Response) {
	// 建立客戶端去Get請(qǐng)求一個(gè)資源，此處以一張圖片為例子
	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("下載失敗"))
	}
	//此處是關(guān)鍵
	response.ResponseWriter.Header().Set(restful.HEADER_ContentType, restful.MIME_OCTET)
	response.ResponseWriter.Header().Set("Content-Disposition", "attachment;filename=20201124032511.png")
	// 將客戶端請(qǐng)求的結(jié)果序列化出來(lái)
	// 不要忘了關(guān)閉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請(qǐng)求一下這個(gè)服務(wù)：