Golang处理文件上传下载：multipart表单与静态服务详解

如何用Go实现文件上传和下载功能？首先，使用net/http包处理multipart/form-data格式的文件上传，通过r.ParseMultipartForm限制内存大小，r.FormFile获取文件句柄，并用io.Copy保存到磁盘；其次，使用http.FileServer提供静态文件服务或将http.ServeFile用于精细控制下载行为；最后，注意文件大小限制、文件名安全、MIME类型识别等常见问题。具体步骤：1.解析multipart请求并限制上传大小；2.获取并保存上传文件；3.使用http.FileServer或http.ServeFile实现下载服务；4.设置Content-Disposition控制浏览器下载或预览；5.防范文件过大、文件名冲突、并发写入等问题。

在Web开发中，文件上传和下载是常见的功能需求。Golang作为一门性能优秀、并发友好的语言，在构建这类服务时非常合适。本文将重点介绍如何用Go处理文件上传（基于multipart表单）和文件下载（静态文件服务），并给出一些实用建议。

使用multipart/form-data处理文件上传

HTTP请求中，文件上传通常使用multipart/form-data格式发送。Go标准库中的net/http包已经内置了对这种格式的支持，我们可以通过以下步骤来处理上传：

• 解析请求中的multipart数据：使用r.ParseMultipartForm()方法，并传入一个最大内存大小参数，例如10 << 20表示最多读取10MB到内存。
• 获取上传的文件句柄：通过r.FormFile("file")获取文件对象，其中"file"是前端传递的字段名。
• 保存文件到磁盘：使用ioutil.TempDir创建临时目录，或者指定目标路径，然后用io.Copy将上传的文件写入本地。
• 注意清理资源：上传完成后记得关闭文件句柄，避免资源泄露。

示例代码片段如下：

func uploadHandler(w http.ResponseWriter, r *http.Request) {
    // 限制上传文件大小为10MB
    r.ParseMultipartForm(10 << 20)

    file, handler, err := r.FormFile("file")
    if err != nil {
        http.Error(w, "Error retrieving the file", http.StatusBadRequest)
        return
    }
    defer file.Close()

    // 创建目标文件
    dst, err := os.Create(handler.Filename)
    if err != nil {
        http.Error(w, "Unable to save the file", http.StatusInternalServerError)
        return
    }
    defer dst.Close()

    // 拷贝上传内容到目标文件
    if _, err := io.Copy(dst, file); err != nil {
        http.Error(w, "Error saving the file", http.StatusInternalServerError)
        return
    }

    fmt.Fprintf(w, "File %s uploaded successfully", handler.Filename)
}

注意：实际部署中应考虑文件重命名、路径安全、文件类型检查等问题。

提供静态文件下载服务

Go也非常适合用来做静态文件服务器，比如提供图片、文档等的下载服务。这可以通过http.FileServer结合http.Handle或http.ServeFile实现。

快速搭建静态文件服务器

如果你希望整个目录可访问，可以使用http.FileServer：

http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("public"))))
log.Fatal(http.ListenAndServe(":8080", nil))

这段代码会把public目录下的内容映射到/static/路径下，用户访问http://localhost:8080/static/filename即可下载对应文件。

精细控制文件下载行为

如果你想对每个文件下载进行更精细的控制，比如记录日志、验证权限、设置自定义响应头等，可以使用http.ServeFile函数：

func downloadHandler(w http.ResponseWriter, r *http.Request) {
    filename := r.URL.Query().Get("file")
    filePath := filepath.Join("uploads", filename)

    // 可以在这里添加权限判断逻辑

    w.Header().Set("Content-Disposition", "attachment")
    http.ServeFile(w, r, filePath)
}

这个例子中，我们强制浏览器下载而不是预览文件，通过设置Content-Disposition: attachment实现。

小技巧：如果想让某些文件支持在线预览（如PDF），可以省略该头部，浏览器会根据MIME类型决定是否直接展示。

常见问题与注意事项

• 上传文件过大导致阻塞：一定要限制上传文件的大小，否则可能耗尽服务器资源。
• 文件名冲突或不安全：上传前应对文件名进行过滤或重命名，防止出现路径穿越攻击（如../../）。
• MIME类型识别错误：对于下载服务，确保服务器能正确识别文件的MIME类型，否则可能导致浏览器处理异常。
• 并发写入冲突：多个用户同时上传同名文件时，需要考虑加锁或使用唯一标识符重命名。

基本上就这些。Go在这方面的标准库已经足够强大，只要稍作封装就能满足大多数场景的需求。