在Golang中实现文件上传到服务器,主要依赖标准库的net/http和mime/multipart包,不需要引入第三方依赖即可完成基础功能,适合大多数中小型项目的需求。实现过程主要分为前端表单配置和后端逻辑处理两部分,需要注意请求体的解析和文件的存储逻辑。

前端表单配置
文件上传的前端需要使用form表单,并且设置正确的编码类型,否则后端无法正确解析上传的文件内容。表单需要满足以下几个要求:
- 表单的
method属性设置为post enctype属性设置为multipart/form-data- 添加
input标签,类型为file,用于选择上传的文件
以下是一个简单的前端上传表单示例:
<!DOCTYPE html>
<html lang="zh-CN">
<head>
<meta charset="UTF-8">
<title>文件上传</title>
</head>
<body>
<form action="/upload" method="post" enctype="multipart/form-data">
<input type="file" name="uploadFile">
<button type="submit">上传文件</button>
</form>
</body>
</html>
后端核心实现步骤
1. 设置路由和处理函数
首先需要使用net/http包设置对应的上传路由,并且编写处理函数。为了避免上传大文件占用过多内存,需要提前设置请求体的最大大小限制。
2. 解析multipart请求
使用r.ParseMultipartForm方法解析请求体,该方法接收一个参数表示最大内存使用量,超过这个大小的文件部分会存储到临时文件中。
3. 获取上传文件并存储
从解析后的请求中获取文件句柄,读取文件内容后写入到服务器的目标路径中,完成文件存储。
以下是完整的后端实现代码示例:
package main
import (
"fmt"
"io"
"net/http"
"os"
"path/filepath"
)
// 上传文件处理函数
func uploadHandler(w http.ResponseWriter, r *http.Request) {
// 限制请求体大小为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()
// 限制允许上传的文件类型
allowedTypes := map[string]bool{
"image/jpeg": true,
"image/png": true,
"text/plain": true,
}
if !allowedTypes[header.Header.Get("Content-Type")] {
http.Error(w, "不允许上传该类型文件", http.StatusBadRequest)
return
}
// 创建存储目录,如果不存在则创建
uploadDir := "./uploads"
if _, err := os.Stat(uploadDir); os.IsNotExist(err) {
os.MkdirAll(uploadDir, 0755)
}
// 生成目标文件路径,避免文件名冲突
filename := filepath.Base(header.Filename)
dstPath := filepath.Join(uploadDir, filename)
// 创建目标文件
dst, err := os.Create(dstPath)
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", dstPath)
}
func main() {
// 设置静态文件服务,用于返回前端上传页面
http.Handle("/", http.FileServer(http.Dir("./static")))
// 设置上传路由
http.HandleFunc("/upload", uploadHandler)
fmt.Println("服务器启动在 :8080 端口")
http.ListenAndServe(":8080", nil)
}
注意事项
- 生产环境中一定要对上传文件的类型和大小做严格限制,避免恶意用户上传可执行文件或者超大文件导致服务器资源耗尽
- 存储文件时不要直接使用用户上传的原始文件名,建议生成唯一的文件名,避免路径遍历攻击和文件名冲突
- 可以结合context设置请求超时时间,避免上传过程占用连接过久
- 如果是分布式部署,建议将上传的文件存储到对象存储服务中,而不是本地服务器磁盘
上述示例中的前端页面需要放在项目的static目录下,启动服务后访问http://ipipp.com:8080即可看到上传表单,选择文件后提交即可完成上传。