如何在 golang net/http 中使用 Transport 添加头信息

Question

我正在尝试通过创建 Trasport 来控制 keep-alives 会话以重用 tcp 连接。

这是我的 sn-p，我不确定如何添加标头信息以进行身份​​验证。

url := "http://localhost:8181/api/v1/resource"
tr := &http.Transport{
    DisableKeepAlives:   false,
    MaxIdleConns:        0,
    MaxIdleConnsPerHost: 0,
    IdleConnTimeout:     time.Second * 10,
}
client := &http.Client{Transport: tr}
resp, err := client.Get(url)

Answer 1

不要混合请求中的Client。
客户端使用Transport 并运行请求：client.Do(req)

你用(h Header) Set(key, value string)在http.Request上设置了标题：

req.Header.Set("name", "value")

Answer 2

这是我发现的：

    package main

    import (
        "fmt"
        "io/ioutil"
        "net/http"
    )

    var URL = "http://httpbin.org/ip"

    func main() {
        tr := &http.Transport{DisableKeepAlives: false}
        req, _ := http.NewRequest("GET", URL, nil)
        req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", "Token"))
        req.Close = false

        res, err := tr.RoundTrip(req)
        if err != nil {
            fmt.Println(err)
        }
        body, _ := ioutil.ReadAll(res.Body)
        fmt.Println(string(body))
    }

而且它有效。

Answer 3

对于您的特定问题，这可能不是您想要的 - 在请求中设置它在您的情况下更有意义，但要直接回答您的问题，您应该能够为通过使用自定义的 RoundTrip 方法进行传输。

查看https://golang.org/pkg/net/http/#RoundTripper

类似：

type CustomTransport struct {
  http.RoundTripper
}

func (ct *CustomTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    req.Header.Add("header-key", "header-value")
    return ct.RoundTripper.RoundTrip(req)
}

url := "http://localhost:8181/api/v1/resource"
tr := &CustomTransport{
    DisableKeepAlives:   false,
    MaxIdleConns:        0,
    MaxIdleConnsPerHost: 0,
    IdleConnTimeout:     time.Second * 10,
}
client := &http.Client{Transport: tr}
resp, err := client.Get(url)

当我无法直接访问 API 客户端库（或每个请求对象）使用的 http 客户端时，我发现这很有用，但它允许我传入传输。