Go语言实现multipart/form-data文件上传教程

package main

import (
    "bytes"
    "fmt"
    "io"
    "mime/multipart"
    "net/http"
    "net/http/httptest" // 用于模拟服务器，便于测试
    "net/http/httputil" // 用于打印请求详情
    "os"
    "strings"
)

// Upload 函数负责构建并发送 multipart/form-data 请求
// client: 用于发送请求的 HTTP 客户端
// url: 目标服务器的 URL
// values: 一个 map，键为表单字段名，值为 io.Reader 接口，可以是文件或字符串
func Upload(client *http.Client, url string, values map[string]io.Reader) (err error) {
    // 创建一个缓冲区来存储 multipart/form-data 请求体
    var b bytes.Buffer
    // 使用 multipart.NewWriter 创建一个写入器，它会将数据写入缓冲区 b
    w := multipart.NewWriter(&b)

    // 遍历所有要上传的字段和文件
    for key, r := range values {
        var fw io.Writer
        // 如果 io.Reader 实现了 io.Closer 接口（如 *os.File），则在函数结束时关闭它
        if x, ok := r.(io.Closer); ok {
            defer x.Close()
        }

        // 判断当前项是否为文件 (*os.File)
        if x, ok := r.(*os.File); ok {
            // 如果是文件，使用 CreateFormFile 创建一个文件字段
            // 第一个参数是表单字段名，第二个参数是文件名（用于 Content-Disposition）
            if fw, err = w.CreateFormFile(key, x.Name()); err != nil {
                return fmt.Errorf("创建文件表单字段失败: %w", err)
            }
        } else {
            // 如果是普通字段（非文件），使用 CreateFormField 创建一个普通表单字段
            if fw, err = w.CreateFormField(key); err != nil {
                return fmt.Errorf("创建普通表单字段失败: %w", err)
            }
        }

        // 将数据从 io.Reader 复制到表单字段写入器中
        if _, err = io.Copy(fw, r); err != nil {
            return fmt.Errorf("复制数据到表单字段失败: %w", err)
        }
    }

    // !!! 非常重要 !!!
    // 关闭 multipart 写入器。如果不关闭，请求体将缺少终止边界，导致服务器无法正确解析。
    w.Close()

    // 现在请求体已经构建完成，可以创建 HTTP 请求了
    req, err := http.NewRequest("POST", url, &b)
    if err != nil {
        return fmt.Errorf("创建 HTTP 请求失败: %w", err)
    }

    // !!! 非常重要 !!!
    // 设置 Content-Type 请求头。必须使用 w.FormDataContentType() 获取正确的 Content-Type，
    // 它包含了 multipart 的边界信息。
    req.Header.Set("Content-Type", w.FormDataContentType())

    // 发送 HTTP 请求
    res, err := client.Do(req)
    if err != nil {
        return fmt.Errorf("发送 HTTP 请求失败: %w", err)
    }
    defer res.Body.Close() // 确保关闭响应体

    // 检查响应状态码
    if res.StatusCode != http.StatusOK {
        return fmt.Errorf("服务器返回错误状态: %s", res.Status)
    }

    fmt.Println("文件上传成功！")
    return nil
}

// mustOpen 是一个辅助函数，用于打开文件，如果失败则触发 panic
// 在生产代码中，应进行更完善的错误处理
func mustOpen(f string) *os.File {
    r, err := os.Open(f)
    if err != nil {
        panic(fmt.Sprintf("无法打开文件 %s: %v", f, err))
    }
    return r
}

func main() {
    var client *http.Client
    var remoteURL string

    // 设置一个模拟的 HTTP 服务器，用于测试上传功能
    // 在实际应用中，这里会是你的真实服务器地址
    ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 打印接收到的请求的详细信息，包括请求体
        b, err := httputil.DumpRequest(r, true)
        if err != nil {
            http.Error(w, "Internal Server Error", http.StatusInternalServerError)
            return
        }
        fmt.Printf("模拟服务器接收到请求:\n%s\n", b)

        // 简单检查是否包含名为 "file" 的文件
        _, _, err = r.FormFile("file")
        if err != nil {
            fmt.Fprintf(w, "服务器响应: 未收到文件 'file': %v\n", err)
            return
        }
        fmt.Fprintf(w, "服务器响应: 文件 'file' 已接收\n")
    }))
    defer ts.Close() // 确保在 main 函数结束时关闭模拟服务器

    client = ts.Client() // 使用模拟服务器的客户端
    remoteURL = ts.URL   // 模拟服务器的 URL

    // 准备要上传的数据
    // "file" 字段对应一个文件 (这里使用 main.go 文件本身作为示例)
    // "other" 字段对应一个普通的文本数据
    values := map[string]io.Reader{
        "file":  mustOpen("main.go"), // 上传当前 main.go 文件
        "other": strings.NewReader("hello world!"),
        "key":   strings.NewReader("YOUR_API_KEY"), // 模拟一个API key字段
    }

    // 调用 Upload 函数进行文件上传
    err := Upload(client, remoteURL, values)
    if err != nil {
        fmt.Printf("文件上传失败: %v\n", err)
        os.Exit(1)
    }
    fmt.Println("主程序：上传过程完成。")
}