Golang中net/http包的使用问题探讨

Golang中net/http包的使用问题探讨 我正在测试一个API，其中有一个用于上传文件的URL，假设是data/import。 其内容类型（contentType）是multipart/form，但只发送文件。

我有一段用于HTTP客户端的代码如下：

     req, _ := http.NewRequest(http.MethodPost, "/data/import", ?)

我不知道在"?"处应该放什么。 是应该直接放文件的路径，还是需要用读取器（reader）读取文件并传递字节切片？

先谢谢了。

3 回复

非常感谢！！！

更多关于Golang中net/http包的使用问题探讨的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


NewRequest 接收一个 io.Reader：

func NewRequest(method, url string, body io.Reader) (*Request, error)

因此，出于测试目的，你可以从字符串或类似的东西中读取数据：

r := strings.NewReader("Test Data")
req, _ := http.NewRequest(http.MethodPost, "/data/import", r)

如果你愿意，也可以从文件中读取数据，类似这样：

file, err := os.Open("testdata.txt") // 用于读取访问。
if err != nil {
	// 处理错误
	log.Fatal(err)
}
defer file.Close()
req, _ := http.NewRequest(http.MethodPost, "/data/import", file)

这段代码完全是未经编译或任何处理直接写出来的，所以你可能需要调整它才能使其真正工作，但这应该能让你入门！

对于上传文件到 multipart/form-data 接口，你需要使用 multipart.Writer 构建请求体，而不是直接传递文件路径或字节切片。以下是完整的示例代码：

package main

import (
    "bytes"
    "io"
    "mime/multipart"
    "net/http"
    "os"
)

func main() {
    // 1. 准备文件
    filePath := "example.txt"
    file, err := os.Open(filePath)
    if err != nil {
        panic(err)
    }
    defer file.Close()

    // 2. 创建multipart请求体
    body := &bytes.Buffer{}
    writer := multipart.NewWriter(body)
    
    // 3. 创建文件字段
    part, err := writer.CreateFormFile("file", "example.txt")
    if err != nil {
        panic(err)
    }
    
    // 4. 复制文件内容
    _, err = io.Copy(part, file)
    if err != nil {
        panic(err)
    }
    
    // 5. 关闭writer以完成multipart编码
    writer.Close()

    // 6. 创建HTTP请求
    req, err := http.NewRequest(
        http.MethodPost, 
        "http://example.com/data/import", 
        body,
    )
    if err != nil {
        panic(err)
    }

    // 7. 设置正确的Content-Type
    req.Header.Set("Content-Type", writer.FormDataContentType())

    // 8. 发送请求
    client := &http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        panic(err)
    }
    defer resp.Body.Close()
}

如果你的API只需要文件而不需要其他表单字段，上面的代码已经足够。如果需要添加额外的表单字段，可以在关闭writer之前添加：

// 添加文本字段
err = writer.WriteField("key", "value")
if err != nil {
    panic(err)
}

关键点：

  1. 使用 multipart.NewWriter 创建multipart编码器
  2. 通过 CreateFormFile 方法创建文件字段
  3. 使用 io.Copy 将文件内容复制到multipart部分
  4. 必须正确设置 Content-Type 头为 writer.FormDataContentType()
  5. 请求体是 bytes.Buffer 类型，包含编码后的multipart数据
回到顶部