Golang执行原生SQL的方法与示例

在使用Golang进行数据库开发时，我们通常会遇到需要执行原生SQL语句的情况。虽然Golang提供了许多优秀的ORM框架，但有时候执行原生SQL语句能为我们带来更高的灵活性和效率。本文将介绍如何使用Golang执行原生SQL，并给出一些示例。

准备工作

在开始之前，我们需要安装一个用于连接数据库的驱动程序。Golang支持多个数据库驱动，如MySQL、PostgreSQL、SQLite等。这里以MySQL为例，在终端运行以下命令安装MySQL驱动：

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

接下来，我们需要在代码中导入驱动库并打开与数据库的连接。

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

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

执行查询操作

当我们需要执行查询操作时，可以用`db.Query()`方法来执行SQL语句并返回结果集。以下是一个查询示例：

rows, err := db.Query("SELECT * FROM users")
if err != nil {
    panic(err)
}
defer rows.Close()

for rows.Next() {
    var id int
    var name string
    var age int

    err := rows.Scan(&id, &name, &age)
    if err != nil {
        panic(err)
    }

    fmt.Println(id, name, age)
}

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

执行插入、更新和删除操作

对于插入、更新和删除操作，我们可以用`db.Exec()`方法来执行SQL语句并返回受影响的行数。以下是一些示例：

// 插入数据
result, err := db.Exec("INSERT INTO users(name, age) VALUES(?,?)", "Alice", 25)
if err != nil {
    panic(err)
}

affectedRows, err := result.RowsAffected()
if err != nil {
    panic(err)
}

fmt.Println("插入成功，受影响的行数为：", affectedRows)

// 更新数据
result, err = db.Exec("UPDATE users SET age=? WHERE name=?", 30, "Alice")
if err != nil {
    panic(err)
}

affectedRows, err = result.RowsAffected()
if err != nil {
    panic(err)
}

fmt.Println("更新成功，受影响的行数为：", affectedRows)

// 删除数据
result, err = db.Exec("DELETE FROM users WHERE name=?", "Alice")
if err != nil {
    panic(err)
}

affectedRows, err = result.RowsAffected()
if err != nil {
    panic(err)
}

fmt.Println("删除成功，受影响的行数为：", affectedRows)

执行事务操作

如果需要执行一个复杂的数据库操作，可能需要使用事务来确保数据的一致性。Golang提供了`db.Begin()`方法开启一个事务，并通过`commit`和`rollback`方法来提交或回滚事务。以下是一个事务操作的示例：

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

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

_, err = stmt.Exec("Alice", 25)
if err != nil {
    tx.Rollback()
    panic(err)
}

res, err := tx.Exec("UPDATE users SET age=? WHERE name=?", 30, "Bob")
if err != nil {
    tx.Rollback()
    panic(err)
}

affectedRows, err := res.RowsAffected()
if err != nil {
    tx.Rollback()
    panic(err)
}

fmt.Println("事务操作成功，受影响的行数为：", affectedRows)

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

总结

通过上述示例，我们了解了如何使用Golang执行原生SQL语句。这种方式可以为我们提供更高的灵活性和效率，特别适用于复杂的数据库操作。在实际的开发中，我们可以根据具体需求选择使用原生SQL或ORM框架，以获得最佳的开发体验。