发布二进制数据（应用程序/八位字节流）

2024-02-05 21:43

短信预约 -IT技能 免费直播动态提醒

问题内容

我想上传一个内容类型设置为application/octet-stream的文件以及请求正文中文件的二进制数据。我将如何在 golang 中做到这一点，这是我的起始代码：

package main

import (
    "fmt"
    "log"
    "net/http"
    "net/http/httputil"
    "os"
)

func main() {
    file, err := os.Open("file.pdf")
    if err != nil {
        log.Fatalln(err)
    }

    req, err := http.NewRequest("POST", fmt.Sprintf("https://example.com/upload"), nil)
    req.Header.Add("Content-Type", "application/octet-stream")

    if err != nil {
        log.Fatalln(err)
    }

    client := &http.Client{
        Transport: &http.Transport{
            DisableCompression: true,
        },
    }

    resp, err := client.Do(req)
    if err != nil {
        log.Fatalln(err)
    }
    defer resp.Body.Close()

    reqDump, err := httputil.DumpRequest(req, true)
    if err != nil {
        log.Fatalln(err)
    }

    fmt.Println(reqDump)
}

正确答案

我整理了一个可以帮助您解决问题的答案。所有代码都属于 main.go 文件，只是为了演示。它有两个主要部分：handler 和 main 函数。让我们从处理程序开始。

http 处理程序

func handlefile(w http.responsewriter, r *http.request) {
    if r.method != http.methodpost {
        w.writeheader(http.statusmethodnotallowed)
        return
    }
    fmt.println("handlefile")
    data, err := io.readall(r.body)
    if err != nil {
        panic(err)
    }
    defer r.body.close()
    w.write(data)
}

在这里，我只是依赖于你使用的（仅go标准库：http、io和fmt包）。我们只需要将从 http 请求中读取的传入有效负载写入响应流。

主要功能

func main() {
    r := http.NewServeMux()
    r.HandleFunc("/example", HandleFile)
    srv := http.Server{
        Addr:    ":8000",
        Handler: r,
    }

    var wg sync.WaitGroup
    wg.Add(1)
    go func() {
        defer wg.Done()
        if err := srv.ListenAndServe(); err != nil {
            fmt.Println(err.Error())
        }
    }()

    file, err := os.Open("file.txt")
    if err != nil {
        panic(err)
    }
    defer file.Close()
    req, err := http.NewRequest(http.MethodPost, "http://localhost:8000/example", file)
    if err != nil {
        panic(err)
    }
    req.Header.Set("Content-Type", "application/octet-stream")
    client := &http.Client{
        Transport: &http.Transport{
            DisableCompression: true,
        },
    }
    res, err := client.Do(req)
    if err != nil {
        panic(err)
    }
    defer res.Body.Close()
    data, err := io.ReadAll(res.Body)
    if err != nil {
        panic(err)
    }
    fmt.Println(string(data))

    srv.Shutdown(context.Background())
    wg.Wait()
}

相反，main函数有更多的逻辑需要处理。让我们在列表中回顾一下它们：

它创建一个 http 服务器来服务传入的请求。我们对其进行检测，以使用我们之前创建的 handlefile 处理程序来回复针对 /example 端点的请求。
我们使用 sync.waitgroup 在单独的 goroutine 中运行服务器。由于这一点，服务器已启动并运行，但我们可以继续我们的程序并在其中发送 http 请求。
我们准备一个 http 请求，其中包含从本地文件系统读取的文件内容（在我们的示例中为 file.txt）。我们将 content-type 标头设置为 application/octet-stream 并将文件句柄传递给 newrequest 函数。
我们发出请求并打印 http 响应负载的内容。
我们关闭服务器并使用 wg.wait 方法告诉 waitgroup 等待所有 goroutine。