gin自定義中間件解決requestBody不可重復(fù)讀問題(最新推薦)
更新時間：2023年04月29日 08:40:44 作者：咸蛋solo粥
這篇文章主要介紹了gin自定義中間件解決requestBody不可重復(fù)讀問題,本文通過示例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
先直接上代碼
r := gin.Default()
// 注冊中間件，使body可以重復(fù)讀取
r.Use(func(context *gin.Context) {
all, err := context.GetRawData() // 讀取body的內(nèi)容
if err != nil {
log.Fatal(err)
}
// 重寫 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方法，都會新生成一個io.ReadCloser，但是底層的byte數(shù)據(jù)，都是all變量緩存的。
context.Request.Body = body
context.Next()
})注意，上面的中間件，需要在第一個執(zhí)行。
分析
在gin中，context.Request.Body 是一個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àn)Read(p []byte) (n int, err error) 和 Close() error 這兩個方法的結(jié)構(gòu)體即可賦值給context.Request.Body，在我們自己實現(xiàn)的方法中實現(xiàn)可重復(fù)讀取即可達(dá)到我們的目的。
到此這篇關(guān)于gin自定義中間件解決requestBody不可重復(fù)讀問題的文章就介紹到這了,更多相關(guān)requestBody不可重復(fù)讀內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
內(nèi)存泄漏是編程中常見的問題,會對程序的性能和穩(wěn)定性產(chǎn)生嚴(yán)重影響,本文將深入詳解?Golang?中的內(nèi)存泄漏的原因、檢測方法以及避免方法,希望對大家有所幫助2023-12-12
Map是一個強(qiáng)大而又有趣的工具，它可以幫助我們高效地存儲和操作鍵值對數(shù)據(jù)，本文主要介紹了Go語言中Map的各種操作，包括增加、查找、刪除、遍歷等，具有一定的參考價值，感興趣的可以了解一下2023-08-08
Golang棧結(jié)構(gòu)和后綴表達(dá)式實現(xiàn)計算器示例
這篇文章主要為大家介紹了Golang棧結(jié)構(gòu)和后綴表達(dá)式實現(xiàn)計算器示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-07-07
這段時間需要使用圖片驗證碼庫,下面這篇文章主要給大家介紹了關(guān)于Go生成base64圖片驗證碼的相關(guān)資料,文中給出了詳細(xì)的實例代碼,需要的朋友可以參考下2023-06-06
Channel是go語言內(nèi)置的一個非常重要的特性,也是go并發(fā)編程的兩大基石之一,下面這篇文章主要給大家介紹了關(guān)于Go中channel的相關(guān)資料,需要的朋友可以參考下2023-02-02
這篇文章主要介紹了Golang實現(xiàn)Redis事務(wù)深入探究，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2024-01-01
這篇文章主要為大家介紹了Go代碼檢查的推薦工具及使用詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-07-07
Go語言中利用http發(fā)起Get和Post請求的方法示例
這篇文章主要給大家介紹了關(guān)于Go語言中利用http發(fā)起Get和Post請求的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧。2017-11-11
這篇文章主要為大家介紹了Go語言開發(fā)k8s之Service操作解析，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-06-06
Go語言擴(kuò)展包同步原語singleflight.Group能夠再一個服務(wù)中抑制對下游的多次重復(fù)請求，它能夠限制對同一個鍵值對的多次重復(fù)請求，減少對下游的瞬時流量，接下來小編就給大家講講Go SingleFlight的具體用法,需要的朋友可以參考下2023-07-07