一切顺利：在 Go 中使用 database/sql 连接数据库的指南

2024-02-15 07:07:17

前言

数据库是现代软件开发中不可或缺的一部分，Go 语言提供了强大的 database/sql 包来帮助开发者轻松连接和操作数据库。本指南将带领您一步步连接到数据库，并执行基本的 CRUD 操作，让您轻松上手 Go 中的数据库操作。

连接数据库

首先，您需要创建一个数据库连接池。连接池是一种用于管理数据库连接的机制，它可以提高应用程序的性能和稳定性。要创建连接池，您可以使用 database/sql 包中的 Open 函数。

import (
	"database/sql"
	_ "github.com/go-sql-driver/mysql" // MySQL 驱动程序
)

func main() {
	// 创建数据库连接池
	db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/database")
	if err != nil {
		panic(err)
	}

	// 关闭数据库连接池
	defer db.Close()
}

上面的代码使用了 MySQL 驱动程序，如果您使用的是其他数据库，需要替换相应的驱动程序。

执行查询

连接到数据库后，您可以使用 Query 函数执行查询。Query 函数返回一个 sql.Rows 对象，其中包含查询结果。

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

// 遍历查询结果
for rows.Next() {
	var id int
	var name string
	err := rows.Scan(&id, &name)
	if err != nil {
		panic(err)
	}

	fmt.Println(id, name)
}

// 关闭查询结果
rows.Close()

上面的代码执行了一个查询，并打印出了查询结果。

执行更新

如果您需要执行更新操作，可以使用 Exec 函数。Exec 函数返回受影响的行数。

result, err := db.Exec("UPDATE users SET name = ? WHERE id = ?", "John Doe", 1)
if err != nil {
	panic(err)
}

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

fmt.Println("Rows affected:", rowsAffected)