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

更新時間：2023年04月10日 08:34:10 作者：中國流浪貓

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

HTTP上傳的文件的原理

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

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

Gin框架文件上傳Demo

代碼邏輯：

通過Gin框架封裝的Form表單獲取數(shù)據(jù)，獲取上傳文件
獲取文件名，并創(chuàng)建新的文件存儲
將上傳的文件內(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)建新的文件存儲
	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ù)來限制 HTTP 請求中讀取的最大字節(jié)數(shù)。這個函數(shù)會返回一個新的 Reader 對象，該對象會在讀取請求的正文時自動檢查字節(jié)數(shù)，如果超過指定的最大字節(jié)數(shù)，則會自動停止讀取，返回錯誤。

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

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

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

運行結(jié)果截圖

文件類型驗證

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

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")
}

文件上傳進度-后臺計算文件上傳進度

實現(xiàn)原理：

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

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，默認32M
	//路由:http://localhost:8080/upload
	router.POST("/upload", uploadFile)
	fmt.Println(router.MaxMultipartMemory)
	router.Run(":8080")
}

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

您可能感興趣的文章: