Golang使用Gin框架實(shí)現(xiàn)HTTP上傳文件過程介紹
HTTP上傳的文件的原理
HTTP協(xié)議的文件上傳是通過HTTP POST請(qǐng)求實(shí)現(xiàn)的,使用multipart/form-data格式將待上傳的文件放入請(qǐng)求體中。
服務(wù)器根據(jù)請(qǐng)求頭中的boundary參數(shù)來解析請(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ù)來限制 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)文章希望大家以后多多支持腳本之家!
相關(guān)文章
基于GORM實(shí)現(xiàn)CreateOrUpdate方法詳解
這篇文章主要為大家介紹了基于GORM實(shí)現(xiàn)CreateOrUpdate方法詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-10-10golang實(shí)現(xiàn)mysql數(shù)據(jù)庫(kù)事務(wù)的提交與回滾
這篇文章主要介紹了golang實(shí)現(xiàn)mysql數(shù)據(jù)庫(kù)事務(wù)的提交與回滾,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2021-04-04go同步原語(yǔ)Phaser和Barrier區(qū)別
這篇文章主要為大家介紹了通過java講解go同步原語(yǔ)Phaser和Barrier區(qū)別,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-12-12go slice 擴(kuò)容實(shí)現(xiàn)原理源碼解析
這篇文章主要為大家介紹了go slice 擴(kuò)容實(shí)現(xiàn)原理源碼解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-01-01詳解golang channel有無(wú)緩沖區(qū)的區(qū)別
這篇文章主要給大家介紹了golang channel有無(wú)緩沖區(qū)的區(qū)別,無(wú)緩沖是同步的,有緩沖是異步的,文中通過代碼示例給大家講解的非常詳細(xì),需要的朋友可以參考下2024-01-01Golang因Channel未關(guān)閉導(dǎo)致內(nèi)存泄漏的解決方案詳解
這篇文章主要為大家詳細(xì)介紹了當(dāng)Golang因Channel未關(guān)閉導(dǎo)致內(nèi)存泄漏時(shí)蓋如何解決,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解一下2023-07-07golang中兩個(gè)協(xié)程交替打印數(shù)字和字母的實(shí)現(xiàn)
這篇文章給大家介紹了golang中兩個(gè)協(xié)程交替打印數(shù)字和字母的實(shí)現(xiàn),文中通過代碼示例講解的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下2024-01-01Golang基于JWT與Casbin身份驗(yàn)證授權(quán)實(shí)例詳解
這篇文章主要為大家介紹了Golang基于JWT與Casbin實(shí)現(xiàn)身份驗(yàn)證授權(quán)實(shí)例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-08-08加速開發(fā):使用Go語(yǔ)言和Gin框架構(gòu)建Web項(xiàng)目的利器
Go語(yǔ)言和Gin框架是構(gòu)建高性能Web項(xiàng)目的利器,Go語(yǔ)言的簡(jiǎn)潔性和并發(fā)性,以及Gin框架的輕量級(jí)和快速路由能力,使開發(fā)者能夠快速構(gòu)建可靠的Web應(yīng)用程序,需要的朋友可以參考下2023-09-09