Go语言sql包

Go语言的database/sql标准库提供了一个通用的接口来与SQL数据库进行交互。这个库定义了数据库驱动的基本接口，允许开发人员通过具体的驱动实现与不同的数据库进行交互。以下是对database/sql标准库的详细介绍，包括其主要类型和方法。

1. 导入包

要使用database/sql包，首先需要导入：

import (
    "database/sql"
    _ "github.com/go-sql-driver/mysql" // 使用MySQL驱动作为示例
)

2. 连接数据库

使用sql.Open方法创建一个数据库连接池：

db, err := sql.Open("mysql", "username:password@tcp(127.0.0.1:3306)/dbname")
if err != nil {
    log.Fatal(err)
}
defer db.Close()

sql.Open并不立即建立数据库连接，而是初始化一个连接池，实际的连接是在第一次执行查询时建立的。

3. 检查连接

可以使用Ping方法来验证与数据库的连接是否正常：

err = db.Ping()
if err != nil {
    log.Fatal(err)
}

4. 查询操作

4.1 查询单行记录

使用QueryRow方法查询单行记录：

var name string
var age int
err = db.QueryRow("SELECT name, age FROM users WHERE id = ?", 1).Scan(&name, &age)
if err != nil {
    log.Fatal(err)
}
fmt.Println(name, age)

QueryRow会返回一个*sql.Row，需要使用Scan方法将结果扫描到目标变量中。

4.2 查询多行记录

使用Query方法查询多行记录：

rows, err := db.Query("SELECT id, name FROM users")
if err != nil {
    log.Fatal(err)
}
defer rows.Close()

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

需要使用Next方法遍历结果集，并使用Scan方法将每行结果扫描到目标变量中。

5. 插入、更新和删除操作

使用Exec方法执行插入、更新和删除操作：

// 插入
result, err := db.Exec("INSERT INTO users (name, age) VALUES (?, ?)", "Alice", 30)
if err != nil {
    log.Fatal(err)
}
insertedID, _ := result.LastInsertId()
fmt.Println("Inserted ID:", insertedID)

// 更新
result, err = db.Exec("UPDATE users SET age = ? WHERE name = ?", 31, "Alice")
if err != nil {
    log.Fatal(err)
}
affectedRows, _ := result.RowsAffected()
fmt.Println("Affected Rows:", affectedRows)

// 删除
result, err = db.Exec("DELETE FROM users WHERE name = ?", "Alice")
if err != nil {
    log.Fatal(err)
}
affectedRows, _ = result.RowsAffected()
fmt.Println("Affected Rows:", affectedRows)

6. 事务处理

使用Begin方法开始一个事务：

tx, err := db.Begin()
if err != nil {
    log.Fatal(err)
}

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

err = tx.Commit()
if err != nil {
    log.Fatal(err)
}

如果事务中的操作出现错误，可以使用Rollback方法回滚事务。

7. 预处理语句

使用Prepare方法预处理SQL语句：

stmt, err := db.Prepare("INSERT INTO users (name, age) VALUES (?, ?)")
if err != nil {
    log.Fatal(err)
}
defer stmt.Close()

_, err = stmt.Exec("Bob", 25)
if err != nil {
    log.Fatal(err)
}

预处理语句可以提高执行效率，特别是当需要多次执行同一SQL语句时。

8. 使用空接口处理任意类型

可以使用空接口interface{}来处理任意类型的查询参数和结果：

func queryAny(db *sql.DB, query string, args ...interface{}) ([]map[string]interface{}, error) {
    // 执行查询语句
    rows, err := db.Query(query, args...)
    if err != nil {
        return nil, err
    }
    defer rows.Close()

    // 获取列名
    columns, err := rows.Columns()
    if err != nil {
        return nil, err
    }

    // 准备存储结果
    results := make([]map[string]interface{}, 0)

    // 遍历每一行
    for rows.Next() {
        // 创建一个map来存储列名到值的映射
        columnsMap := make(map[string]interface{})

        // 创建一个切片来存储每一列的指针
        columnPointers := make([]interface{}, len(columns))
        for i := range columnPointers {
            columnPointers[i] = new(interface{})
        }

        // 将列的值扫描到columnPointers中
        if err := rows.Scan(columnPointers...); err != nil {
            return nil, err
        }

        // 将扫描结果存入columnsMap
        for i, colName := range columns {
            columnsMap[colName] = *(columnPointers[i].(*interface{}))
        }

        // 将结果添加到results中
        results = append(results, columnsMap)
    }

    // 检查rows的错误
    if err := rows.Err(); err != nil {
        return nil, err
    }

    return results, nil
}

9. 常用接口和类型

sql.DB：表示一个数据库连接池。
sql.Tx：表示一个数据库事务。
sql.Stmt：表示一个预处理语句。
sql.Row：表示一个查询结果的单行。
sql.Rows：表示一个查询结果的多行。
sql.Result：表示一个执行结果，包含插入ID和影响行数。

10. 错误处理

在执行SQL操作时，需要处理可能发生的错误：

if err != nil {
    if err == sql.ErrNoRows {
        log.Println("No rows found")
    } else {
        log.Fatal(err)
    }
}

sql.ErrNoRows是查询操作未返回结果时的特殊错误。

通过以上内容，你可以全面了解和使用database/sql标准库来进行数据库操作。这些示例涵盖了从基础的CRUD操作到事务处理、预处理语句等各个方面。根据实际项目的需要，可以进一步扩展和优化这些功能。