在req项目中实现多MIME类型的multipart/form-data请求

在HTTP请求中，multipart/form-data是一种常见的编码方式，特别适用于文件上传和表单提交。但在某些场景下，我们需要为表单中的不同字段设置不同的MIME类型。本文将详细介绍如何使用req库实现这一需求。

需求背景

在实际开发中，我们可能会遇到这样的场景：

• 需要提交一个JSON格式的字段
• 同时包含纯文本说明
• 附带CSV格式的数据
• 还要上传PDF文件

这些不同类型的数据需要设置不同的Content-Type，传统的表单提交方式难以满足这种需求。

req库的解决方案

req库提供了SetFileUpload方法，允许我们为每个字段单独设置MIME类型。核心思路是使用FileUpload结构体，其中包含以下关键属性：

1. ParamName: 表单字段名
2. FileName: 文件名（必填）
3. ContentType: 内容类型
4. GetFileContent: 获取内容的方法

实现示例

我们可以创建一个辅助函数来简化操作：

func createMultipartField(data map[string]string, fieldName string, contentType string) req.FileUpload {
    return req.FileUpload{
        ParamName:   fieldName,
        ContentType: contentType,
        FileName:    fieldName, // 必须提供文件名
        GetFileContent: func() (io.ReadCloser, error) {
            return io.NopCloser(strings.NewReader(data[fieldName])), nil
        },
    }
}

使用示例：

formValues := map[string]string{
    "x1":    `{"a": "bcd", "e": "fgh"}`,
    "x2":    "This is some explanation in plain text",
    "xdata": "a,bcd\ne,fgh",
}

resp, err := netClient.R().
    EnableForceMultipart().
    SetFileUpload(
        createMultipartField(formValues, "x1", "application/json"),
        createMultipartField(formValues, "x2", "text/plain"),
        createMultipartField(formValues, "xdata", "text/csv"),
    ).
    Post("/upload")

注意事项

1. 资源管理：req库会自动关闭GetFileContent返回的io.ReadCloser，开发者无需手动处理。
2. 文件名必填：即使不是真正的文件上传，也必须为FileName属性赋值。
3. 性能考虑：对于大文件，建议使用流式处理而非完全加载到内存。

高级用法

除了文本内容，我们也可以上传真正的文件：

fileUpload := req.FileUpload{
    ParamName: "file",
    FileName: "document.pdf",
    ContentType: "application/pdf",
    GetFileContent: func() (io.ReadCloser, error) {
        return os.Open("path/to/document.pdf")
    },
}

通过这种灵活的方式，req库能够满足各种复杂的multipart/form-data请求场景，为开发者提供了极大的便利性。