Golang使用Gin框架实现HTTP上传文件过程介绍

更新时间：2023年04月10日 08:34:10 作者：中国流浪猫

由于需求中有文件上传这一个需求，在这里我们就学习一下go语言如何上传文件。本文主要通过表单的方式进行文件上传操作，本文实例为大家分享了Go实现文件上传操作的具体代码，供大家参考，具体内容如下

HTTP上传的文件的原理

HTTP协议的文件上传是通过HTTP POST请求实现的，使用multipart/form-data格式将待上传的文件放入请求体中。

服务器根据请求头中的boundary参数来解析请求体，并根据Content-Disposition字段获取文件名等信息，根据Content-Type字段判断文件类型并保存到相应位置。

Gin框架文件上传Demo

代码逻辑：

通过Gin框架封装的Form表单获取数据，获取上传文件
获取文件名，并创建新的文件存储
将上传的文件内容写入新的文件
返回上传成功信息

package main
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"io"
	"net/http"
	"os"
)
func uploadFile(c *gin.Context) {
	//form表单
	file, header, err := c.Request.FormFile("upload")
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("上传文件失败: %s", err.Error()))
		return
	}
	// 获取文件名，并创建新的文件存储
	filename := header.Filename
	out, err := os.Create(filename)
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("创建文件: %s", err.Error()))
		return
	}
	defer out.Close()
	//将读取的文件流写到文件中
	_, err = io.Copy(out, file)
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("读取文件失败: %s", err.Error()))
		return
	}
	c.String(http.StatusCreated, "上传成功 \n")
}
func main() {
	router := gin.Default()
	//路由:http://localhost:8080/upload
	router.POST("/upload", uploadFile)
	router.Run(":8080")
}

限制文件上传的大小

使用 http.MaxBytesReader() 函数来限制 HTTP 请求中读取的最大字节数。这个函数会返回一个新的 Reader 对象，该对象会在读取请求的正文时自动检查字节数，如果超过指定的最大字节数，则会自动停止读取，返回错误。

//限制大小为2M
c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, int64(2<<20))
	file, header, err := c.Request.FormFile("upload")
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("上传文件失败: %s", err.Error()))
		return
	}

该代码不能限制文件上传大小，只是设置内存大小，即使文件大小比这个大，也会写入临时文件

router := gin.Default()
router.MaxMultipartMemory = 2 * 1024 //2M Byte，默认32M

运行结果截图

文件类型验证

验证上传的文件类型，以确保上传的文件是我们期望的类型，借助“github.com/h2non/filetype”实现对文件类型的判断

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/h2non/filetype"
	"io"
	"net/http"
)
func uploadFile(c *gin.Context) {
	//form表单
	c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, int64(2<<20))
	file, _, err := c.Request.FormFile("upload")
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("上传文件失败: %s", err.Error()))
		return
	}
	content, err := io.ReadAll(file)
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("读取失败: %s", err.Error()))
		return
	}
	// 解析文件类型
	kind, err := filetype.Match(content)
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("文件类型判断失败: %s", err.Error()))
		return
	}
	fmt.Println(kind)
	// 验证文件类型
	if kind == filetype.Unknown {
		c.String(http.StatusCreated, "未知类型 \n")
        return
	}
	if filetype.IsImage(content) {
		c.String(http.StatusCreated, "图片 上传成功 \n")
		return
	}
	c.String(http.StatusCreated, "上传成功 \n")
}

文件上传进度-后台计算文件上传进度

实现原理：

要实现 Gin 框架中的文件上传进度，在文件上传中，计算已上传的字节数，并将其与文件的总大小进行比较，以确定上传的进度。

package main
import (
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
	"os"
)
func uploadFile(c *gin.Context) {
	//form表单
	//c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, int64(2<<20))
	file, fileHeader, err := c.Request.FormFile("upload")
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("上传文件失败: %s", err.Error()))
		return
	}
	filename := fileHeader.Filename
	out, err := os.Create(filename)
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("创建文件: %s", err.Error()))
		return
	}
	defer out.Close()
	count := 0
	for {
		buf := make([]byte, 10000)
		n, err := file.Read(buf)
		if err != nil {
			c.String(http.StatusBadRequest, fmt.Sprintf("读取失败: %s", err.Error()))
			return
		}
		if n == 0 {
			break
		}
		count = count + n
		out.Write(buf)
		fmt.Println(count, float64(fileHeader.Size))
		progress := float64(count) / float64(fileHeader.Size) * 100
		fmt.Println(fmt.Sprintf("%.2f%%", progress))
	}
	c.String(http.StatusCreated, "上传成功 \n")
}
func main() {
	router := gin.Default()
	router.MaxMultipartMemory = 2 * 1024 //2M Byte，默认32M
	//路由:http://localhost:8080/upload
	router.POST("/upload", uploadFile)
	fmt.Println(router.MaxMultipartMemory)
	router.Run(":8080")
}

到此这篇关于Golang使用Gin框架实现HTTP上传文件过程介绍的文章就介绍到这了,更多相关Go HTTP上传文件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

如何避免go的map竞态问题的方法
本文主要介绍了如何避免go的map竞态问题的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2023-02-02
Golang中多个if代码优化小技巧
这篇文章主要为大家详细介绍了Golang中一些常用的if代码优化小技巧,w文中的示例代码简洁易懂,具有一定的借鉴价值,感兴趣的小伙伴可以跟随小编一起学习一下
2023-10-10
Go语言实现服务端消息接收和发送
这篇文章主要为大家详细介绍了Go语言实现服务端消息接收和发送功能，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2022-07-07
mayfly-go部署和使用详解
这篇文章主要介绍了mayfly-go部署和使用详解,此处部署基于CentOS7.4部署，结合实例代码图文给大家讲解的非常详细，需要的朋友可以参考下
2022-09-09
Go语言参数传递是传值还是传引用
Go 语言到底是传值（值传递），还是传引用（引用传递）？本文就详细介绍一下，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-12-12
golang 常用定时任务汇总
这篇文章主要介绍了golang 常用定时任务汇总，golang利用goroutine外加github.com/gorhill/cronexpr库就可实现定时任务，具体代码介绍，需要的小伙伴可以参考一下
2022-09-09
golang实现并发控制的方法和技巧
golang 是一门支持并发的编程语言,它提供了 goroutine 和 channel 等强大的特性,让我们可以轻松地创建和管理多个执行单元,实现高效的任务处理,在本文中,我们将介绍一些 golang 的并发控制的方法和技巧,希望对你有所帮助
2024-03-03
Go 1.18新特性之泛型的全面讲解
本文力求能让未接触过泛型编程的人也能较好理解Go的泛型，所以行文可能略显啰嗦。但是请相信我，看完这篇文章你能获得对Go泛型非常全面的了解
2023-03-03
使用go语言解析xml的实现方法(必看篇)
下面小编就为大家带来一篇使用go语言解析xml的实现方法(必看篇)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-06-06
Go 语言数据结构之双链表学习教程
这篇文章主要为大家介绍了Go 语言数据结构之双链表学习教程详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-08-08