request_body_limit.go 318 B

123456789101112131415
  1. package middleware
  2. import (
  3. "net/http"
  4. "github.com/gin-gonic/gin"
  5. )
  6. // RequestBodyLimit 使用 MaxBytesReader 限制请求体大小。
  7. func RequestBodyLimit(maxBytes int64) gin.HandlerFunc {
  8. return func(c *gin.Context) {
  9. c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, maxBytes)
  10. c.Next()
  11. }
  12. }