在Golang开发中,实现文件上传到服务器需要结合HTTP协议特性和标准库的相关功能,同时针对不同的业务场景做性能优化,才能满足实际的使用需求。

Golang实现基础文件上传
前端表单配置
文件上传需要前端表单设置正确的编码类型,否则后端无法正确解析文件数据,表单的核心配置如下:
<form action="/upload" method="post" enctype="multipart/form-data">
<input type="file" name="uploadFile">
<input type="submit" value="上传文件">
</form>
后端接收处理逻辑
Golang的net/http库提供了处理multipart表单的能力,核心是通过r.ParseMultipartForm解析请求,再从表单中获取文件句柄,完整实现代码如下:
package main
import (
"fmt"
"io"
"net/http"
"os"
)
func uploadHandler(w http.ResponseWriter, r *http.Request) {
// 解析multipart表单,最大内存设置为10MB,超出部分会存储到临时文件
err := r.ParseMultipartForm(10 << 20)
if err != nil {
http.Error(w, "解析表单失败", http.StatusBadRequest)
return
}
// 获取上传的文件
file, header, err := r.FormFile("uploadFile")
if err != nil {
http.Error(w, "获取文件失败", http.StatusBadRequest)
return
}
defer file.Close()
// 创建目标文件
dst, err := os.Create("./uploads/" + header.Filename)
if err != nil {
http.Error(w, "创建文件失败", http.StatusInternalServerError)
return
}
defer dst.Close()
// 拷贝文件内容
_, err = io.Copy(dst, file)
if err != nil {
http.Error(w, "保存文件失败", http.StatusInternalServerError)
return
}
fmt.Fprintf(w, "文件上传成功,文件名:%s,大小:%d字节", header.Filename, header.Size)
}
func main() {
// 创建上传目录
os.MkdirAll("./uploads", 0755)
http.HandleFunc("/upload", uploadHandler)
http.ListenAndServe(":8080", nil)
}
文件上传性能优化方案
限制上传文件大小
为了避免恶意上传大文件占用服务器资源,可以通过http.MaxBytesReader限制请求体的最大大小,代码如下:
func uploadHandler(w http.ResponseWriter, r *http.Request) {
// 限制请求体最大为20MB
r.Body = http.MaxBytesReader(w, r.Body, 20<<20)
// 后续解析逻辑和之前一致
}
大文件分片上传
对于超过100MB的大文件,建议采用分片上传的方式,将文件拆分为多个小块分别传输,后端再合并分片,这样可以避免单次传输失败导致全部重传,分片上传的后端处理逻辑如下:
package main
import (
"fmt"
"io"
"net/http"
"os"
"strconv"
)
func chunkUploadHandler(w http.ResponseWriter, r *http.Request) {
// 获取分片参数
chunkIndex, _ := strconv.Atoi(r.FormValue("chunkIndex"))
totalChunks, _ := strconv.Atoi(r.FormValue("totalChunks"))
fileName := r.FormValue("fileName")
// 获取当前分片文件
file, _, err := r.FormFile("chunk")
if err != nil {
http.Error(w, "获取分片失败", http.StatusBadRequest)
return
}
defer file.Close()
// 创建临时分片存储目录
os.MkdirAll("./chunks/"+fileName, 0755)
chunkPath := fmt.Sprintf("./chunks/%s/chunk_%d", fileName, chunkIndex)
dst, err := os.Create(chunkPath)
if err != nil {
http.Error(w, "保存分片失败", http.StatusInternalServerError)
return
}
defer dst.Close()
io.Copy(dst, file)
// 如果是最后一个分片,合并所有分片
if chunkIndex == totalChunks-1 {
finalPath := "./uploads/" + fileName
finalFile, _ := os.Create(finalPath)
defer finalFile.Close()
for i := 0; i < totalChunks; i++ {
chunkPath := fmt.Sprintf("./chunks/%s/chunk_%d", fileName, i)
chunkFile, _ := os.Open(chunkPath)
io.Copy(finalFile, chunkFile)
chunkFile.Close()
os.Remove(chunkPath)
}
os.RemoveAll("./chunks/" + fileName)
fmt.Fprintf(w, "文件合并完成")
} else {
fmt.Fprintf(w, "分片%d上传成功", chunkIndex)
}
}
优化内存使用
默认情况下ParseMultipartForm会将小文件存储在内存中,大文件存到临时磁盘,可以通过调整内存阈值适配业务场景,同时及时关闭临时文件句柄,避免内存泄漏。另外可以开启Golang的HTTP服务的读写超时设置,避免长时间占用连接:
func main() {
os.MkdirAll("./uploads", 0755)
server := &http.Server{
Addr: ":8080",
ReadTimeout: 30 * 1000000000, // 30秒读超时
WriteTimeout: 30 * 1000000000, // 30秒写超时
}
http.HandleFunc("/upload", uploadHandler)
http.HandleFunc("/chunk_upload", chunkUploadHandler)
server.ListenAndServe()
}
常见问题与注意事项
- 上传目录需要提前创建,并且设置正确的读写权限,否则会出现文件创建失败的问题
- 文件名校验需要过滤特殊字符,避免路径遍历攻击,比如禁止文件名包含../等路径符号
- 生产环境建议将上传文件存储到对象存储服务,而不是服务器本地磁盘,避免磁盘空间不足的问题
- 可以结合Golang的并发控制,比如使用协程池处理多个上传请求,提升并发场景下的处理能力