前言
在Go语言中，连接和操作数据库通常使用database/sql包，它提供了一个数据库抽象层，支持多种数据库引擎，如MySQL、PostgreSQL、SQLite等。下面我将以MySQL为例，详细讲解如何使用Go语言连接和操作数据库。
1. 安装MySQL驱动
首先，你需要安装一个MySQL驱动，以便能够与MySQL数据库进行通信。推荐使用github.com/go-sql-driver/mysql，可以通过以下命令安装：
go get -u github.com/go-sql-driver/mysql
2. 连接数据库
使用sql.Open函数建立与MySQL数据库的连接。该函数接受两个参数：数据库驱动名称和连接字符串。连接字符串包含了连接数据库所需的所有信息，如用户名、密码、主机名、端口和数据库名等。
示例代码：
package main
import (
"database/sql"
"fmt"
"log"
_ "github.com/go-sql-driver/mysql"
)
func main() {
// 定义连接字符串
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)
}
fmt.Println("Connected to database successfully!")
}
3. 执行查询
使用db.Query方法执行SQL查询，并获取*sql.Rows结果集。你可以使用rows.Next()方法遍历结果集，并使用rows.Scan()方法将每行数据扫描到变量中。
示例代码：
package main
import (
"database/sql"
"fmt"
"log"
_ "github.com/go-sql-driver/mysql"
)
func main() {
// 连接数据库
db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/dbname")
if err != nil {
log.Fatal(err)
}
defer db.Close()
// 执行查询
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.Printf("ID: %d, Name: %s\n", id, name)
}
// 检查遍历过程中是否出现错误
if err := rows.Err(); err != nil {
log.Fatal(err)
}
}
4. 执行插入、更新和删除操作
使用db.Exec方法执行SQL语句，如插入、更新和删除操作。db.Exec方法返回一个sql.Result对象，你可以使用它来获取受影响的行数。
示例代码：
package main
import (
"database/sql"
"fmt"
"log"
_ "github.com/go-sql-driver/mysql"
)
func main() {
// 连接数据库
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.Exec("INSERT INTO users (name, age) VALUES (?, ?)", "Alice", 25)
if err != nil {
log.Fatal(err)
}
fmt.Println("Data inserted successfully!")
// 更新数据
_, err = db.Exec("UPDATE users SET age = ? WHERE id = ?", 30, 1)
if err != nil {
log.Fatal(err)
}
fmt.Println("Data updated successfully!")
// 删除数据
_, err = db.Exec("DELETE FROM users WHERE id = ?", 1)
if err != nil {
log.Fatal(err)
}
fmt.Println("Data deleted successfully!")
}
以上是使用Go语言连接和操作MySQL数据库的基本步骤。
