golang连接mysql需使用database/sql包及驱动。1.安装推荐的mysql驱动github.com/go-sql-driver/mysql;2.通过sql.open创建连接池并用db.ping测试连接;3.查询时使用rows.scan读取数据并确保关闭rows;4.更新操作使用db.exec获取受影响行数;5.合理配置连接池参数如最大连接数和空闲数;6.使用预编译语句防止sql注入;7.事务处理通过db.begin开启,tx.commit提交或tx.rollback回滚以保证一致性。

直接操作数据库,对于任何应用来说,都是绕不开的一环。Golang在数据库操作上提供了标准库database/sql,但要真正用好它,还是需要一些技巧和最佳实践的。这篇文章就来聊聊Golang连接MySQL的一些实战经验。

连接MySQL,实际上就是通过database/sql包,加上MySQL的驱动,来实现数据的增删改查。选择一个合适的MySQL驱动至关重要,我个人比较推荐github.com/go-sql-driver/mysql,因为它比较成熟稳定,社区支持也好。

解决方案
-
安装MySQL驱动:
立即学习“go语言免费学习笔记(深入)”;
go get -u github.com/go-sql-driver/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("Successfully connected to MySQL!") }注意,
sql.Open只是创建了一个数据库连接池,并没有真正建立连接。db.Ping()可以用来测试连接是否成功。 另外,_ "github.com/go-sql-driver/mysql"这行代码是必须的,它会注册MySQL驱动到database/sql包中。 -
执行查询:
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.Println(id, name) } err = rows.Err() if err != nil { log.Fatal(err) }这里需要注意的是,一定要关闭
rows,防止资源泄露。 另外,rows.Err()用于检查在rows.Next()过程中是否发生了错误。 -
执行更新:
result, err := db.Exec("UPDATE users SET name = ? WHERE id = ?", "new_name", 1) if err != nil { log.Fatal(err) } rowsAffected, err := result.RowsAffected() if err != nil { log.Fatal(err) } fmt.Println("Rows affected:", rowsAffected)db.Exec用于执行更新、插入、删除等操作。result.RowsAffected()可以获取受影响的行数。
如何优雅地处理数据库连接池?
数据库连接是昂贵的资源,频繁地创建和销毁连接会严重影响性能。所以,连接池是必须的。database/sql包本身就实现了连接池。但是,我们需要合理配置连接池的大小。
-
db.SetMaxOpenConns(n int): 设置连接池的最大连接数。 -
db.SetMaxIdleConns(n int): 设置连接池的最大空闲连接数。 -
db.SetConnMaxLifetime(d time.Duration): 设置连接的最大生存时间。
一个合理的配置,需要根据你的应用场景来调整。例如,如果你的应用并发量很高,可以适当增加MaxOpenConns。如果你的应用对数据库的延迟很敏感,可以适当增加MaxIdleConns。
如何防止SQL注入?
SQL注入是Web应用安全中一个非常常见的漏洞。在Golang中,我们可以使用预编译语句来防止SQL注入。
stmt, err := db.Prepare("SELECT id, name FROM users WHERE name = ?")
if err != nil {
log.Fatal(err)
}
defer stmt.Close()
rows, err := stmt.Query("user' OR '1'='1") // 即使这里有恶意代码,也不会被执行
if err != nil {
log.Fatal(err)
}
defer rows.Close()
// ... 处理 rows使用预编译语句,我们可以将SQL语句和参数分开,从而防止SQL注入。 永远不要直接拼接SQL语句,这是一个非常危险的做法。
如何处理事务?
事务是保证数据一致性的重要手段。在Golang中,我们可以使用db.Begin()来开启一个事务,然后使用tx.Commit()来提交事务,或者使用tx.Rollback()来回滚事务。
tx, err := db.Begin()
if err != nil {
log.Fatal(err)
}
defer tx.Rollback() // 确保在函数退出时回滚事务
_, err = tx.Exec("UPDATE accounts SET balance = balance - ? WHERE id = ?", 100, 1)
if err != nil {
log.Fatal(err)
}
_, err = tx.Exec("UPDATE accounts SET balance = balance + ? WHERE id = ?", 100, 2)
if err != nil {
log.Fatal(err)
}
err = tx.Commit()
if err != nil {
log.Fatal(err)
}
fmt.Println("Transaction completed successfully!")在事务中,如果任何一个步骤失败,我们都需要回滚事务,保证数据的一致性。 另外,需要注意的是,defer tx.Rollback() 必须在 tx.Commit() 之前, 这样才能保证即使在 Commit 之前发生错误,事务也能被回滚。










