Go-RESTful实现下载功能思路详解

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

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

ws.Route(ws.GET("/img").To(download.DownLoadRequest).Doc("Add user").
		Returns(http.StatusOK, "下载成功", ""))
	//注册webservice
	restful.Add(ws)
	log.Fatal(http.ListenAndServe(":8080", nil)) // 启动监听

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("文件读取失败"))
	}
}