迹忆客 专注技术分享

HTTP 1.1 协议支持 HTTP 持久连接，也称为 HTTP Keep-Alive。 这允许客户端和服务器在发送多个 HTTP 请求/响应时重新使用相同的底层 TCP 连接。 因此，客户端不是为每个 HTTP 请求建立连接，而是多次重复使用先前创建的 TCP 连接。 出于性能原因以及向同一主机发送多个请求时，这特别有用。

未重用 HTTP 连接

这是对同一主机的一系列 HTTP 请求的测试示例。 我们将使用 httptrace 包来检查底层 http 连接信息的状态。 在这篇博文 中查看更多关于 httptrace 用法的信息。

package main

import (
    "context"
    "log"
    "net/http"
    "net/http/httptrace"
)

func main() {
    // client trace to log whether the request's underlying tcp connection was re-used
    clientTrace := &httptrace.ClientTrace{
        GotConn: func(info httptrace.GotConnInfo) { log.Printf("conn was reused: %t", info.Reused) },
    }
    traceCtx := httptrace.WithClientTrace(context.Background(), clientTrace)

    // 1st request
    req, err := http.NewRequestWithContext(traceCtx, http.MethodGet, "http://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }
    _, err = http.DefaultClient.Do(req)
    if err != nil {
        log.Fatal(err)
    }
    // 2nd request
    req, err = http.NewRequestWithContext(traceCtx, http.MethodGet, "http://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }
    _, err = http.DefaultClient.Do(req)
    if err != nil {
        log.Fatal(err)
    }
}

预期输出如下。

即使向同一主机发出请求，也不会重用 HTTP 底层 tcp 连接。 这是由于以下原因，特别是我们在 Go HTTP 响应中使用响应主体的方式。

$ go doc http.Response.Body
...
// The default HTTP client's Transport
// may not reuse HTTP/1.x "keep-alive" TCP connections 
// if the Body is not read to completion and closed.
...
Body io.ReadCloser

重用 HTTP 连接

为了让客户端重用底层连接，我们只需要在发出新的 HTTP 请求之前完全读取正文并关闭它。 让我们看一个实际的例子。

package main

import (
    "context"
    "io"
    "io/ioutil"
    "log"
    "net/http"
    "net/http/httptrace"
)

func main() {
    // client trace to log whether the request's underlying tcp connection was re-used
    clientTrace := &httptrace.ClientTrace{
        GotConn: func(info httptrace.GotConnInfo) { log.Printf("conn was reused: %t", info.Reused) },
    }
    traceCtx := httptrace.WithClientTrace(context.Background(), clientTrace)

    // 1st request
    req, err := http.NewRequestWithContext(traceCtx, http.MethodGet, "http://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }
    res, err := http.DefaultClient.Do(req)
    if err != nil {
        log.Fatal(err)
    }
    if _, err := io.Copy(ioutil.Discard, res.Body); err != nil {
        log.Fatal(err)
    }
    res.Body.Close()
    // 2nd request
    req, err = http.NewRequestWithContext(traceCtx, http.MethodGet, "http://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }
    _, err = http.DefaultClient.Do(req)
    if err != nil {
        log.Fatal(err)
    }
}

这现在可以按预期正确地重用 tcp 连接。