4. 连接池生命周期：连接的创建、复用、回收、销毁全过程

连接池的核心价值在于对连接实例进行全生命周期管理，避免每次请求都经历三次握手与鉴权开销。一个连接从“出生”到“消亡”会经历四个关键阶段：创建、复用、回收、销毁。理解每个阶段的触发条件与内部状态转换，是编写高水位连接池的基石。

4.1 连接创建：按需预热与懒加载

连接创建并非一次性全部建立，而是遵循“懒加载 + 按需预热”策略。创建时机分为以下三种场景：

// 连接创建核心逻辑示例
func (p *Pool) createConnection(ctx context.Context) (*Conn, error) {
    p.mu.Lock()
    if p.total >= p.maxTotal {
        p.mu.Unlock()
        return nil, ErrPoolExhausted
    }
    p.total++
    p.mu.Unlock()

    // 实际网络拨号
    conn, err := net.DialTimeout("tcp", p.addr, p.dialTimeout)
    if err != nil {
        p.mu.Lock()
        p.total--
        p.mu.Unlock()
        return nil, err
    }

    // 执行认证、设置参数等初始化
    if err := p.initConn(conn); err != nil {
        conn.Close()
        p.mu.Lock()
        p.total--
        p.mu.Unlock()
        return nil, err
    }

    return &Conn{raw: conn, createTime: time.Now()}, nil
}

4.2 连接复用：借与还的契约

复用阶段是连接池性能的核心体现。调用方通过 Get() 从池中“借”出连接，使用完毕后通过 Put() 或 Close() 将连接“还”回池中。复用过程必须遵守以下契约：

• 借出策略：优先从空闲队列头部取出最老连接（FIFO），避免连接长期未被使用而老化。
• 健康检查：借出前执行 Ping() 或 IsBroken() 检测，失效连接直接进入销毁流程，不返回给调用方。
• 归还判定：归还时检查连接是否超过 MaxLifetime 或 MaxIdleTime，超龄连接直接销毁，不归还池中。
• 空闲队列上限：归还时若空闲队列长度已达 MaxIdle，则销毁当前连接，保持空闲池水位。

// 连接复用核心逻辑示例
func (p *Pool) Get(ctx context.Context) (*Conn, error) {
    for {
        select {
        case conn := <-p.idleCh:
            // 健康检查
            if p.isBroken(conn) || p.isExpired(conn) {
                p.destroy(conn)
                continue
            }
            return conn, nil
        default:
            // 空闲队列为空，尝试创建新连接
            return p.createConnection(ctx)
        }
    }
}

func (p *Pool) Put(conn *Conn) {
    if p.isExpired(conn) || p.isBroken(conn) {
        p.destroy(conn)
        return
    }

    select {
    case p.idleCh <- conn:
        // 成功归还至空闲队列
    default:
        // 空闲队列已满，销毁连接
        p.destroy(conn)
    }
}

4.3 连接回收：被动检测与主动驱逐

回收阶段负责将“不健康”或“超龄”的连接从池中移除，防止调用方获取到失效连接。回收机制分为被动回收和主动回收两种：

4.4 连接销毁：优雅关闭与资源清理

销毁是连接生命周期的终点，必须确保底层 socket、缓冲区、goroutine 等资源被彻底释放。销毁流程应遵循以下步骤：

1. 从池中移除：先从空闲队列或活跃集合中删除连接引用，避免被其他 goroutine 误用。
2. 发送关闭指令：若协议支持（如 MySQL COM_QUIT），先发送优雅关闭指令，让服务端及时释放资源。
3. 关闭底层连接：调用 net.Conn.Close() 关闭 TCP 连接。
4. 更新计数器：原子递减 total 计数器，释放总连接数配额。
5. 触发事件回调：若有注册的 OnClose 回调，执行自定义清理逻辑（如释放内存、记录日志）。

// 连接销毁核心逻辑示例
func (p *Pool) destroy(conn *Conn) {
    p.mu.Lock()
    defer p.mu.Unlock()

    // 1. 从活跃集合中移除
    delete(p.activeConns, conn.id)

    // 2. 发送优雅关闭指令（协议相关）
    if conn.raw != nil {
        conn.raw.SetDeadline(time.Now().Add(2 * time.Second))
        conn.raw.Write([]byte("QUIT\r\n")) // 示例：Redis 协议
    }

    // 3. 关闭底层连接
    if conn.raw != nil {
        conn.raw.Close()
    }

    // 4. 更新计数器
    p.total--

    // 5. 触发回调
    if p.onClose != nil {
        p.onClose(conn)
    }
}

4.5 全生命周期状态流转图

下图完整展示了连接从创建到销毁的四个阶段及其状态转换关系：

4.6 生命周期各阶段水位变化对比

下表总结了四个阶段中三个关键水位指标的变化情况，帮助理解每个操作对池状态的影响：