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（可以是*os.File或*strings.Reader等）
func Upload(client *http.Client, url string, values map[string]io.Reader) (err error) {
    // 使用bytes.Buffer来存储构建的multipart请求体
    var requestBody bytes.Buffer
    writer := multipart.NewWriter(&requestBody)

    for key, reader := range values {
        // 确保文件句柄在操作完成后关闭，避免资源泄露
        if closer, ok := reader.(io.Closer); ok {
            defer closer.Close()
        }

        var partWriter io.Writer
        // 通过类型断言判断是否为文件类型 (*os.File)
        if file, ok := reader.(*os.File); ok {
            // 如果是os.File类型，则创建文件表单字段
            if partWriter, err = writer.CreateFormFile(key, file.Name()); err != nil {
                return fmt.Errorf("创建文件表单字段失败: %w", err)
            }
        } else {
            // 否则创建普通表单字段
            if partWriter, err = writer.CreateFormField(key); err != nil {
                return fmt.Errorf("创建普通表单字段失败: %w", err)
            }
        }

        // 将数据从reader复制到表单字段的writer中
        if _, err = io.Copy(partWriter, reader); err != nil {
            return fmt.Errorf("复制数据到表单字段失败: %w", err)
        }
    }

    // 关闭multipart writer，这会写入请求体的终止边界。
    // 这一步非常关键，否则请求体将不完整，服务器可能无法正确解析。
    if err = writer.Close(); err != nil {
        return fmt.Errorf("关闭multipart writer失败: %w", err)
    }

    // 创建HTTP POST请求
    req, err := http.NewRequest("POST", url, &requestBody)
    if err != nil {
        return fmt.Errorf("创建HTTP请求失败: %w", err)
    }

    // 设置Content-Type头部。必须使用writer.FormDataContentType()获取的完整值，
    // 它包含了由multipart writer生成的边界信息。
    req.Header.Set("Content-Type", writer.FormDataContentType())

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

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

    fmt.Printf("文件上传成功，状态码: %s\n", resp.Status)
    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() {
    // 设置一个模拟的HTTP服务器，用于演示和测试
    // 实际应用中，这里会是你的目标服务器URL
    testServer := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 打印接收到的完整请求，以便观察其结构
        dump, err := httputil.DumpRequest(r, true)
        if err != nil {
            http.Error(w, "Failed to dump request", http.StatusInternalServerError)
            return
        }
        fmt.Printf("--- 模拟服务器接收到的请求 ---\n%s\n---------------------------\n", dump)

        // 简单地返回成功，表示接收到请求
        w.WriteHeader(http.StatusOK)
        fmt.Fprint(w, "Upload received successfully!")
    }))
    defer testServer.Close() // 确保测试服务器在main函数结束时关闭

    // 准备要上传的数据
    // "image" 字段对应一个文件 (这里以main.go自身为例)
    // "key" 字段对应一个普通文本数据
    uploadValues := map[string]io.Reader{
        "image": mustOpen("main.go"), // 假设需要上传一个名为 "image" 的文件
        "key":   strings.NewReader("my_secret_key"), // 假设需要上传一个名为 "key" 的文本字段
    }

    // 调用Upload函数发送请求
    err := Upload(testServer.Client(), testServer.URL, uploadValues)
    if err != nil {
        fmt.Printf("上传失败: %v\n", err)
        os.Exit(1)
    }
}