Golang HTTP 大文件上传处理详解

本文档详细介绍了如何使用 Golang 的 net/http 包处理大文件上传。通过 ParseMultipartForm 函数,我们可以将上传的文件存储在内存或磁盘中,有效应对 Expect: 100 Continue 请求,并提供了前端 JavaScript 示例,帮助开发者构建稳定可靠的大文件上传服务。

在构建 Web 服务器时,处理大文件上传是一个常见的需求。Golang 的 net/http 包提供了强大的功能,可以轻松实现这一目标。本文将深入探讨如何使用 net/http 包处理大文件上传,包括处理 Expect: 100 Continue 请求,以及前端的配合。

后端实现:使用 ParseMultipartForm 处理上传

net/http 包中的 ParseMultipartForm 函数是处理 multipart/form-data 请求的关键。它可以将整个请求体解析为多个部分,并将文件部分存储在内存或磁盘上。

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "log"
)

func handler(w http.ResponseWriter, r *http.Request) {
    // 设置最大内存限制,超出部分将存储在磁盘上
    err := r.ParseMultipartForm(10 << 20) // 10MB
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 获取文件数据
    file, header, err := r.FormFile("file") // "file" 是前端传递的文件字段名
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
    defer file.Close()

    // 读取文件内容
    fileBytes, err := ioutil.ReadAll(file)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 这里可以对文件进行处理,例如保存到磁盘、数据库等
    // 以下示例将文件内容打印到控制台
    fmt.Printf("Uploaded File: %+v\n", header.Filename)
    fmt.Printf("File Size: %+v\n", len(fileBytes))
    fmt.Printf("MIME Header: %+v\n", header.Header)

    // 将文件保存到磁盘 (示例)
    err = ioutil.WriteFile("./" + header.Filename, fileBytes, 0644)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    fmt.Fprintf(w, "Successfully Uploaded File\n")
}

func main() {
    http.HandleFunc("/upload", handler)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

代码解释:

  1. r.ParseMultipartForm(10 这行代码解析 multipart/form-data 请求。 10
  2. r.FormFile("file"): 从请求中获取名为 "file" 的文件。 "file" 必须与前端代码中用于上传文件的字段名匹配。
  3. ioutil.ReadAll(file): 读取文件内容到 fileBytes 变量中。
  4. ioutil.WriteFile("./" + header.Filename, fileBytes, 0644): 将文件保存到当前目录,文件名为上传的文件名。

注意事项:

  • ParseMultipartForm 必须在读取任何请求体之前调用。
  • 根据服务器的内存大小和预期上传文件的大小,合理设置 ParseMultipartForm 的内存限制。
  • 在生产环境中,应该使用更安全的文件存储方式,例如对象存储服务(如 AWS S3、阿里云 OSS)。
  • 需要进行错误处理,例如检查文件类型、大小等。
  • 需要考虑安全性,例如防止恶意文件上传。

前端实现:使用 JavaScript 上传文件

以下是一个使用 JavaScript 上传文件的示例:




    File Upload


    
    

    

代码解释:

  1. FormData 对象: 用于构建 multipart/form-data 请求。
  2. formData.append('file', file): 将文件添加到 FormData 对象中。 'file' 是后端代码中 r.FormFile("file") 使用的字段名,必须保持一致。
  3. fetch('/upload', ...): 使用 fetch API 发送 POST 请求到 /upload 路由。

注意事项:

  • 确保前端代码中的文件字段名与后端代码中的 r.FormFile() 参数匹配。
  • 可以使用 XMLHttpRequest 对象代替 fetch API。
  • 可以添加上传进度条,提高用户体验。
  • 可以限制上传文件类型和大小。

处理 Expect: 100 Continue 请求

当客户端发送 Expect: 100 Continue 请求头时,服务器应该在接收到请求头后,先检查是否可以处理该请求。如果可以,则返回 HTTP/1.1 100 Continue 响应,通知客户端继续发送请求体。否则,返回错误响应,例如 HTTP/1.1 417 Expectation Failed。

Golang 的 net/http 包会自动处理 Expect: 100 Continue 请求。如果服务器可以处理请求,它会自动发送 HTTP/1.1 100 Continue 响应。如果服务器无法处理请求,则会返回相应的错误响应。

因此,在使用 ParseMultipartForm 处理大文件上传时,无需手动处理 Expect: 100 Continue 请求。

总结

本文介绍了如何使用 Golang 的 net/http 包处理大文件上传。通过 ParseMultipartForm 函数,可以方便地解析 multipart/form-data 请求,并将文件存储在内存或磁盘上。同时,也提供了前端 JavaScript 示例,帮助开发者构建完整的大文件上传功能。最后,强调了 net/http 包会自动处理 Expect: 100 Continue 请求,简化了开发过程。 掌握这些技巧,可以构建稳定可靠的大文件上传服务。