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，在我們自己實現的方法中實現可重覆讀取即可達到我們的目的。