gin怎么自定義中間件解決requestBody不可重復讀問題

今天小編給大家分享一下gin怎么自定義中間件解決requestBody不可重復讀問題的相關知識點，內容詳細，邏輯清晰，相信大部分人都還太了解這方面的知識，所以分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后有所收獲，下面我們一起來了解一下吧。

先直接上代碼

r := gin.Default()
	// 注冊中間件，使body可以重復讀取
	r.Use(func(context *gin.Context) {
		all, err := context.GetRawData() // 讀取body的內容
		if err != nil {
			log.Fatal(err)
		}
		// 重寫 GetBody 方法，以便后續的其他操作
		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()  // 每次調用GetBody方法，都會新生成一個io.ReadCloser，但是底層的byte數據，都是all變量緩存的。
		context.Request.Body = body
		context.Next()
	})

注意，上面的中間件，需要在第一個執行。

分析

在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
}

我們發現io.ReaderCloser接口的本質就是Read(p []byte) (n int, err error) 和 Close() error 的組合。

所以我們只需要自己編寫實現Read(p []byte) (n int, err error) 和 Close() error 這兩個方法的結構體即可賦值給context.Request.Body，在我們自己實現的方法中實現可重復讀取即可達到我們的目的。

以上就是“gin怎么自定義中間件解決requestBody不可重復讀問題”這篇文章的所有內容，感謝各位的閱讀！相信大家閱讀完這篇文章都有很大的收獲，小編每天都會為大家更新不同的知識，如果還想學習更多的知識，請關注億速云行業資訊頻道。