在 golang 项目中配置文件上传大小限制:设置 http.request 中 fileheader 类型属性的 maxfilesize 字段。创建文件上传处理程序并设置上传文件大小限制。解析多部分表单并提取上传文件。将文件保存到磁盘。返回成功响应或错误响应(如果文件大小超出限制)。
如何在 Golang 项目中配置文件上传大小限制
在 Golang 项目中,配置上传大小限制涉及设置 MaxFileSize 字段,该字段是 http.Request 的 FileHeader 类型属性。以下是如何实现的:
package main
import (
"fmt"
"log"
"net/http"
"os"
)
func main() {
// 设置上传文件大小限制为 10 MB
maxFileSize := int64(10 * 1024 * 1024)
// 创建文件上传处理程序
http.HandleFunc("/upload", func(w http.ResponseWriter, r *http.Request) {
// 检查请求是否是多部分表单
if r.Method == "POST" && r.MultipartForm == nil {
http.Error(w, "请求不是多部分表单", http.StatusBadRequest)
return
}
// 解析多部分表单
if err := r.ParseMultipartForm(maxFileSize); err != nil {
http.Error(w, fmt.Sprintf("解析表单时出错:%v", err), http.StatusInternalServerError)
return
}
// 提取上传文件
file, fileHeader, err := r.FormFile("file")
if err != nil {
http.Error(w, fmt.Sprintf("提取文件时出错:%v", err), http.StatusBadRequest)
return
}
defer file.Close()
// 将文件保存到磁盘
dst, err := os.Create("uploaded_file")
if err != nil {
http.Error(w, fmt.Sprintf("保存文件时出错:%v", err), http.StatusInternalServerError)
return
}
defer dst.Close()
if _, err = io.Copy(dst, file); err != nil {
http.Error(w, fmt.Sprintf("复制文件时出错:%v", err), http.StatusInternalServerError)
return
}
// 返回成功响应
fmt.Fprintf(w, "文件上传成功!")
})
// 启动服务器
log.Fatal(http.ListenAndServe(":8080", nil))
}
实战案例:
使用 Postman 或 cURL 来上传大于 10 MB 的文件,你将收到一个错误响应,如下所示:
{
"error": "文件大小超出了限制"
}
以上就是如何在 Golang 项目中配置文件上传大小限制?的详细内容,更多请关注编程网其它相关文章!