17370845950

Golang如何用 database/sql 连接数据库_Golang database/sql 数据库操作实战
Go语言通过database/sql包操作数据库简洁高效,需先安装驱动如go-sql-driver/mysql并导入。使用sql.Open("mysql", DSN)连接数据库,DSN格式为用户:密码@协议(地址:端口)/数据库名,再调用db.Ping()测试连接。执行插入用db.Exec("INSERT...")获取结果和ID,查询单行用db.QueryRow().Scan()绑定变量,多行查询用db.Query()遍历rows。为防SQL注入且提升性能,应使用db.Prepare()创建预处理语句。事务通过db.Begin()开启,tx.Exec()执行操作,成功则tx.Commit()提交,失败需tx.Rollback()回滚。务必处理error并defer关闭db、rows、stmt等资源。掌握这些核心方法即可覆盖多数数据库场景。

Go语言通过database/sql包操作数据库非常简洁高效。它提供了一套标准接口,配合对应的驱动(如MySQL、PostgreSQL、SQLite等),可以实现数据库的连接、查询、插入、更新和事务处理。下面以MySQL为例,详细介绍如何使用database/sql进行数据库操作。

1. 安装驱动并导入包

Go的'database/sql'是标准库,但不包含具体数据库驱动,需要额外引入。以MySQL为例,常用驱动是github.com/go-sql-driver/mysql

安装命令:

go get -u github.com/go-sql-driver/mysql

在代码中导入:

import (
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

注意:下划线表示初始化驱动,让sql.Open能识别"mysql"类型。

2. 连接数据库

使用sql.Open创建数据库连接对象,传入驱动名和数据源名称(DSN)。

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

// 测试连接
err = db.Ping()
if err != nil {
    log.Fatal(err)
}

常见DSN格式:用户名:密码@协议(地址:端口)/数据库名。可添加参数如charset=utf8mb4&parseTime=True

3. 执行SQL操作

连接成功后,就可以执行各类SQL语句。

  • 插入数据
    result, err := db.Exec("INSERT INTO users(name, age) VALUES(?, ?)", "Alice", 25)
    if err != nil {
        log.Fatal(err)
    }
    id, _ := result.LastInsertId()
    fmt.Printf("插入ID: %d\n", id)
    
  • 查询单行:使用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.Printf("姓名: %s, 年龄: %d\n", name, age)
    
  • 查询多行:使用Query
    rows, err := db.Query("SELECT id, name, age FROM users")
    if err != nil {
        log.Fatal(err)
    }
    defer rows.Close()
    
    for rows.Next() {
        var id int
        var name string
        var age int
        rows.Scan(&id, &name, &age)
        fmt.Printf("ID: %d, 姓名: %s, 年龄: %d\n", id, name, age)
    }
    

4. 使用预处理语句防止SQL注入

频繁执行相同SQL时,建议使用Prepare提升性能并增强安全性。

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

_, err = stmt.Exec("Bob", 30)
_, err = stmt.Exec("Charlie", 35)

5. 事务处理

使用Begin开启事务,Commit提交,Rollback回滚。

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

_, err = tx.Exec("INSERT INTO users(name, age) VALUES(?, ?)", "David", 28)
if err != nil {
    tx.Rollback()
    log.Fatal(err)
}

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

基本上就这些。掌握Open、Exec、Query、Prepare和事务,就能应对大多数数据库操作场景。关键是记得处理error和关闭资源。不复杂但容易忽略细节。