关闭 Go http.Client 的连接池

Question

出于测试目的，我试图net/http.Client在 Go 中创建一个禁用了连接池的对象。我想要实现的是为每个 HTTP/1.x 请求的地址建立一个新的 TCP 连接。

目前我有：

        c = &http.Client{
            Transport: &http.Transport{
                DialContext: (&net.Dialer{
                    Timeout:   5 * time.Second,
                    KeepAlive: 5 * time.Second,
                }).DialContext,
                TLSHandshakeTimeout:   5 * time.Second,
                ResponseHeaderTimeout: 5 * time.Second,
                ExpectContinueTimeout: 1 * time.Second,
            },
        }

任何想法我应该如何调整这个？

我看到如果我设置c.Transport.MaxIdleConns = 1它可以工作，但我不确定这是否仍然允许 1 个使用中 + 1 个空闲（总共 2 个）TCP 连接：

    // MaxIdleConns controls the maximum number of idle (keep-alive)
    // connections across all hosts. Zero means no limit.
    MaxIdleConns int

同样，似乎c.Dialer.KeepAlive = -1也可以这样做：

    // KeepAlive specifies the keep-alive period for an active
    // network connection.
    // If zero, keep-alives are enabled if supported by the protocol
    // and operating system. Network protocols or operating systems
    // that do not support keep-alives ignore this field.
    // If negative, keep-alives are disabled.

但我不确定 TCP 连接 + Keep-Alive + HTTP 的行为。

另一种方法是尝试尽快杀死空闲的 TCP 连接，因此我设置了c.Transport.IdleConnTimeout = 1*time.Nanosecond.

当我这样做时，我Client.Do()现在偶尔会返回错误：

tls: use of closed connection

我怀疑这是一个 Go stdlib 问题（可能是一场竞赛），它使用了一个应该从池中取出的连接。

Answer 1

连接被添加到函数Transport.tryPutIdleConn中的池中。如果Transport.DisableKeepAlives为 true 或 Transport.MaxIdleConnsPerHost小于零，则不会合并连接。

设置任一值都会禁用池化。Connection: close当DisableKeepAlives 为true 时，传输会添加请求标头。这可能是理想的，也可能不是理想的，具体取决于您正在测试的内容。

设置DisableKeepAlives的方法如下：

t := http.DefaultTransport.(*http.Transport).Clone()
t.DisableKeepAlives = true
c := &http.Client{Transport: t}

在操场上运行DisableKeepAlives = true 的演示。

设置 MaxIdleConnsPerHost 的方法如下：

t := http.DefaultTransport.(*http.Transport).Clone()
t.MaxIdleConnsPerHost = -1
c := &http.Client{Transport: t}

在操场上运行 MaxIdleConnsPerHost = -1 的演示。

上面的代码克隆了默认传输以确保使用默认传输选项。如果您明确想要问题中的选项，请使用

    c = &http.Client{
        Transport: &http.Transport{
            DialContext: (&net.Dialer{
                Timeout:   5 * time.Second,
                KeepAlive: 5 * time.Second,
            }).DialContext,
            TLSHandshakeTimeout:   5 * time.Second,
            ResponseHeaderTimeout: 5 * time.Second,
            ExpectContinueTimeout: 1 * time.Second,
            DisableKeepAlives: true,
        },
    }

或者

    c = &http.Client{
        Transport: &http.Transport{
            DialContext: (&net.Dialer{
                Timeout:   5 * time.Second,
                KeepAlive: 5 * time.Second,
            }).DialContext,
            TLSHandshakeTimeout:   5 * time.Second,
            ResponseHeaderTimeout: 5 * time.Second,
            ExpectContinueTimeout: 1 * time.Second,
            MaxIdleConnsPerHost: -1,
        },
    }

MaxIdleConnsPerHost 不限制每个主机的活动连接数。 请参阅此游乐场示例进行演示。

通过将Dialer.KeepAlive设置为 -1，不会禁用池 。请参阅此答案以获取解释。