Golang数据库操作

CoProphet · · 11437 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

Golang支持DB操作位于database包下,支持基本CRUD操作、事务和Prepared Statement,本文以MySQL为例。

MySQL驱动安装 
使用数据库之前,必须安装相关MySQL数据驱动[1],如果未安装驱动或者没有引入驱动库会报以下错误: 
failed to open database: sql: unknown driver "mysql" (forgotten import?) 
安装golang mysql driver 
go get github.com/go-sql-driver/mysql 
代码中还需注册数据库驱动,通过引入空白倒入[2]mysql包来完成

  1. import(
  2. "database/sql"
  3. // 引入数据库驱动注册及初始化
  4. _ "github.com/go-sql-driver/mysql" )

这段空白倒入代码实际执行mysql包的初始化代码,位于%GOPATH%/github.com/go-sql-driver/mysql/driver.go

  1. func init() {
  2. sql.Register("mysql", &MySQLDriver{})
  3. }

准备测试数据 
连接MySQL 
mysql -uroot -p 
选择数据库test 
use test 
创建测试用的users表和order表,并插入测试数据

  1. #创建user表
  2. DROP TABLE IF EXISTS `order`;
  3. DROP TABLE IF EXISTS `user`;
  4. CREATE TABLE IF NOT EXISTS `user` (`uid` SERIAL PRIMARY KEY, `name` VARCHAR(20) NOT NULL, `password` VARCHAR(20) NOT NULL) ENGINE=`innodb`, CHARACTER SET=utf8;
  5. #创建order表
  6. CREATE TABLE IF NOT EXISTS `order`(`oid` SERIAL PRIMARY KEY, `uid` BIGINT(20) UNSIGNED NOT NULL, `date` TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP, FOREIGN KEY (`uid`) REFERENCES `user`(`uid`))ENGINE=innodb,CHARACTER SET=utf8;
  7. #插入测试数据
  8. INSERT INTO `user`(`name`,`password`) VALUES('nick', 'nick'),('jacky', 'jacky');
  9. INSERT INTO `order`(`uid`) VALUES(1),(2);

连接数据库 
连接数据的DSN格式为: 
username:password@protocol(address)/dbname?param=value 
示例代码:

  1. db, err := sql.Open("mysql", "root:root@tcp(127.0.0.1:3306)/test?charset=utf8")
  2. if err != nil {
  3. fmt.Println("failed to open database:", err.Error())
  4. return
  5. }
  6. defer db.Close()

返回的DB对象,实际封装了一个数据库连接池,对于goroutine是线程安全的,可以放心使用。这个数据库连接池由"database/sql"包负责自动创建和回收。连接池的大小可以由SetMaxIdleConns指定。 
需要注意的是,创建DB对象成功,并不代表已经成功的连接了数据库,数据库连接只有在真正需要的时候才会被创建。因此如果,在创建DB对象后想验证数据库连接是否有效,可以调用Ping()或者通过

关闭数据库

  1. defer db.Close()

关闭数据库并释放全部已打开的资源。实际上,很少需要进行关闭操作,DB对象实际上长期存活并在所有的goroutine之间共享

CRUD 
DB中执行SQL通过Exec和Query方法,查询操作是通过Query完成,它会返回一个sql.Rows的结果集,包含一个游标用来遍历查询结果;Exec方法返回的是sql.Result对象,用于检测操作结果,及被影响记录数

查询

  1. // 获取USERS表中的前十行记录
  2. rows, err := db.Query("SELECT * FROM user")
  3. if err != nil {
  4. fmt.Println("fetech data failed:", err.Error())
  5. return
  6. }
  7. defer rows.Close()
  8. for rows.Next() {
  9. var uid int
  10. var name, password string
  11. rows.Scan(&uid, &name, &password)
  12. fmt.Println("uid:", uid, "name:", name, "password:", password)
  13. }

插入

  1. // 插入一条新数据
  2. result, err := db.Exec("INSERT INTO `user`(`name`,`password`) VALUES('tom', 'tom')")
  3. if err != nil {
  4. fmt.Println("insert data failed:", err.Error())
  5. return
  6. }
  7. id, err := result.LastInsertId()
  8. if err != nil {
  9. fmt.Println("fetch last insert id failed:", err.Error())
  10. return
  11. }
  12. fmt.Println("insert new record", id)

更新

  1. // 更新一条数据
  2. result, err = db.Exec("UPDATE `user` SET `password`=? WHERE `name`=?", "tom_new_password", "tom")
  3. if err != nil {
  4. fmt.Println("update data failed:", err.Error())
  5. return
  6. }
  7. num, err := result.RowsAffected()
  8. if err != nil {
  9. fmt.Println("fetch row affected failed:", err.Error())
  10. return
  11. }
  12. fmt.Println("update recors number", num)

删除

  1. // 删除数据
  2. result, err = db.Exec("DELETE FROM `user` WHERE `name`=?", "tom")
  3. if err != nil {
  4. fmt.Println("delete data failed:", err.Error())
  5. return
  6. }
  7. num, err = result.RowsAffected()
  8. if err != nil {
  9. fmt.Println("fetch row affected failed:", err.Error())
  10. return
  11. }
  12. fmt.Println("delete record number", num)

事务支持 
sql.Tx用来支持事务处理

  1. // 事务处理
  2. // 完全删除用户编号为2的用户数据
  3. tx, err := db.Begin()
  4. result, err = tx.Exec("DELETE FROM `order` WHERE uid=? ", 2)
  5. if err != nil {
  6. fmt.Println("delete data failed:", err.Error())
  7. return
  8. }
  9. num, err = result.RowsAffected()
  10. if err != nil {
  11. fmt.Println("fetch row affected failed:", err.Error())
  12. return
  13. }
  14. fmt.Println("delete record number", num)
  15. result, err = tx.Exec("DELETE FROM `user` WHERE uid=? ", 2)
  16. if err != nil {
  17. fmt.Println("delete data failed:", err.Error())
  18. return
  19. }
  20. num, err = result.RowsAffected()
  21. if err != nil {
  22. fmt.Println("fetch row affected failed:", err.Error())
  23. return
  24. }
  25. fmt.Println("delete record number", num)
  26. // 根据条件回滚或者提交
  27. // tx.Rollback()
  28. tx.Commit()

Prepared Statement 
sql.Stmt支持预备表达式,可以用来优化SQL查询提高性能,减少SQL注入的风险, DB.Prepare()和Tx.Prepare()都提供了对于预备表达式的支持。

  1. // 预备表达式
  2. stmt, err := db.Prepare("DELETE FROM `order` WHERE `oid`=?")
  3. if err != nil {
  4. fmt.Println("fetch row affected failed:", err.Error())
  5. return
  6. }
  7. result, err = stmt.Exec(1)
  8. if err != nil {
  9. fmt.Println("delete data failed:", err.Error())
  10. return
  11. }
  12. num, err = result.RowsAffected()
  13. if err != nil {
  14. fmt.Println("fetch row affected failed:", err.Error())
  15. return
  16. }
  17. fmt.Println("delete record number", num)

参考资料

本文来自:CSDN博客

感谢作者:CoProphet

查看原文:Golang数据库操作

入群交流(和以上内容无关):Go中文网 QQ 交流群:798786647 或加微信入微信群:274768166 备注:入群;关注公众号:Go语言中文网

11437 次点击  
加入收藏 微博
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传