关闭 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 问题(可能是一场竞赛),它使用了应该从池中取出的连接。


函数式编程
浏览 159回答 3
3回答

哆啦的时光机

连接被添加到函数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,不会禁用池 。

饮歌长啸

您需要将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

红颜莎娜

http.Transport 有一个名为MaxConnsPerHostMaxConnsPerHost 可以选择限制每个主机的连接总数,包括处于拨号、活动和空闲状态的连接。包括拨号、活动和空闲状态
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Go