关闭 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，不会禁用池 。

Answer 2

您需要将DisableKeepAlives、true和MaxIdleConnsPerHost-1 设置为。

从文档中：

// DisableKeepAlives, if true, disables HTTP keep-alives and

// will only use the connection to the server for a single

// HTTP request.

https://golang.org/src/net/http/transport.go，第 166 和 187 行

因此，您的客户端必须按如下方式初始化

c = &http.Client{

    Transport: &http.Transport{

        DialContext: (&net.Dialer{

            Timeout:   5 * time.Second,

        }).DialContext,

        TLSHandshakeTimeout:   5 * time.Second,

        ResponseHeaderTimeout: 5 * time.Second,

        ExpectContinueTimeout: 1 * time.Second,

        DisableKeepAlives: true,

        MaxIdleConnsPerHost: -1

    },

}

如果您使用 1.7 之前的 Go 版本，那么您需要消耗主体的所有缓冲区，并且只有在调用request.Body.Close(). 相反，如果您使用的版本大于或等于 1.7，则可以推迟关闭，而无需采取额外的预防措施。

禁用连接池但仍然能够执行并行请求的示例库： https: //github.com/alessiosavi/Requests

Answer 3

http.Transport 有一个名为MaxConnsPerHost

MaxConnsPerHost 可以选择限制每个主机的连接总数，包括处于拨号、活动和空闲状态的连接。

包括拨号、活动和空闲状态