feat(): add max file size for upload
This commit is contained in:
parent
ea6a7b272a
commit
48528f3a56
9
main.go
9
main.go
@ -12,8 +12,9 @@ import (
|
||||
)
|
||||
|
||||
const (
|
||||
port = ":8080"
|
||||
filesDir = "./files"
|
||||
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()
|
||||
|
Loading…
Reference in New Issue
Block a user