رفتن به محتوا

محدودیت حجم آپلود

از http.MaxBytesReader برای محدود کردن دقیق حداکثر حجم فایل‌های آپلود شده استفاده کنید. وقتی از محدودیت تجاوز شود، reader خطا برمی‌گرداند و می‌توانید با وضعیت 413 Request Entity Too Large پاسخ دهید.

این برای جلوگیری از حملات انکار سرویس مهم است، جایی که کلاینت‌ها فایل‌های بسیار بزرگ ارسال می‌کنند تا حافظه یا فضای دیسک سرور را تمام کنند.

نحوه کار

تعریف محدودیت — یک ثابت MaxUploadSize (۱ مگابایت) سقف سخت آپلودها را تعیین می‌کند.
اعمال محدودیت — http.MaxBytesReader بدنه c.Request.Body را بسته‌بندی می‌کند. اگر کلاینت بیش از حد مجاز بایت ارسال کند، reader متوقف شده و خطا برمی‌گرداند.
تجزیه و بررسی — c.Request.ParseMultipartForm خواندن را فعال می‌کند. کد *http.MaxBytesError را بررسی می‌کند تا وضعیت 413 با پیام واضح برگرداند.

package main

import (
  "fmt"
  "net/http"

  "github.com/gin-gonic/gin"
)

const (
  MaxUploadSize = 1 << 20 // 1 MB
)

func uploadHandler(c *gin.Context) {
  // Wrap the body reader so only MaxUploadSize bytes are allowed
  c.Request.Body = http.MaxBytesReader(c.Writer, c.Request.Body, MaxUploadSize)

  // Parse multipart form
  if err := c.Request.ParseMultipartForm(MaxUploadSize); err != nil {
    if _, ok := err.(*http.MaxBytesError); ok {
      c.JSON(http.StatusRequestEntityTooLarge, gin.H{
        "error": fmt.Sprintf("file too large (max: %d bytes)", MaxUploadSize),
      })
      return
    }
    c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
    return
  }

  file, _, err := c.Request.FormFile("file")
  if err != nil {
    c.JSON(http.StatusBadRequest, gin.H{"error": "file form required"})
    return
  }
  defer file.Close()

  c.JSON(http.StatusOK, gin.H{
    "message": "upload successful",
  })
}

func main() {
  r := gin.Default()
  r.POST("/upload", uploadHandler)
  r.Run(":8080")
}

تست

# Upload a small file (under 1 MB) -- succeeds
curl -X POST http://localhost:8080/upload \
  -F "file=@/path/to/small-file.txt"
# Output: {"message":"upload successful"}

# Upload a large file (over 1 MB) -- rejected
curl -X POST http://localhost:8080/upload \
  -F "file=@/path/to/large-file.zip"
# Output: {"error":"file too large (max: 1048576 bytes)"}

همچنین ببینید