Go-SQLite3 使用指南

go-sqlite3 是 Go 语言的一个 SQLite3 数据库驱动，它通过 cgo 封装了 SQLite3 的 C 语言 API。下面我将详细介绍如何使用这个库。

安装

首先需要安装 go-sqlite3：

go get github.com/mattn/go-sqlite3

注意：使用此库需要安装 GCC 编译环境，因为它使用了 cgo。

基本使用

1. 连接数据库

package main

import (
	"database/sql"
	"fmt"
	"log"
	
	_ "github.com/mattn/go-sqlite3"
)

func main() {
	// 打开数据库连接
	db, err := sql.Open("sqlite3", "./test.db")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()
	
	// 测试连接
	err = db.Ping()
	if err != nil {
		log.Fatal(err)
	}
	
	fmt.Println("成功连接到SQLite数据库")
}

2. 创建表

func createTable(db *sql.DB) {
	sqlStmt := `
	CREATE TABLE IF NOT EXISTS users (
		id INTEGER PRIMARY KEY AUTOINCREMENT,
		name TEXT NOT NULL,
		age INTEGER,
		email TEXT UNIQUE
	);
	`
	_, err := db.Exec(sqlStmt)
	if err != nil {
		log.Printf("%q: %s\n", err, sqlStmt)
		return
	}
	fmt.Println("表创建成功")
}

3. 插入数据

func insertUser(db *sql.DB, name string, age int, email string) {
	stmt, err := db.Prepare("INSERT INTO users(name, age, email) VALUES(?, ?, ?)")
	if err != nil {
		log.Fatal(err)
	}
	defer stmt.Close()

	res, err := stmt.Exec(name, age, email)
	if err != nil {
		log.Fatal(err)
	}

	id, err := res.LastInsertId()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("插入成功，ID: %d\n", id)
}

4. 查询数据

func queryUsers(db *sql.DB) {
	rows, err := db.Query("SELECT id, name, age, email FROM users")
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	for rows.Next() {
		var id int
		var name string
		var age int
		var email string
		err = rows.Scan(&id, &name, &age, &email)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Printf("ID: %d, Name: %s, Age: %d, Email: %s\n", id, name, age, email)
	}

	err = rows.Err()
	if err != nil {
		log.Fatal(err)
	}
}

5. 更新数据

func updateUser(db *sql.DB, id int, newName string) {
	stmt, err := db.Prepare("UPDATE users SET name = ? WHERE id = ?")
	if err != nil {
		log.Fatal(err)
	}
	defer stmt.Close()

	res, err := stmt.Exec(newName, id)
	if err != nil {
		log.Fatal(err)
	}

	affected, err := res.RowsAffected()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("更新了 %d 行\n", affected)
}

6. 删除数据

func deleteUser(db *sql.DB, id int) {
	stmt, err := db.Prepare("DELETE FROM users WHERE id = ?")
	if err != nil {
		log.Fatal(err)
	}
	defer stmt.Close()

	res, err := stmt.Exec(id)
	if err != nil {
		log.Fatal(err)
	}

	affected, err := res.RowsAffected()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("删除了 %d 行\n", affected)
}

高级功能

1. 事务处理

func transactionExample(db *sql.DB) {
	tx, err := db.Begin()
	if err != nil {
		log.Fatal(err)
	}

	// 在事务中执行多个操作
	_, err = tx.Exec("INSERT INTO users(name, age, email) VALUES(?, ?, ?)", "Alice", 25, "alice@example.com")
	if err != nil {
		tx.Rollback()
		log.Fatal(err)
	}

	_, err = tx.Exec("UPDATE users SET age = ? WHERE name = ?", 26, "Bob")
	if err != nil {
		tx.Rollback()
		log.Fatal(err)
	}

	err = tx.Commit()
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("事务执行成功")
}

2. 使用连接池

func connectionPoolExample() {
	db, err := sql.Open("sqlite3", "./test.db")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	// 设置连接池参数
	db.SetMaxOpenConns(10) // 最大打开连接数
	db.SetMaxIdleConns(5)  // 最大空闲连接数
	db.SetConnMaxLifetime(time.Hour) // 连接最大存活时间

	// 使用数据库...
}

3. 使用预编译语句

func preparedStatementExample(db *sql.DB) {
	// 准备语句
	stmt, err := db.Prepare("SELECT name FROM users WHERE age > ?")
	if err != nil {
		log.Fatal(err)
	}
	defer stmt.Close()

	// 多次使用同一语句
	rows, err := stmt.Query(20)
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	for rows.Next() {
		var name string
		err = rows.Scan(&name)
		if err != nil {
			log.Fatal(err)
		}
		fmt.Println(name)
	}
}

注意事项

1. 并发访问：SQLite3 是文件数据库，多个写操作需要串行执行。go-sqlite3 内部已经处理了并发问题。

2. 内存模式：可以使用 :memory: 作为数据库文件名创建内存数据库：

   db, err := sql.Open("sqlite3", ":memory:")
   

3. 性能优化：对于大量插入操作，考虑使用事务可以显著提高性能。

4. CGO依赖：由于使用了 cgo，交叉编译时需要特别注意。

5. 扩展支持：go-sqlite3 支持 SQLite3 的扩展，可以通过 Register 函数注册自定义扩展。

希望这个指南能帮助你开始使用 go-sqlite3。根据你的具体需求，可以进一步探索更高级的功能。