4. Go HTTP客户端进阶：设置请求头、超时控制、连接池管理、Cookie与Session处理

在契约压测场景中，HTTP 客户端不仅仅是发送请求并等待响应。为了模拟真实用户行为、保障压测稳定性、复用连接资源，必须掌握 Go net/http 包的高级配置。本章将深入请求头定制、超时控制、连接池调优以及 Cookie/Session 状态管理，这些是构建高保真压测脚本的核心能力。

4.1 请求头设置：从静态到动态

请求头（Header）是 REST API 契约验证的第一道关卡。压测时需精确控制 Content-Type、Authorization、User-Agent 等字段。

4.1.1 基础设置方法

// 创建请求
req, _ := http.NewRequest("POST", "https://api.example.com/order", body)
// 设置请求头
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+token)
req.Header.Set("User-Agent", "GoPTest/1.0")
req.Header.Set("X-Request-ID", uuid.New().String())

4.1.2 动态请求头与契约验证

压测时请求头常需动态生成，例如时间戳签名、随机 TraceID。以下表格列出常见动态头及其生成策略：

4.2 超时控制：三层防护体系

压测中最怕“悬空连接”耗尽 goroutine。Go 的 http.Client 提供三层超时机制，必须逐层配置。

4.2.1 三层超时模型

1. DialContext 超时：TCP 连接建立的最长等待时间。
2. TLS 握手超时：HTTPS 场景下 TLS 协商的最长等待。
3. ResponseHeader 超时：等待服务端返回响应头的最大时间。

transport := &http.Transport{
    DialContext: (&net.Dialer{
        Timeout:   5 * time.Second,   // 连接超时
        KeepAlive: 30 * time.Second,  // 连接保活
    }).DialContext,
    TLSHandshakeTimeout:   3 * time.Second,   // TLS 握手超时
    ResponseHeaderTimeout: 5 * time.Second,   // 响应头超时
    ExpectContinueTimeout: 1 * time.Second,
}

client := &http.Client{
    Timeout:   30 * time.Second,  // 整体请求超时（兜底）
    Transport: transport,
}

4.2.2 超时配置对比

4.3 连接池管理：压测性能的基石

Go 的 http.Transport 默认维护一个连接池。压测时若配置不当，会出现端口耗尽、连接泄漏或吞吐瓶颈。

4.3.1 核心参数调优

transport := &http.Transport{
    MaxIdleConns:        100,              // 所有主机的最大空闲连接数
    MaxIdleConnsPerHost: 20,               // 每台主机的最大空闲连接数
    MaxConnsPerHost:     50,               // 每台主机的最大并发连接数（含活跃+空闲）
    IdleConnTimeout:     90 * time.Second, // 空闲连接关闭时间
    DisableCompression:  false,            // 启用 gzip 压缩
}

4.3.2 连接状态监控

通过 expvar 或自定义指标可监控连接池状态：

• Goroutines 数：异常增长说明连接未复用。
• TCP 连接数（ss -s）：远高于 MaxConnsPerHost 说明配置失效。
• TIME_WAIT 状态：大量短连接导致端口耗尽。

4.4 Cookie 与 Session 处理

许多 REST API 依赖 Cookie 维持会话状态。压测时需模拟登录态、Token 刷新和 Session 隔离。

4.4.1 CookieJar 的使用

Go 标准库提供 cookiejar 自动管理 Cookie：

jar, _ := cookiejar.New(nil)
client := &http.Client{
    Jar: jar,
}
// 首次请求后，Cookie 自动存储
resp, _ := client.Get("https://api.example.com/login")
// 后续请求自动携带 Cookie
resp2, _ := client.Get("https://api.example.com/profile")

4.4.2 手动 Cookie 控制

压测中常需手动注入或校验 Cookie：

// 手动设置 Cookie
req.AddCookie(&http.Cookie{
    Name:  "session_id",
    Value: "abc123",
    Path:  "/",
})

// 读取响应中的 Set-Cookie
for _, c := range resp.Cookies() {
    if c.Name == "session_id" {
        sessionToken = c.Value
    }
}

4.4.3 Session 隔离策略

多用户压测时，每个虚拟用户应有独立 Session：

4.5 综合配置示例

以下代码整合了本章所有进阶配置，可直接用于契约压测脚本：

func NewBenchClient() *http.Client {
    transport := &http.Transport{
        DialContext: (&net.Dialer{
            Timeout:   5 * time.Second,
            KeepAlive: 30 * time.Second,
        }).DialContext,
        TLSHandshakeTimeout:   3 * time.Second,
        ResponseHeaderTimeout: 5 * time.Second,
        MaxIdleConns:          100,
        MaxIdleConnsPerHost:   20,
        MaxConnsPerHost:       50,
        IdleConnTimeout:       90 * time.Second,
    }

    jar, _ := cookiejar.New(nil)

    return &http.Client{
        Timeout:   30 * time.Second,
        Transport: transport,
        Jar:       jar,
    }
}

// 使用示例
func SendOrder(client *http.Client, token string, orderData []byte) (*http.Response, error) {
    req, _ := http.NewRequest("POST", "https://api.example.com/order", bytes.NewReader(orderData))
    req.Header.Set("Content-Type", "application/json")
    req.Header.Set("Authorization", "Bearer "+token)
    req.Header.Set("X-Request-ID", uuid.New().String())
    return client.Do(req)
}