Go 标准库 database/sql 的连接池机制

Go 语言没有内置通用的连接池库，但 database/sql 包为所有 SQL 数据库驱动提供了统一的连接池管理框架。开发者无需手动实现连接复用、保活、超时等逻辑，只需通过 sql.DB 对象即可获得一个生产可用的连接池。

核心结构：sql.DB 与连接池的关系

sql.DB 不是单个数据库连接，而是一个连接池的抽象。它内部维护了一个连接队列，所有数据库操作（Query、Exec、Begin 等）都从池中获取连接，使用完毕后归还。

连接池的默认行为

如果不做任何配置，sql.Open 返回的 sql.DB 会采用以下默认策略：

连接的生命周期管理

连接池中的每个连接都会经历以下状态流转：

1. 创建（Create）： 当没有空闲连接可用且未达到 MaxOpenConns 上限时，池会创建一个新连接。
2. 分配（Borrow）： 从空闲队列头部取出一个连接，标记为繁忙，交给调用方使用。
3. 使用（Use）： 调用方执行 SQL 操作，连接处于繁忙状态。
4. 归还（Return）： 操作完成后，连接被归还到空闲队列尾部。
5. 清理（Evict）： 如果空闲连接数超过 MaxIdleConns，或连接已超过 ConnMaxLifetime/ConnMaxIdleTime，连接会被关闭并从池中移除。

连接池的并发安全机制

database/sql 的连接池内部使用 sync.Mutex 和 sync.Cond 来保证并发安全：

• 互斥锁： 保护空闲连接队列、繁忙连接集合等共享数据结构的读写操作。
• 条件变量： 当所有连接都在繁忙且已达到 MaxOpenConns 上限时，新的请求会通过 sync.Cond 阻塞等待，直到有连接被归还。
• 连接健康检查： 在分配连接时，如果连接已过期或底层网络异常，池会自动丢弃该连接并尝试创建新连接。

代码示例：默认行为观察

以下代码演示了连接池的默认行为，以及如何通过 db.Stats() 获取池的实时状态：

package main

import (
    "database/sql"
    "fmt"
    "log"
    "time"

    _ "github.com/lib/pq"
)

func main() {
    // 使用默认配置创建连接池
    db, err := sql.Open("postgres", "host=localhost port=5432 user=test dbname=test sslmode=disable")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 观察默认参数
    fmt.Printf("默认 MaxOpenConns: %d\n", db.Stats().MaxOpenConnections)
    fmt.Printf("默认 OpenConnections: %d\n", db.Stats().OpenConnections)
    fmt.Printf("默认 InUse: %d\n", db.Stats().InUse)
    fmt.Printf("默认 Idle: %d\n", db.Stats().Idle)

    // 执行一次查询，观察连接变化
    var now time.Time
    err = db.QueryRow("SELECT NOW()").Scan(&now)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Printf("\n查询后状态:\n")
    fmt.Printf("OpenConnections: %d\n", db.Stats().OpenConnections)
    fmt.Printf("InUse: %d\n", db.Stats().InUse)
    fmt.Printf("Idle: %d\n", db.Stats().Idle)

    // 等待连接归还空闲池
    time.Sleep(100 * time.Millisecond)
    fmt.Printf("\n归还后状态:\n")
    fmt.Printf("OpenConnections: %d\n", db.Stats().OpenConnections)
    fmt.Printf("InUse: %d\n", db.Stats().InUse)
    fmt.Printf("Idle: %d\n", db.Stats().Idle)
}

连接池参数配置方法

通过 sql.DB 的 Set 方法可以显式配置连接池行为：

// 设置最大打开连接数
db.SetMaxOpenConns(25)

// 设置最大空闲连接数
db.SetMaxIdleConns(10)

// 设置连接最大存活时间
db.SetConnMaxLifetime(5 * time.Minute)

// 设置连接最大空闲时间（Go 1.15+）
db.SetConnMaxIdleTime(2 * time.Minute)