Golang使用Gin框架实现HTTP上传文件过程介绍
作者:中国流浪猫 发布时间:2024-04-28 09:18:12
标签:Go,HTTP,Gin,上传文件
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")
}
来源:https://blog.csdn.net/a41888313/article/details/129979126
0
投稿
猜你喜欢
- 使用ENUM代替字符串类型有时候, 可以通过使用ENUM来代理常规的字符串类型。一个ENUM列能够存储65535个不同的字符串值,MySQL
- 题目[1]:格式输出练习。在交互式状态下完成以下练习。运行结果截图:题目[2]:格式输出练习。在.py的文件中完成以下练习代码:num =
- 如下所示:from osgeo import gdalimport numpy as npdef read_tiff(inpath): &n
- 本文针对Python的全局变量实现方法简述如下:先来看下面一段测试程序:count = 0def Fuc(count): pri
- 一、环境安装 1.安装apache2 sudo apt-get install apache2 安装后在浏览器中打开:http://loca
- 这篇文章主要介绍了python Jupyter运行时间实例过程解析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价
- 问题你想解析某个XML文档,文档中使用了XML命名空间。解决方案考虑下面这个使用了命名空间的文档:<?xml version=&quo
- 在我们学习python的过程中,学习序列是一门必修课。当我们掌握了序列过后,便会学习常用的两个排序函数sort()与sorted()。但很少
- python取余运算符是什么?python取余运算符是%,即表示取模,返回除法的余数。假设变量: a=10,b=20:那么b % a 输出结
- 概述何为路由守卫?路由守卫有点类似于ajax的请求 * ,就是请求发送之前先给你拦截住做一些事情之后再去发送请求,同样这里的路由守卫意思差不
- 经常会有小朋友问我,“我想做个黑客,我该学什么编程语言?”,或者有的小朋友会说:“我要学c,我要做病毒”。其实对于这些小朋友而言他们基本都没
- 1.选中数据库,右键属性/选项,把恢复模式从“完整”改为“简单”。 2.选中数据库,右键任务/收缩文件,选择日志文件,设置将文件收缩为0M,
- 向数据库加入字符类型的数据时,我老爱写错数据中含有单引号的语句而导致操作失败像:正确的SQL语句:INSERT baby 
- 本文实例为大家分享了小程序实现横向滑动日历效果的具体代码,供大家参考,具体内容如下<scroll-view class="s
- sql 使用系统存储过程 sp_send_dbmail 发送电子邮件语法:sp_send_dbmail [ [ @profile_name
- 算法思路Knuth-Morris-Pratt(KMP)算法是解决字符串匹配问题的经典算法,下面通过一个例子来演示一下:给定字符串"
- 朋友需要一个工具,将图片中的文字提取出来。我帮他在网上找了一些OCR的应用,都不好用。所以准备自己研究,写一个Web APP供他使用。OCR
- 背景重装系统,发现之前装在E盘的python可以直接使用,就只是将python的安装目录加入到环境变量中,也一直没有管它,今天跟天软交互的时
- 1.包: package PaintBrush; /** * * @author lucifer */ public class Paint
- 本文从算法入手同时通过实例来讲解如何在SQL Server数据库中实现最优最简的整个过程。算法是计算机科学中一个重要的研究方向,是解决复杂问