Go语言集成SQLite3数据库：使用go-sqlite3库的实践指南

go get github.com/mattn/go-sqlite3

sqlite3.go: In function ‘_cgo_7e09c699097a_Cfunc_sqlite3_prepare_v2’:
sqlite3.go:198:2: warning: passing argument 5 of ‘sqlite3_prepare_v2’ from incompatible pointer type [enabled by default]
/usr/local/include/sqlite3.h:2924:16: note: expected ‘const char **’ but argument is of type ‘char **’

package main

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

    _ "github.com/mattn/go-sqlite3" // 导入驱动，但不在代码中直接使用
)

const dbFileName = "test.db"

func main() {
    // 1. 打开或创建数据库文件
    db, err := sql.Open("sqlite3", dbFileName)
    if err != nil {
        log.Fatalf("无法打开数据库: %v", err)
    }
    defer db.Close() // 确保数据库连接在函数结束时关闭

    // 2. 创建表
    err = createTable(db)
    if err != nil {
        log.Fatalf("无法创建表: %v", err)
    }
    fmt.Println("表创建成功或已存在。")

    // 3. 插入数据
    lastID, err := insertData(db, "Alice", 30)
    if err != nil {
        log.Fatalf("插入数据失败: %v", err)
    }
    fmt.Printf("插入成功，最后插入ID: %d\n", lastID)

    _, err = insertData(db, "Bob", 25)
    if err != nil {
        log.Fatalf("插入数据失败: %v", err)
    }

    // 4. 查询所有数据
    fmt.Println("\n--- 查询所有数据 ---")
    err = queryAllData(db)
    if err != nil {
        log.Fatalf("查询所有数据失败: %v", err)
    }

    // 5. 根据条件查询数据
    fmt.Println("\n--- 查询年龄大于28的数据 ---")
    err = queryDataByAge(db, 28)
    if err != nil {
        log.Fatalf("条件查询失败: %v", err)
    }

    // 6. 更新数据
    rowsAffected, err := updateData(db, "Alicia", 32, lastID)
    if err != nil {
        log.Fatalf("更新数据失败: %v", err)
    }
    fmt.Printf("\n更新成功，影响行数: %d\n", rowsAffected)
    fmt.Println("--- 更新后查询所有数据 ---")
    err = queryAllData(db)
    if err != nil {
        log.Fatalf("查询所有数据失败: %v", err)
    }

    // 7. 删除数据
    rowsAffected, err = deleteData(db, lastID)
    if err != nil {
        log.Fatalf("删除数据失败: %v", err)
    }
    fmt.Printf("\n删除成功，影响行数: %d\n", rowsAffected)
    fmt.Println("--- 删除后查询所有数据 ---")
    err = queryAllData(db)
    if err != nil {
        log.Fatalf("查询所有数据失败: %v", err)
    }

    // 8. 清理：删除数据库文件 (可选)
    // defer os.Remove(dbFileName)
}

// createTable 创建一个名为 'users' 的表
func createTable(db *sql.DB) error {
    sqlStmt := `
    CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        age INTEGER
    );`
    _, err := db.Exec(sqlStmt)
    return err
}

// insertData 插入一条新用户数据
func insertData(db *sql.DB, name string, age int) (int64, error) {
    stmt, err := db.Prepare("INSERT INTO users(name, age) VALUES(?, ?)")
    if err != nil {
        return 0, fmt.Errorf("准备插入语句失败: %w", err)
    }
    defer stmt.Close()

    res, err := stmt.Exec(name, age)
    if err != nil {
        return 0, fmt.Errorf("执行插入失败: %w", err)
    }

    id, err := res.LastInsertId()
    if err != nil {
        return 0, fmt.Errorf("获取最后插入ID失败: %w", err)
    }
    return id, nil
}

// queryAllData 查询并打印所有用户数据
func queryAllData(db *sql.DB) error {
    rows, err := db.Query("SELECT id, name, age FROM users")
    if err != nil {
        return fmt.Errorf("查询所有数据失败: %w", 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.Printf("扫描行失败: %v", err)
            continue
        }
        fmt.Printf("ID: %d, Name: %s, Age: %d\n", id, name, age)
    }
    return rows.Err() // 检查迭代过程中是否有错误
}

// queryDataByAge 根据年龄查询数据
func queryDataByAge(db *sql.DB, minAge int) error {
    stmt, err := db.Prepare("SELECT id, name, age FROM users WHERE age > ?")
    if err != nil {
        return fmt.Errorf("准备查询语句失败: %w", err)
    }
    defer stmt.Close()

    rows, err := stmt.Query(minAge)
    if err != nil {
        return fmt.Errorf("执行查询失败: %w", 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.Printf("扫描行失败: %v", err)
            continue
        }
        fmt.Printf("ID: %d, Name: %s, Age: %d\n", id, name, age)
    }
    return rows.Err()
}

// updateData 更新指定ID的用户数据
func updateData(db *sql.DB, newName string, newAge int, id int64) (int64, error) {
    stmt, err := db.Prepare("UPDATE users SET name = ?, age = ? WHERE id = ?")
    if err != nil {
        return 0, fmt.Errorf("准备更新语句失败: %w", err)
    }
    defer stmt.Close()

    res, err := stmt.Exec(newName, newAge, id)
    if err != nil {
        return 0, fmt.Errorf("执行更新失败: %w", err)
    }

    rowsAffected, err := res.RowsAffected()
    if err != nil {
        return 0, fmt.Errorf("获取影响行数失败: %w", err)
    }
    return rowsAffected, nil
}

// deleteData 删除指定ID的用户数据
func deleteData(db *sql.DB, id int64) (int64, error) {
    stmt, err := db.Prepare("DELETE FROM users WHERE id = ?")
    if err != nil {
        return 0, fmt.Errorf("准备删除语句失败: %w", err)
    }
    defer stmt.Close()

    res, err := stmt.Exec(id)
    if err != nil {
        return 0, fmt.Errorf("执行删除失败: %w", err)
    }

    rowsAffected, err := res.RowsAffected()
    if err != nil {
        return 0, fmt.Errorf("获取影响行数失败: %w", err)
    }
    return rowsAffected, nil
}