22 lines
414 B
Go
22 lines
414 B
Go
package middleware
|
||
|
||
import (
|
||
"bytes"
|
||
"io"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// CacheBody 缓存 request body（body 只能读一次，中间件提前读取并缓存）
|
||
func CacheBody() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
body, err := io.ReadAll(c.Request.Body)
|
||
if err != nil {
|
||
body = []byte{}
|
||
}
|
||
c.Request.Body = io.NopCloser(bytes.NewBuffer(body))
|
||
c.Set("raw_body", body)
|
||
c.Next()
|
||
}
|
||
}
|