gin自定義中間件解決requestBody不可重復(fù)讀問(wèn)題(最新推薦)
先直接上代碼
r := gin.Default()
// 注冊(cè)中間件，使body可以重復(fù)讀取
r.Use(func(context *gin.Context) {
all, err := context.GetRawData() // 讀取body的內(nèi)容
if err != nil {
log.Fatal(err)
}
// 重寫(xiě) GetBody 方法，以便后續(xù)的其他操作
context.Request.GetBody = func() (io.ReadCloser, error) {
context.Request.Body = io.NopCloser(bytes.NewBuffer(all))
buffer := bytes.NewBuffer(all)
closer := io.NopCloser(buffer)
return closer, nil
}
body, _ := context.Request.GetBody() // 每次調(diào)用GetBody方法，都會(huì)新生成一個(gè)io.ReadCloser，但是底層的byte數(shù)據(jù)，都是all變量緩存的。
context.Request.Body = body
context.Next()
})注意，上面的中間件，需要在第一個(gè)執(zhí)行。
分析
在gin中，context.Request.Body 是一個(gè)io.ReadCloser的接口，如下圖
查看io.ReadCloser接口定義
type ReadCloser interface {
Reader
Closer
}
type Reader interface {
Read(p []byte) (n int, err error)
}
type Closer interface {
Close() error
}我們發(fā)現(xiàn)io.ReaderCloser接口的本質(zhì)就是Read(p []byte) (n int, err error) 和 Close() error 的組合。
所以我們只需要自己編寫(xiě)實(shí)現(xiàn)Read(p []byte) (n int, err error) 和 Close() error 這兩個(gè)方法的結(jié)構(gòu)體即可賦值給context.Request.Body，在我們自己實(shí)現(xiàn)的方法中實(shí)現(xiàn)可重復(fù)讀取即可達(dá)到我們的目的。
到此這篇關(guān)于gin自定義中間件解決requestBody不可重復(fù)讀問(wèn)題的文章就介紹到這了,更多相關(guān)requestBody不可重復(fù)讀內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
淺析Go中函數(shù)的健壯性,panic異常處理和defer機(jī)制
這篇文章主要為大家詳細(xì)介紹了Go中函數(shù)的健壯性,panic異常處理和defer機(jī)制的相關(guān)知識(shí),文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解一下2023-10-10
Golang開(kāi)發(fā)Go依賴(lài)管理工具dep安裝驗(yàn)證實(shí)現(xiàn)過(guò)程
這篇文章主要為大家介紹了Golang開(kāi)發(fā)Go依賴(lài)管理工具dep安裝驗(yàn)證及初始化一系列實(shí)現(xiàn)過(guò)程，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步2021-11-11
go語(yǔ)言使用第三方包 json化結(jié)構(gòu)體操作示例
這篇文章主要介紹了go語(yǔ)言使用第三方包 json化結(jié)構(gòu)體操作,結(jié)合實(shí)例形式分析了Go語(yǔ)言ffjson包git安裝及結(jié)構(gòu)體轉(zhuǎn)json字符串相關(guān)操作技巧,需要的朋友可以參考下2019-06-06
Go語(yǔ)言模型:string的底層數(shù)據(jù)結(jié)構(gòu)與高效操作詳解
這篇文章主要介紹了Go語(yǔ)言模型:string的底層數(shù)據(jù)結(jié)構(gòu)與高效操作詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-12-12
本文主要講述SetMaxOpenConns(),?SetMaxIdleConns()?和?SetConnMaxLifetime()方法，?您可以使用它們來(lái)配置sql.DB的行為并改變其性能，感興趣的可以了解一下2021-12-12
Go中的fuzz模糊測(cè)試使用實(shí)戰(zhàn)詳解
這篇文章主要為大家介紹了Go中的fuzz模糊測(cè)試使用實(shí)戰(zhàn)詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-12-12
拾遺主要是收集和golang相關(guān)的瑣碎知識(shí)，這篇文章主要是為大家整理了Golang中指針和接口的使用方法，文中的示例代碼講解詳細(xì)，需要的可以參考一下2023-02-02
Golang實(shí)現(xiàn)事務(wù)型內(nèi)存數(shù)據(jù)庫(kù)的方法詳解
內(nèi)存數(shù)據(jù)庫(kù)經(jīng)我們經(jīng)常用到，例如Redis，那么如何從零實(shí)現(xiàn)一個(gè)內(nèi)存數(shù)據(jù)庫(kù)呢，本文旨在介紹如何使用Golang編寫(xiě)一個(gè)KV內(nèi)存數(shù)據(jù)庫(kù)MossDB2023-03-03
這篇文章主要為大家詳細(xì)介紹了如何利用Go語(yǔ)言實(shí)現(xiàn)分金幣游戲，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
Go?編程復(fù)雜數(shù)據(jù)類(lèi)型?Map
這篇文章主要介紹了Go編程復(fù)雜數(shù)據(jù)類(lèi)型Map，Go中的Map是一組無(wú)需的K-V類(lèi)型的數(shù)據(jù)，與Python中的字典Dict和Java中的HashMap結(jié)構(gòu)類(lèi)似。未被初始化的Map為nil2022-08-08