Golang数据库操作指南_go连接MySQL实战

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，因为它比较成熟稳定，社区支持也好。

解决方案

1. 安装MySQL驱动：

   立即学习“go语言免费学习笔记（深入）”；

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

   登录后复制

2. 连接数据库：

   

   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包中。

3. 执行查询：

   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()过程中是否发生了错误。

   

   怪兽AI数字人

   数字人短视频创作，数字人直播，实时驱动数字人

   44

   查看详情

4. 执行更新：

   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 之前发生错误，事务也能被回滚。

以上就是Golang数据库操作指南_go连接MySQL实战的详细内容，更多请关注php中文网其它相关文章！