如何在Golang中上传文件到服务器_Golang 文件上传实践

答案是实现Golang文件上传需处理multipart请求并保存文件。首先设置HTTP路由,使用ParseMultipartForm解析请求,通过r.FormFile获取文件,再用io.Copy保存到指定路径,最后关闭文件句柄并返回响应。

在Golang中实现文件上传到服务器是一个常见需求,比如用户上传头像、文档或附件。实现过程并不复杂,核心是处理HTTP的multipart/form-data请求,并将接收到的文件保存到服务端指定位置。

处理前端文件上传请求

浏览器通过表单提交文件时,通常使用multipart/form-data编码格式。Go的net/http包能自动解析这种请求。你只需要设置路由并调用ParseMultipartForm方法。

示例代码:

main.go

package main

import (
    "io"
    "net/http"
    "os"
    "path/filepath"
)

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    // 只允许POST方法
    if r.Method != "POST" {
        http.Error(w, "只允许POST请求", http.StatusMethodNotAllowed)
        return
    }

    // 解析 multipart 表单,最大内存 32MB
    err := r.ParseMultipartForm(32 << 20)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }

    // 获取名为 "file" 的文件字段
    file, handler, err := r.FormFile("file")
    if err != nil {
        http.Error(w, "无法获取上传的文件", http.StatusBadRequest)
        return
    }
    defer file.Close()

    // 创建本地保存文件
    dst, err := os.Create(filepath.Join("uploads", handler.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
    }

    w.Write([]byte("文件上传成功: " + handler.Filename))
}

func main() {
    // 确保 uploads 目录存在
    os.MkdirAll("uploads", os.ModePerm)

    http.HandleFunc("/upload", uploadHandler)
    http.Handle("/", http.FileServer(http.Dir("."))) // 静态页面用于测试

    http.ListenAndServe(":8080", nil)
}

前端HTML测试页面

为了测试上传功能,可以创建一个简单的HTML表单:

index.html

将该HTML放在项目根目录下,访问http://localhost:8080即可看到上传界面。

安全与优化建议

在生产环境中,直接保存原始文件名和不加验证可能带来风险。以下是几个实用建议:

  • 重命名文件:使用UUID或时间戳避免覆盖和路径注入问题
  • 限制文件大小:在ParseMultipartForm中设置合理上限
  • 检查文件类型:通过magic number判断真实MIME类型,而非依赖扩展名
  • 防病毒扫描(可选):对重要文件集成杀毒服务
  • 设置权限:保存文件时使用os.Create后调整权限为0644

支持多文件上传

若需同时上传多个文件,可使用r.MultipartForm.File获取所有文件:

for _, fHeaders := range r.MultipartForm.File["files"] {
    file, err := fHeaders.Open()
    if err != nil { continue }
    defer file.Close()

    dst, _ := os.Create(filepath.Join("uploads", fHeaders.Filename))
    defer dst.Close()
    io.Copy(dst, file)
}

基本上就这些。Golang标准库已足够应对大多数文件上传场景,无需引入额外框架。只要注意边界情况和安全性,就能稳定运行。