Go 言語での同時データベース接続の接続プール管理をどのように扱うか?

type Pool struct {
    pool     chan *sql.DB
    capacity int
    count    int
}

func NewPool(dbURL string, capacity int) (*Pool, error) {
    // 创建连接池并初始化
    pool := make(chan *sql.DB, capacity)
    for i := 0; i < capacity; i++ {
        db, err := sql.Open("mysql", dbURL)
        if err != nil {
            return nil, err
        }
        pool <- db
    }

    return &Pool{
        pool:     pool,
        capacity: capacity,
        count:    capacity,
    }, nil
}

func (p *Pool) Get() (*sql.DB, error) {
    // 从连接池获取一个连接
    db := <-p.pool
    p.count--

    return db, nil
}

func (p *Pool) Put(db *sql.DB) {
    // 将连接放回连接池
    p.pool <- db
    p.count++
}

func (p *Pool) Expand() error {
    // 增加连接池中的连接容量
    db, err := sql.Open("mysql", dbURL)
    if err != nil {
        return err
    }
    p.pool <- db
    p.count++

    return nil
}

func (p *Pool) Shrink() error {
    // 减少连接池中的连接容量
    db := <-p.pool
    db.Close()
    p.count--

    return nil
}

func main() {
    dbURL := "username:password@tcp(hostname:port)/dbname"
    capacity := 10

    // 创建连接池
    pool, err := NewPool(dbURL, capacity)
    if err != nil {
        log.Fatal(err)
    }

    // 并发查询
    var wg sync.WaitGroup
    for i := 0; i < 100; i++ {
        wg.Add(1)
        go func(id int) {
            defer wg.Done()

            // 从连接池获取一个连接
            db, err := pool.Get()
            if err != nil {
                log.Println(err)
                return
            }
            defer pool.Put(db)

            // 执行查询
            rows, err := db.Query("SELECT * FROM users")
            if err != nil {
                log.Println(err)
                return
            }
            defer rows.Close()

            // 处理查询结果
            for rows.Next() {
                var name string
                err := rows.Scan(&name)
                if err != nil {
                    log.Println(err)
                    return
                }
                log.Println("Query result:", name)
            }
        }(i)
    }

    // 等待所有goroutine完成
    wg.Wait()
}