Golang文件上传需解析multipart/form-data，使用r.ParseMultipartForm限制内存使用，通过r.FormFile获取文件，验证文件类型与大小，使用io.Copy流式写入避免内存溢出，错误时返回对应HTTP状态码并确保资源释放。
Golang文件上传涉及对
multipart/form-data的处理，核心在于解析HTTP请求中的表单数据，提取文件内容和元数据，并将其保存到服务器上。这需要理解
multipart的结构和使用Golang标准库中的
mime/multipart包。
解决方案
首先，你需要一个处理上传请求的HTTP handler。在这个handler中，你需要调用
r.ParseMultipartForm来解析请求体。这个函数会将上传的文件和表单字段解析到内存中（或者磁盘上，如果文件太大）。
package main
import (
"fmt"
"io"
"log"
"net/http"
"os"
)
func uploadFile(w http.ResponseWriter, r *http.Request) {
// 限制上传大小，例如 10MB
err := r.ParseMultipartForm(10 << 20)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
// 获取上传的文件
file, handler, err := r.FormFile("myFile") // "myFile" 是 HTML 表单中 file input 的 name 属性
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
defer file.Close()
fmt.Fprintf(w, "Uploaded File: %+v\n", handler.Filename)
fmt.Fprintf(w, "File Size: %+v bytes\n", handler.Size)
fmt.Fprintf(w, "MIME Header: %+v\n", handler.Header)
// 创建文件
dst, err := os.Create("./uploads/" + handler.Filename)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer dst.Close()
// 将上传的文件内容拷贝到新创建的文件
if _, err := io.Copy(dst, file); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
fmt.Fprintf(w, "Successfully Uploaded File\n")
}
func setupRoutes() {
http.HandleFunc("/upload", uploadFile)
// 静态文件服务
fs := http.FileServer(http.Dir("./uploads"))
http.Handle("/files/", http.StripPrefix("/files/", fs))
log.Fatal(http.ListenAndServe(":8080", nil))
}
func main() {
// 确保 uploads 目录存在
if _, err := os.Stat("./uploads"); os.IsNotExist(err) {
os.Mkdir("./uploads", os.ModeDir|0755)
}
fmt.Println("Server listening on port 8080")
setupRoutes()
}这段代码首先解析
multipart/form-data，然后从表单中提取文件，创建目标文件，并将上传的文件内容复制到目标文件中。注意错误处理，这对于健壮的应用程序至关重要。同时，为了简单起见，这里直接使用了文件名作为存储文件名，实际应用中应该生成唯一的文件名，避免覆盖。
立即学习“go语言免费学习笔记（深入）”；
如何处理大型文件上传？
处理大型文件上传时，将整个文件加载到内存中是不现实的。
r.ParseMultipartForm函数允许你指定一个最大内存大小，超过这个大小的文件会被写入临时文件。你可以通过检查
r.MultipartForm.File来确定文件是否被写入了磁盘。
为了更高效地处理大型文件，可以使用
io.Copy函数直接将上传的文件流复制到目标文件。避免一次性读取整个文件到内存中。
此外，可以考虑使用分块上传技术，将大文件分割成小块，逐个上传。这可以提高上传的稳定性和可恢复性，特别是在网络环境不稳定的情况下。
如何验证上传文件的类型和大小？
验证上传文件的类型和大小是安全性的重要组成部分。你可以通过检查
handler.Header中的
Content-Type字段来确定文件的 MIME 类型。
contentType := handler.Header.Get("Content-Type")
if contentType != "image/jpeg" && contentType != "image/png" {
http.Error(w, "Invalid file type", http.StatusBadRequest)
return
}同时，你可以通过
handler.Size字段获取文件的大小，并与允许的最大大小进行比较。
maxFileSize := int64(10 << 20) // 10MB
if handler.Size > maxFileSize {
http.Error(w, "File too large", http.StatusBadRequest)
return
}这些验证步骤应该在将文件保存到磁盘之前执行，以避免存储无效或恶意文件。
如何处理上传过程中的错误？
文件上传过程中可能发生各种错误，例如磁盘空间不足、权限问题、网络中断等。良好的错误处理可以提高应用程序的健壮性。
在每个可能出错的地方，都应该检查错误，并采取适当的措施。例如，如果创建文件失败，应该返回一个 HTTP 500 错误，并记录错误日志。如果复制文件内容失败，应该删除已创建的文件，以避免残留的垃圾文件。
使用
defer语句可以确保资源在使用完毕后被释放，例如关闭文件。这可以避免资源泄漏。同时，应该向客户端返回清晰的错误信息，方便用户排查问题。
