Go 与数据库：连接池是怎么被榨干的

package main

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

    _ "modernc.org/sqlite" // 注册名为 "sqlite"
)

func main() {
    db, err := sql.Open("sqlite", "file:demo.db")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    if _, err := db.Exec(`CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY, name TEXT, age INTEGER)`); err != nil {
        log.Fatal(err)
    }

    res, err := db.Exec("INSERT INTO users (name, age) VALUES (?, ?)", "小明", 25)
    if err != nil {
        log.Fatal(err)
    }
    id, _ := res.LastInsertId()

    var name string
    var age int
    if err := db.QueryRow("SELECT name, age FROM users WHERE id = ?", id).
        Scan(&name, &age); err != nil {
        log.Fatal(err)
    }
    fmt.Printf("插入成功 id=%d，查询到 %s，%d 岁\n", id, name, age)
}

插入成功 id=1，查询到 小明，25 岁

// 单行：ErrNoRows 单独处理
var name string
var age int
err := db.QueryRow("SELECT name, age FROM users WHERE id = $1", 1).Scan(&name, &age)
if err == sql.ErrNoRows {
    fmt.Println("没有找到记录")
} else if err != nil {
    log.Fatal(err)
}

// 多行：defer Close + 循环后检查 Err
rows, err := db.Query("SELECT id, name, age FROM users WHERE age > $1", 18)
if err != nil {
    log.Fatal(err)
}
defer rows.Close()

for rows.Next() {
    var id int
    var name string
    var age int
    if err := rows.Scan(&id, &name, &age); err != nil {
        log.Fatal(err)
    }
    fmt.Printf("ID: %d, %s, %d\n", id, name, age)
}
if err = rows.Err(); err != nil {
    log.Fatal(err) // 迭代中途的网络断连等错误在这里浮出
}

result, err := db.Exec("UPDATE users SET age = $1 WHERE id = $2", 26, 1)
if err != nil {
    log.Fatal(err)
}
affected, _ := result.RowsAffected()
fmt.Printf("更新了 %d 行\n", affected)

var newID int
err = db.QueryRow(
    "INSERT INTO users (name, age) VALUES ($1, $2) RETURNING id",
    "小红", 22,
).Scan(&newID)

func TransferMoney(db *sql.DB, fromID, toID int, amount float64) error {
    tx, err := db.Begin()
    if err != nil {
        return err
    }
    // 无论哪条路径退出都尝试回滚；已 Commit 的事务 Rollback 返回
    // ErrTxDone，无副作用，因此"无条件 defer + 成功后 Commit"
    // 是比按条件判断更简洁也更不容易漏的写法
    defer tx.Rollback()

    result, err := tx.Exec(
        "UPDATE accounts SET balance = balance - $1 WHERE id = $2 AND balance >= $1",
        amount, fromID)
    if err != nil {
        return err
    }
    if affected, _ := result.RowsAffected(); affected == 0 {
        return fmt.Errorf("余额不足或账户不存在")
    }

    if _, err := tx.Exec(
        "UPDATE accounts SET balance = balance + $1 WHERE id = $2", amount, toID); err != nil {
        return err
    }
    return tx.Commit()
}

db.SetMaxOpenConns(25)                 // 最大打开连接数
db.SetMaxIdleConns(10)                 // 最大空闲连接数
db.SetConnMaxLifetime(5 * time.Minute) // 连接最大存活时间
db.SetConnMaxIdleTime(1 * time.Minute) // 空闲连接最大存活时间

ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()

var name string
err := db.QueryRowContext(ctx, "SELECT name FROM users WHERE id = $1", 1).Scan(&name)

tx, err := db.BeginTx(ctx, nil) // 事务同样接受 context，超时自动回滚

// 解法一：sql.NullXxx 类型
var name sql.NullString
db.QueryRow("SELECT name FROM users WHERE id = $1", 1).Scan(&name)
if name.Valid {
    fmt.Println(name.String)
}

// 解法二：指针（通常更干净）
var nickname *string
db.QueryRow("SELECT nickname FROM users WHERE id = $1", 1).Scan(&nickname)
if nickname != nil {
    fmt.Println(*nickname)
}