Go语言数据库：连接池管理

1. 连接池基础概念

数据库连接池（Connection Pool）用于管理数据库连接的创建、复用和释放，避免频繁建立/关闭连接带来的性能开销。Go语言通过`database/sql`标准库和第三方库支持连接池管理。

连接复用：减少TCP三次握手/四次挥手的开销
连接限制：控制最大并发连接数，防止数据库过载
健康检查：自动关闭失效连接，确保连接可用性

2. 标准库连接池配置

`database/sql`通过`DB`对象管理连接池，支持设置最大打开连接数、最大空闲连接数和连接超时时间：

示例：连接池配置

package main

import (
    "database/sql"
    "fmt"
    "github.com/go-sql-driver/mysql"
)

func main() {
    cfg := mysql.Config{
        User:   "root",
        Passwd: "password",
        Net:    "tcp",
        Addr:   "localhost:3306",
        DBName: "test",
    }
    db, err := sql.Open("mysql", cfg.FormatDSN())
    if err != nil {
        panic(err)
    }
    defer db.Close()

    // 配置连接池
    db.SetMaxOpenConns(20)   // 最大打开连接数
    db.SetMaxIdleConns(5)    // 最大空闲连接数
    db.SetConnMaxLifetime(30 * time.Minute)  // 连接最大存活时间
    db.SetConnMaxIdleTime(10 * time.Minute)  // 连接最大空闲时间

    // 验证连接可用性
    if err := db.Ping(); err != nil {
        panic(err)
    }
    fmt.Println("连接池初始化成功")
}

关键参数说明：

`SetMaxOpenConns`：限制同时打开的连接数（包括正在使用和空闲的）
`SetMaxIdleConns`：限制空闲连接数（超过会被关闭）
`SetConnMaxLifetime`：连接存活时间（防止连接长时间占用）
`SetConnMaxIdleTime`：空闲连接最大保留时间（避免无效连接占用资源）

3. GORM连接池配置

GORM基于`database/sql`的连接池，通过`gorm.Config`的`ConnPool`字段配置，支持与标准库一致的参数：

示例：GORM连接池配置

package main

import (
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
    "time"
)

func main() {
    dsn := "root:password@tcp(localhost:3306)/test?charset=utf8&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
        ConnPool: &sql.DB{
            // 复用database/sql的连接池配置
            SetMaxOpenConns: 20,
            SetMaxIdleConns: 5,
            SetConnMaxLifetime: 30 * time.Minute,
            SetConnMaxIdleTime: 10 * time.Minute,
        },
    })
    if err != nil {
        panic("连接数据库失败：" + err.Error())
    }
}

注意事项：

需导入`database/sql`包以访问连接池方法
配置参数需根据业务场景调整（如高并发场景增大`MaxOpenConns`）
定期调用`DB.Stats()`监控连接池状态

4. 连接池监控与调优

通过`DB.Stats()`获取连接池状态，包括当前使用连接数、空闲连接数等，帮助定位性能瓶颈：

示例：连接池状态监控

func printPoolStats(db *sql.DB) {
    stats := db.Stats()
    fmt.Printf("当前使用连接数: %d\n", stats.InUse)
    fmt.Printf("空闲连接数: %d\n", stats.Idle)
    fmt.Printf("总打开连接数: %d\n", stats.OpenConnections)
    fmt.Printf("等待连接数: %d\n", stats.WaitCount)
}

调优策略：

若`InUse`接近`MaxOpenConns`，考虑增大`MaxOpenConns`或优化查询减少连接占用时间
若`Idle`长期高于`MaxIdleConns`，检查`SetMaxIdleConns`配置是否合理
若`WaitCount`持续增加，说明连接池无法满足需求，需扩容或优化业务逻辑

5. 连接池常见问题与解决方案

连接池使用中可能遇到连接泄漏、超时等问题，需针对性解决：

示例：连接泄漏处理

// 错误示例：未关闭Rows导致连接泄漏
rows, err := db.Query("SELECT * FROM users")
if err != nil {
    // 处理错误
}
// 未调用rows.Close()

// 正确示例：使用defer关闭Rows
rows, err := db.Query("SELECT * FROM users")
if err != nil {
    // 处理错误
}
	defer rows.Close()

常见问题：

连接泄漏：未正确关闭`Rows`或`Stmt`，导致连接无法返回池
连接超时：数据库防火墙关闭空闲连接，需设置合理的`ConnMaxLifetime`
连接争用：高并发场景下连接池过小，导致大量线程等待

解决方案：

始终使用`defer rows.Close()`关闭查询结果
设置`ConnMaxLifetime`小于数据库的`wait_timeout`参数
根据压测结果调整`MaxOpenConns`和`MaxIdleConns`

6. 实战：高并发场景连接池配置

针对高并发Web应用，推荐以下连接池配置策略：

示例：高并发配置

func initDB() *sql.DB {
    cfg := mysql.Config{/* 数据库配置 */}
    db, err := sql.Open("mysql", cfg.FormatDSN())
    if err != nil {
        panic(err)
    }

    // 高并发配置
    db.SetMaxOpenConns(runtime.NumCPU() * 10)  // CPU核心数×10
    db.SetMaxIdleConns(runtime.NumCPU() * 2)   // 保持少量空闲连接
    db.SetConnMaxLifetime(15 * time.Minute)    // 小于数据库wait_timeout（通常30分钟）
    db.SetConnMaxIdleTime(5 * time.Minute)     // 空闲5分钟后关闭

    return db
}

配置依据：

`MaxOpenConns`：根据CPU核心数设置，避免超过数据库最大连接限制
`MaxIdleConns`：保留部分空闲连接应对突发请求
`ConnMaxLifetime`：确保连接在数据库关闭前被替换