问题内容
我想上传一个内容类型设置为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。
最终考虑因素
这里写的代码肯定可以改进。无论如何，只是为了演示，我更愿意让它尽可能接近原始文件，以便让您更好地理解如何在 http 服务器上发布二进制文件以及如何同时运行客户端和服务器同一个程序。如果有不清楚的地方，请告诉我，谢谢！
