关闭 Go http.Client 的连接池

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《关闭 Go http.Client 的连接池》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

问题内容

出于测试目的，我正在尝试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 问题（可能是一场比赛），它使用了一个应该从池中取出的连接。

正确答案

在函数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不会禁用池 化。请参阅[此答案](https://stackoverflow.com/questions/59656164/what-is-the- difference-between-net-dialerkeepalive-and-http- transportidletimeo/59656223#59656223)以获取解释。

今天关于《关闭 Go http.Client 的连接池》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！