Golang Gin框架实现文件下载功能

最新推荐文章于 2024-09-04 09:20:33 发布

JYZzz.

最新推荐文章于 2024-09-04 09:20:33 发布

阅读量2.1w

点赞数 7

分类专栏： Go语言学习文章标签： golang

本文链接：https://blog.csdn.net/weixin_43823363/article/details/105946099

版权

Go语言学习专栏收录该内容

12 篇文章 0 订阅

订阅专栏

Layui框架实现文件上传

基本的思路就是随便创建一个元素，然后使用layui的upload组件对创建的元素进行渲染，详见代码

<!DOCTYPE html>
<html lang="en">
<head>
    <script src="jquery-3.5.0.min.js" type="text/javascript"></script> <!-- jquery包，记得替换为你本地的路径 -->

    <link rel="stylesheet" href="layui/css/layui.css"> <!--layui框架的样式库 同样你本地的路径 -->
    <script src="pkg/layui/layui.js"></script><!--enmmm 没错，路径-->
</head>	
<body>
<!-- 这里我是用的a标签，当然你可以使用别的 label、button等均可-->
 <a href="javascript:(0);" id="attachment-upload">+添加附件</a>
</body>
<script>
	//以layui.js的方式引入使用时需要手动加载layui的组件
	layui.use(['layer','upload'],function(){ // PS：个人理解，该行只是为了引入layui框架的组件，html文件加载完毕后执行，内部的代码和直接写在script标签内没啥区别，只是可以在代码中使用layer了
	// 此处引入layer只是为了打印一些东西
		var layer = layui.layer,
		upload = layui.upload;
		
	// 渲染元素，使其支持文件上传
	// 详情请移步 https://www.layui.com/doc/modules/upload.html
	// https://www.layui.com/demo/upload.html
	 upload.render({
            elem: '#attachment-upload', // 感觉利用了jquery的id选择器，这里用#id(上传附件的标签、按钮值类型元素id)就可以了
            url: dev_url + "fc/upload", // 后端接收上传文件的接口地址
            headers: {   // 这里官方文档中没有，是在请求后端接口时在request header中增加一些请求参数，实测可行
                'api_token': uInfo
            },
            accept: 'file', // 这里可以限定支持上传文件的类型，详见文档
            done: function (res) { // 调用后端上传文件接口后的返回
           // 这里解析下后端的返回字段，具体怎么着看业务吧
                if (res.success == false) {
                    layer.msg("上传文件失败!" + res.msg);
                }
        });
})
</script>
</html>

PS：注意下使用layui上传文件的请求格式为multiply/form-data，参数为file,(binary)格式

layui上传文件组件请求

Gin框架获取前端上传的文件

func UploadFileControl(c *gin.Context) {
	logrus.Infof("[UploadFileControl] user_id =%d", userId)
	// GIN框架获取前端上传文件
	// 这里的参数其实是上传文件时的字段名，也就是上面图片中的file，如果前端是自己定义的其他字段名，需要替换下
	uploadFile, fileHeader, err := c.Request.FormFile("file")
	if err != nil {
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"msg":     "获取文件信息失败!" + err.Error(),
		})
	}
	if uploadFile != nil { // 记得及时关闭文件，避免内存泄漏
		defer uploadFile.Close()
	}

// 读取上传文件的内容
// 其实这里直接读取所有不太合理,如果文件过大时会占用很多内存，可以考虑使用缓冲区读取
	fileContent, err := ioutil.ReadAll(uploadFile)
	if err != nil {
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"msg":     "读取文件内容失败!" + err.Error(),
		})
	}

// 接受到文件后的处理方式，可以保存到本地，也可以上传到文件对象，看自己了
	/*fileId, err := oss.GetOssClient().UploadFile(userId, fileHeader.Filename, fileContent)
	if err != nil {
		logrus.Errorf("[UploadFile] error,user_id = %d,err_msg= %s", userId, err.Error())
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"msg":     "上传文件失败!请稍后再试",
		})
	}*/

// 这里向前端返回下上传成功的信息
	c.JSON(http.StatusOK, gin.H{
		"success":   true,
		"msg":       "",
	})
}

Gin框架的文件下载

        Response Header中的Content-Type指定了服务器端返回数据的类型，浏览器自己是可以处理这些类型的，当返回的数据为文件类型时，浏览器会自行下载。具体的类型和content-type的对应方式可见https://tool.oschina.net/commons/.
        当Content-Type是一些文件类型时，使用Content-Disposition可以指定浏览器下载文件时的默认文件名
        因此，我们想要使用gin实现下载文件的功能，只需要在接口返回时设置Response-Header中的Content-Type为文件类型，并设置Content-Disposition指定默认的文件名，然后将文件数据返回浏览器即可，具体代码如下：

func DownloadFileControl(c *gin.Context) {
	//  查询一些必要的参数 进行一些必要的验证
	attchIdStr := c.Query("attachment_id")
	attachmentName = c.Query("attachment_name")
	
	// 获取要返回的文件数据流
	// 看你文件存在哪里了，本地就直接os.Open就可以了，总之是要获取一个[]byte
	/*	fileContent, err := oss.GetOssClient().DownloadFile(req.AttachmentId, req.AttachmentName)
	if err != nil {
		logrus.Errorf("[DownloadFile] download file error,file_id = %d,file_name = %s,user_is = %d,err_msg= %s", req.UserId, req.AttachmentId, req.AttachmentName, err.Error())
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"msg":     "下载文件失败,请联系管理员",
		})
		return
	}*/
	
// 设置返回头并返回数据
	fileContentDisposition := "attachment;filename=\"" + attachmentName + "\""
	c.Header("Content-Type", "application/zip") // 这里是压缩文件类型 .zip
	c.Header("Content-Disposition", fileContentDisposition)
	c.Data(http.StatusOK, contentType, fileContent)