Go语言中SQL连接的惯用共享模式与并发安全

package main

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

    _ "github.com/mattn/go-sqlite3" // 导入SQLite驱动
    "yourproject/repository"       // 假设有一个名为repository的包
)

// Db 是一个全局变量，用于存储数据库句柄
// 注意：虽然方便，但全局变量可能降低模块的可测试性，依赖注入通常更推荐。
var Db *sql.DB

func main() {
    var err error
    // 初始化数据库连接
    Db, err = sql.Open("sqlite3", "./foo.db")
    if err != nil {
        log.Fatalf("无法打开数据库: %v", err)
    }

    // 设置连接池参数（可选但推荐）
    Db.SetMaxOpenConns(25)                 // 最大打开连接数
    Db.SetMaxIdleConns(10)                 // 最大空闲连接数
    Db.SetConnMaxLifetime(5 * time.Minute) // 连接最大生命周期

    // 确保在程序退出前关闭数据库连接
    defer func() {
        if err := Db.Close(); err != nil {
            log.Printf("关闭数据库连接失败: %v", err)
        }
    }()

    // 尝试ping数据库以验证连接
    if err = Db.Ping(); err != nil {
        log.Fatalf("无法连接到数据库: %v", err)
    }
    fmt.Println("数据库连接成功并准备就绪。")

    // 示例：将Db实例传递给其他包的函数
    // 这是一个依赖注入的例子，repository包中的函数接收Db实例
    userRepo := repository.NewUserRepository(Db)
    err = userRepo.CreateTable()
    if err != nil {
        log.Fatalf("创建用户表失败: %v", err)
    }

    // 更多业务逻辑...
    fmt.Println("应用程序启动完成。")

    // 模拟长时间运行
    select {}
}

// 假设yourproject/repository包的结构
package repository

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

// UserRepository 结构体，包含一个sql.DB实例
type UserRepository struct {
    db *sql.DB
}

// NewUserRepository 创建并返回一个新的UserRepository实例
func NewUserRepository(db *sql.DB) *UserRepository {
    return &UserRepository{db: db}
}

// CreateTable 在数据库中创建用户表
func (r *UserRepository) CreateTable() error {
    query := `
    CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        email TEXT UNIQUE NOT NULL
    );`
    _, err := r.db.Exec(query)
    if err != nil {
        return fmt.Errorf("执行创建表查询失败: %w", err)
    }
    log.Println("用户表已创建或已存在。")
    return nil
}

// AddUser 向用户表中添加新用户
func (r *UserRepository) AddUser(name, email string) error {
    query := `INSERT INTO users (name, email) VALUES (?, ?)`
    _, err := r.db.Exec(query, name, email)
    if err != nil {
        return fmt.Errorf("添加用户失败: %w", err)
    }
    log.Printf("用户 %s (%s) 已添加。", name, email)
    return nil
}

// GetUserByID 根据ID获取用户
func (r *UserRepository) GetUserByID(id int) (string, string, error) {
    query := `SELECT name, email FROM users WHERE id = ?`
    row := r.db.QueryRow(query, id)

    var name, email string
    err := row.Scan(&name, &email)
    if err != nil {
        if err == sql.ErrNoRows {
            return "", "", fmt.Errorf("未找到ID为 %d 的用户", id)
        }
        return "", "", fmt.Errorf("查询用户失败: %w", err)
    }
    return name, email, nil
}