快捷導(dǎo)航

Golang使用Gin框架實(shí)現(xiàn)HTTP上傳文件過程介紹

更新時(shí)間：2023年04月10日 08:34:10 作者：中國(guó)流浪貓

由于需求中有文件上傳這一個(gè)需求，在這里我們就學(xué)習(xí)一下go語(yǔ)言如何上傳文件。本文主要通過表單的方式進(jìn)行文件上傳操作，本文實(shí)例為大家分享了Go實(shí)現(xiàn)文件上傳操作的具體代碼，供大家參考，具體內(nèi)容如下

HTTP上傳的文件的原理

HTTP協(xié)議的文件上傳是通過HTTP POST請(qǐng)求實(shí)現(xiàn)的，使用multipart/form-data格式將待上傳的文件放入請(qǐng)求體中。

服務(wù)器根據(jù)請(qǐng)求頭中的boundary參數(shù)來(lái)解析請(qǐng)求體，并根據(jù)Content-Disposition字段獲取文件名等信息，根據(jù)Content-Type字段判斷文件類型并保存到相應(yīng)位置。

Gin框架文件上傳Demo

代碼邏輯：

通過Gin框架封裝的Form表單獲取數(shù)據(jù)，獲取上傳文件
獲取文件名，并創(chuàng)建新的文件存儲(chǔ)
將上傳的文件內(nèi)容寫入新的文件
返回上傳成功信息

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
	}
	// 獲取文件名，并創(chuàng)建新的文件存儲(chǔ)
	filename := header.Filename
	out, err := os.Create(filename)
	if err != nil {
		c.String(http.StatusBadRequest, fmt.Sprintf("創(chuàng)建文件: %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() 函數(shù)來(lái)限制 HTTP 請(qǐng)求中讀取的最大字節(jié)數(shù)。這個(gè)函數(shù)會(huì)返回一個(gè)新的 Reader 對(duì)象，該對(duì)象會(huì)在讀取請(qǐng)求的正文時(shí)自動(dòng)檢查字節(jié)數(shù)，如果超過指定的最大字節(jié)數(shù)，則會(huì)自動(dòng)停止讀取，返回錯(cuò)誤。

//限制大小為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
	}

該代碼不能限制文件上傳大小，只是設(shè)置內(nèi)存大小，即使文件大小比這個(gè)大，也會(huì)寫入臨時(shí)文件

router := gin.Default()
router.MaxMultipartMemory = 2 * 1024 //2M Byte，默認(rèn)32M

運(yùn)行結(jié)果截圖

文件類型驗(yàn)證

驗(yàn)證上傳的文件類型，以確保上傳的文件是我們期望的類型，借助“github.com/h2non/filetype”實(shí)現(xiàn)對(duì)文件類型的判斷

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)
	// 驗(yàn)證文件類型
	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")
}

文件上傳進(jìn)度-后臺(tái)計(jì)算文件上傳進(jìn)度

實(shí)現(xiàn)原理：

要實(shí)現(xiàn) Gin 框架中的文件上傳進(jìn)度，在文件上傳中，計(jì)算已上傳的字節(jié)數(shù)，并將其與文件的總大小進(jìn)行比較，以確定上傳的進(jìn)度。

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("創(chuàng)建文件: %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，默認(rèn)32M
	//路由:http://localhost:8080/upload
	router.POST("/upload", uploadFile)
	fmt.Println(router.MaxMultipartMemory)
	router.Run(":8080")
}

到此這篇關(guān)于Golang使用Gin框架實(shí)現(xiàn)HTTP上傳文件過程介紹的文章就介紹到這了,更多相關(guān)Go HTTP上傳文件內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: