在go语言中,连接mysql数据库需使用database/sql包并导入github.com/go-sql-driver/mysql驱动,通过sql.open()创建数据库连接池并用db.ping()验证连接,结合setmaxopenconns、setmaxidleconns和setconnmaxlifetime优化连接池,使用预处理语句防止sql注入并提升性能,同时通过errors.as等方法精细化处理各类数据库错误,最终在程序退出时调用db.close()释放资源,完成安全高效的数据库操作。

在Go语言中,要连接MySQL数据库,通常我们会依赖标准库
database/sql
github.com/go-sql-driver/mysql
package main
import (
"database/sql"
"fmt"
"log"
_ "github.com/go-sql-driver/mysql" // 导入MySQL驱动,注意下划线,表示只导入其init函数
)
func main() {
// 构建数据源名称 (DSN)
// 格式通常是 user:password@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local
// parseTime=True 很关键,它会将MySQL的DATETIME/TIMESTAMP类型自动转换为Go的time.Time
// loc=Local 确保时间解析基于本地时区
dsn := "user:password@tcp(127.0.0.1:3306)/testdb?charset=utf8mb4&parseTime=True&loc=Local"
// 使用 sql.Open() 建立数据库连接
// 注意:sql.Open() 不会立即建立网络连接,它只是验证参数并返回一个DB对象。
// 实际的网络连接会在第一次需要时(如Ping、Query、Exec等)才建立。
db, err := sql.Open("mysql", dsn)
if err != nil {
// 这里捕获的错误通常是DSN格式错误或驱动加载问题
log.Fatalf("无法打开数据库连接: %v", err)
}
// 确保在函数退出时关闭数据库连接,释放资源
defer func() {
if cerr := db.Close(); cerr != nil {
log.Printf("关闭数据库连接失败: %v", cerr)
}
}()
// 通过 Ping() 方法验证数据库连接是否有效
// 这是实际建立连接并测试其可用性的第一步
if err = db.Ping(); err != nil {
log.Fatalf("无法连接到数据库: %v", err)
}
fmt.Println("成功连接到MySQL数据库!")
// 接下来可以执行数据库操作,例如查询数据
var version string
err = db.QueryRow("SELECT VERSION()").Scan(&version)
if err != nil {
log.Fatalf("查询版本失败: %v", err)
}
fmt.Printf("MySQL版本: %s\n", version)
// 示例:插入数据
// stmt, err := db.Prepare("INSERT INTO users(name, email) VALUES(?, ?)")
// if err != nil {
// log.Fatalf("准备语句失败: %v", err)
// }
// defer stmt.Close()
// res, err := stmt.Exec("Alice", "alice@example.com")
// if err != nil {
// log.Fatalf("执行插入失败: %v", err)
// }
// lastID, _ := res.LastInsertId()
// rowsAffected, _ := res.RowsAffected()
// fmt.Printf("插入成功,ID: %d, 影响行数: %d\n", lastID, rowsAffected)
// 示例:查询多行数据
// rows, err := db.Query("SELECT id, name, email FROM users")
// if err != nil {
// log.Fatalf("查询多行数据失败: %v", err)
// }
// defer rows.Close() // 确保关闭rows,释放底层连接
// for rows.Next() {
// var id int
// var name, email string
// if err := rows.Scan(&id, &name, &email); err != nil {
// log.Printf("扫描行数据失败: %v", err)
// continue
// }
// fmt.Printf("ID: %d, Name: %s, Email: %s\n", id, name, email)
// }
// if err := rows.Err(); err != nil { // 检查迭代过程中是否发生错误
// log.Fatalf("遍历行时发生错误: %v", err)
// }
}database/sql
database/sql
sql.Open
*sql.DB
但仅仅依赖默认设置是不够的,尤其是在高并发场景下。我们需要通过一些方法来优化这个连接池的行为:
立即学习“go语言免费学习笔记(深入)”;
db.SetMaxOpenConns(n int)
db.SetMaxIdleConns(n int)
MaxOpenConns
db.SetConnMaxLifetime(d time.Duration)
wait_timeout
wait_timeout
一个常见的误区是,很多人以为
db.Close()
*sql.DB
db.Close()
Open
Close
log.Fatal
在实际的Go应用中,数据库操作的错误处理远比示例代码中的
log.Fatal
log.Fatal
首先,
database/sql
sql.ErrNoRows
QueryRow().Scan()
rows.Next()
sql.ErrNoRows
var name string
err := db.QueryRow("SELECT name FROM users WHERE id = ?", userID).Scan(&name)
if err == sql.ErrNoRows {
fmt.Println("用户不存在")
return nil // 或者返回自定义的 ErrNotFound
}
if err != nil {
return fmt.Errorf("查询用户失败: %w", err) // 包装其他错误
}
// 处理找到的用户网络错误或连接问题: 当数据库服务器宕机、网络中断或连接池耗尽时,会返回这类错误。这些通常是
net.OpError
errors.Is
errors.As
数据库约束错误: 例如,插入重复的唯一键、违反外键约束等。这些错误通常由数据库驱动包装后返回。例如,MySQL驱动可能会返回一个包含特定错误码的错误。你可以检查这些错误码来判断具体的错误类型,并给予用户友好的提示。
// 假设 dbErr 是一个来自数据库操作的错误
var mysqlErr *mysql.MySQLError
if errors.As(dbErr, &mysqlErr) {
switch mysqlErr.Number {
case 1062: // Duplicate entry for key 'PRIMARY'
return fmt.Errorf("该记录已存在: %w", dbErr)
case 1452: // Cannot add or update a child row: a foreign key constraint fails
return fmt.Errorf("关联数据不存在或无效: %w", dbErr)
default:
return fmt.Errorf("数据库操作失败: %w", dbErr)
}
}
return fmt.Errorf("未知错误: %w", dbErr)迭代器错误: 当使用
db.Query()
rows.Scan()
rows.Err()
rows, err := db.Query("SELECT id, name FROM users")
if err != nil {
return fmt.Errorf("查询用户列表失败: %w", err)
}
defer rows.Close() // 关键:确保关闭rows
for rows.Next() {
// ... scan data ...
}
if err = rows.Err(); err != nil { // 检查循环过程中是否出错
return fmt.Errorf("遍历用户列表时出错: %w", err)
}记住,
defer rows.Close()
for rows.Next()
rows.Close()
在进行数据库操作时,尤其是涉及用户输入的数据,预处理语句(Prepared Statements)是不可或缺的。它主要解决了两个核心问题:SQL注入攻击和重复执行相同SQL语句的性能问题。
SQL注入攻击:这是最直接也最危险的问题。如果你直接将用户输入拼接到SQL字符串中,恶意用户可以通过输入特定的字符来改变SQL语句的意图,从而窃取、修改或删除数据。例如:
// 危险的做法:直接拼接
query := fmt.Sprintf("SELECT * FROM users WHERE username = '%s' AND password = '%s'", username, password)
// 如果 username 输入 ' OR '1'='1 -- ,密码随便输,SQL就变成了:
// SELECT * FROM users WHERE username = '' OR '1'='1' -- ' AND password = ''
// 这会导致绕过密码验证,登录成功。预处理语句通过将SQL语句的结构与数据分离来解决这个问题。你先定义一个带有占位符的SQL模板,然后将数据作为参数绑定到这些占位符上。数据库驱动会负责正确地转义这些参数,确保它们被视为数据而不是SQL代码的一部分。
性能提升:当你的应用程序需要频繁执行相同的SQL语句(例如,批量插入、更新或查询),但每次只有参数不同时,预处理语句可以显著提升性能。数据库收到预处理语句后,会对其进行解析、编译和优化,生成一个执行计划。后续的执行只需要将新的参数传递给这个已编译的计划,而无需重复解析和编译的开销。这在高并发场景下尤为明显。
在Go的
database/sql
准备语句 (db.Prepare(query string)
db.Prepare()
*sql.Stmt
*sql.Stmt
stmt, err := db.Prepare("INSERT INTO users(name, email) VALUES(?, ?)")
if err != nil {
log.Fatalf("准备插入语句失败: %v", err)
}
// 确保在函数退出时关闭预处理语句,释放资源
defer stmt.Close()注意:
defer stmt.Close()
defer db.Close()
stmt.Close()
执行语句 (stmt.Exec(args ...interface{})stmt.Query(args ...interface{})
Exec
Query
*sql.Stmt
// 插入数据
res, err := stmt.Exec("Bob", "bob@example.com")
if err != nil {
log.Fatalf("执行插入失败: %v", err)
}
lastID, _ := res.LastInsertId()
rowsAffected, _ := res.RowsAffected()
fmt.Printf("插入成功,ID: %d, 影响行数: %d\n", lastID, rowsAffected)
// 查询数据 (如果stmt是SELECT语句)
// rows, err := stmt.Query("some_id")
// defer rows.Close()
// ...使用预处理语句是构建安全、高效数据库应用程序的基石。它不仅是防御SQL注入的第一道防线,也是提升应用程序性能的关键优化手段。在我看来,任何涉及参数的数据库操作,都应该优先考虑使用预处理语句。
以上就是如何用Golang连接MySQL数据库 集成database/sql标准库的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号