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) } // 重寫 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
的組合。
所以我們只需要自己編寫實(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)文章
Golang棧結(jié)構(gòu)和后綴表達(dá)式實(shí)現(xiàn)計(jì)算器示例
這篇文章主要為大家介紹了Golang棧結(jié)構(gòu)和后綴表達(dá)式實(shí)現(xiàn)計(jì)算器示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-07-07Go生成base64圖片驗(yàn)證碼實(shí)例(超詳細(xì)工具類)
這段時(shí)間需要使用圖片驗(yàn)證碼庫(kù),下面這篇文章主要給大家介紹了關(guān)于Go生成base64圖片驗(yàn)證碼的相關(guān)資料,文中給出了詳細(xì)的實(shí)例代碼,需要的朋友可以參考下2023-06-06Golang實(shí)現(xiàn)Redis事務(wù)深入探究
這篇文章主要介紹了Golang實(shí)現(xiàn)Redis事務(wù)深入探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01Go語(yǔ)言中利用http發(fā)起Get和Post請(qǐng)求的方法示例
這篇文章主要給大家介紹了關(guān)于Go語(yǔ)言中利用http發(fā)起Get和Post請(qǐng)求的相關(guān)資料,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧。2017-11-11Go語(yǔ)言開(kāi)發(fā)k8s之Service操作解析
這篇文章主要為大家介紹了Go語(yǔ)言開(kāi)發(fā)k8s之Service操作解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-06-06Go擴(kuò)展原語(yǔ)之SingleFlight的用法詳解
Go語(yǔ)言擴(kuò)展包同步原語(yǔ)singleflight.Group能夠再一個(gè)服務(wù)中抑制對(duì)下游的多次重復(fù)請(qǐng)求,它能夠限制對(duì)同一個(gè)鍵值對(duì)的多次重復(fù)請(qǐng)求,減少對(duì)下游的瞬時(shí)流量,接下來(lái)小編就給大家講講Go SingleFlight的具體用法,需要的朋友可以參考下2023-07-07