feat(): add max file size for upload

This commit is contained in:
Lucas Barbieri 2024-08-19 16:39:57 -03:00
parent ea6a7b272a
commit 48528f3a56

View File

@ -14,6 +14,7 @@ import (
const (
port = ":8080"
filesDir = "./files"
maxFileSize = 20 * 1024 * 1024 // 20MiB
)
var url string = os.Getenv("URL")
@ -41,9 +42,11 @@ func uploadHandler(w http.ResponseWriter, r *http.Request) {
return
}
r.Body = http.MaxBytesReader(w, r.Body, maxFileSize)
file, _, err := r.FormFile("file")
if err != nil {
http.Error(w, "Error retrieving the file", http.StatusBadRequest)
http.Error(w, "Error retrieving the file or file size exceeds limit", http.StatusBadRequest)
return
}
defer file.Close()