本文深入探讨了 Go 语言中使用 `net/http` 包发起 HTTP(S) 请求时，客户端连接无法复用的问题。通过分析示例代码，解释了连接复用的关键因素，并提供了确保连接复用的正确方法，包括读取完整响应体和关闭响应体。同时，简要介绍了如何通过 `time.Tick` 限制请求速率，以避免连接数过多。
在使用 Go 语言的 net/http 包进行 HTTP(S) 请求时，连接复用对于性能至关重要。默认情况下，http.Client 会尝试复用连接，但如果不注意某些细节，可能会导致连接无法复用，从而增加延迟和资源消耗。
连接复用的关键：读取完整响应体和关闭响应体
Go 官方文档明确指出，为了实现连接复用，必须在读取完响应体后关闭它。具体来说，需要完成以下两个步骤：
- 读取完整响应体： 确保从 resp.Body 中读取所有数据。可以使用 io.Copy(ioutil.Discard, resp.Body) 将响应体的内容读取并丢弃，或者使用 ioutil.ReadAll(resp.Body) 将所有内容读取到内存中。
- 关闭响应体： 在读取完响应体后，务必调用 resp.Body.Close() 关闭响应体。
示例代码
以下代码演示了如何正确地进行 HTTP(S) 请求并复用连接：
package main
import (
"fmt"
"io"
"io/ioutil"
"net/http"
)
func main() {
client := &http.Client{}
for i := 0; i < 10; i++ {
req, err := http.NewRequest("GET", "https://www.example.com", nil)
if err != nil {
fmt.Println("Error creating request:", err)
return
}
resp, err := client.Do(req)
if err != nil {
fmt.Println("Error performing request:", err)
return
}
// 确保读取完整响应体
io.Copy(ioutil.Discard, resp.Body)
// 或者使用
// ioutil.ReadAll(resp.Body)
// 关闭响应体
resp.Body.Close()
fmt.Printf("Request %d completed\n", i+1)
}
}在这个示例中，我们创建了一个 http.Client 实例，并在循环中发送了 10 个 GET 请求。每次请求后，我们都使用 io.Copy(ioutil.Discard, resp.Body) 读取完整响应体，并调用 resp.Body.Close() 关闭响应体，从而确保连接可以被复用。
注意事项
- 如果忘记关闭响应体，可能会导致连接泄漏，最终耗尽系统资源。
- 即使响应体为空，也需要调用 resp.Body.Close()。
- 使用 defer resp.Body.Close() 可以确保响应体在函数退出时被关闭，即使发生错误。
连接池大小限制
默认情况下，http.Client 使用一个全局的连接池，该连接池的大小没有明确的限制。在高并发场景下，可能会导致连接数过多，从而影响性能。虽然Go 1.1 中没有直接设置最大连接数的选项，但可以通过自定义 Transport 来间接控制连接池的行为。
transport := &http.Transport{
MaxIdleConns: 100,
MaxIdleConnsPerHost: 10,
DisableKeepAlives: false,
}
client := &http.Client{Transport: transport}MaxIdleConns 设置总共的最大空闲连接数，而 MaxIdleConnsPerHost 设置每个 host 的最大空闲连接数。
限制请求速率
除了控制连接池大小，还可以通过限制请求速率来避免连接数过多。可以使用 time.Tick 来创建一个定时器，控制请求的发送频率。
import "time"
requests_per_second := 5
throttle := time.Tick(time.Second / time.Duration(requests_per_second))
for i := 0; i < 16; i++ {
<-throttle
// 发送请求
}在这个示例中，我们创建了一个每秒发送 5 个请求的定时器。在每次发送请求之前，我们都会从 throttle 通道中接收一个值，从而确保请求的发送频率不会超过限制。
总结
要确保 Go 语言的 net/http 客户端能够复用连接，必须读取完整响应体并关闭响应体。同时，可以通过自定义 Transport 控制连接池大小，或者通过限制请求速率来避免连接数过多。通过遵循这些最佳实践，可以提高 HTTP(S) 客户端的性能和可靠性。
