Go语言中SQL数据库访问：database/sql 包与驱动生态

package main

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

    // 导入MySQL驱动，通常使用空白导入，仅用于执行其init()函数注册驱动
    _ "github.com/go-sql-driver/mysql" 
)

func main() {
    // 1. 连接数据库
    // sql.Open(driverName, dataSourceName)
    // driverName: 注册的驱动名称，如"mysql", "postgres"等
    // dataSourceName: 数据库连接字符串，格式由具体驱动定义
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/testdb?charset=utf8mb4&parseTime=True&loc=Local")
    if err != nil {
        // sql.Open不会立即建立连接，而是验证参数。
        // 真正的连接会在第一次需要时建立。
        log.Fatalf("打开数据库连接失败: %v", err)
    }
    defer db.Close() // 确保在函数结束时关闭数据库连接

    // 2. 验证数据库连接
    // db.Ping() 尝试与数据库建立连接并验证其可用性
    err = db.Ping()
    if err != nil {
        log.Fatalf("数据库连接验证失败: %v", err)
    }
    fmt.Println("成功连接到数据库！")

    // 3. 插入数据 (使用预处理语句防止SQL注入)
    // db.Prepare() 返回一个预处理语句对象，可重复使用
    stmt, err := db.Prepare("INSERT INTO users(name, email) VALUES(?, ?)")
    if err != nil {
        log.Fatalf("预处理插入语句失败: %v", err)
    }
    defer stmt.Close() // 确保关闭预处理语句

    res, err := stmt.Exec("Alice", "alice@example.com")
    if err != nil {
        log.Fatalf("执行插入失败: %v", err)
    }
    lastInsertID, err := res.LastInsertId()
    if err != nil {
        log.Fatalf("获取最后插入ID失败: %v", err)
    }
    rowsAffected, err := res.RowsAffected()
    if err != nil {
        log.Fatalf("获取影响行数失败: %v", err)
    }
    fmt.Printf("插入成功，ID: %d, 影响行数: %d\n", lastInsertID, rowsAffected)

    // 4. 查询数据
    // db.Query() 用于执行查询，返回 Rows 对象
    rows, err := db.Query("SELECT id, name, email FROM users WHERE id = ?", lastInsertID)
    if err != nil {
        log.Fatalf("执行查询失败: %v", err)
    }
    defer rows.Close() // 确保关闭 Rows 对象

    // 遍历查询结果
    for rows.Next() {
        var id int
        var name, email string
        // rows.Scan() 将当前行的数据扫描到变量中
        if err := rows.Scan(&id, &name, &email); err != nil {
            log.Fatalf("扫描行数据失败: %v", err)
        }
        fmt.Printf("查询结果 - ID: %d, Name: %s, Email: %s\n", id, name, email)
    }
    // 检查迭代过程中是否发生错误
    if err := rows.Err(); err != nil {
        log.Fatalf("迭代查询结果错误: %v", err)
    }

    // 5. 更新数据
    updateRes, err := db.Exec("UPDATE users SET email = ? WHERE id = ?", "alice.new@example.com", lastInsertID)
    if err != nil {
        log.Fatalf("执行更新失败: %v", err)
    }
    updateRowsAffected, err := updateRes.RowsAffected()
    if err != nil {
        log.Fatalf("获取更新影响行数失败: %v", err)
    }
    fmt.Printf("更新成功，影响行数: %d\n", updateRowsAffected)

    // 6. 删除数据
    deleteRes, err := db.Exec("DELETE FROM users WHERE id = ?", lastInsertID)
    if err != nil {
        log.Fatalf("执行删除失败: %v", err)
    }
    deleteRowsAffected, err := deleteRes.RowsAffected()
    if err != nil {
        log.Fatalf("获取删除影响行数失败: %v", err)
    }
    fmt.Printf("删除成功，影响行数: %d\n", deleteRowsAffected)
}