Golang如何优雅连接MySQL数据库?

数据库 MySQL
Go原生就支持连接数据库,所以在使用 Golang 开发时,当需要数据库交互时,即可使用database/sql包。

 Go原生就支持连接数据库,所以在使用 Golang 开发时,当需要数据库交互时,即可使用database/sql包。


在Go中访问DB需用sql.DB接口:可创建语句(statement)和事务(transaction),执行查询,获取结果。

使用DB时,除database/sql包,还需引入想使用的特定DB驱动。官方不提供实现,需要先下载三方实现,点击这里查看各种各样的实现版本。

通常DB选型MySQL,所以选型驱动为:github.com/go-sql-driver/mysql,需引入包:

"database/sql"_ "github.com/go-sql-driver/mysql"

 包名前的"_" 

import 下划线(如:import _ github/demo)的作用:当导入一个包时,该包下的文件里所有init()函数都会被执行,然而,有些时候我们并不需要把整个包都导入进来,仅仅是是希望它执行init()函数而已。这个时候就可以使用 import _ 引用该包。

上面的MySQL驱动中引入的就是MySQL包中各个init()方法,你无法通过包名来调用包中的其他函数。导入时,驱动的初始化函数会调用sql.Register将自己注册在database/sql包的全局变量sql.drivers中,以便以后通过sql.Open访问。

 案例用数据表

 初始化数据库连接[[353659]] 

 sql.Open()中的数据库连接串格式为:"用户名:密码@tcp(IP:端口)/数据库?charset=utf8"。DB的类型为:*sql.DB,有DB后即可执行CRUD。

Go将数据库操作分为两类:Query与Exec

  • Query表示查询,它会从数据库获取查询结果(一系列行,可能为空)。
  • Exec表示执行语句,它不会返回行。

常见数据库操作模式:

  • QueryRow只返回一行的查询,作为Query的一个常见特例。
  • Prepare准备一个需要多次使用的语句,供后续执行用。

查询操作

  1. var user User 
  2. rows, e := DB.Query("select * from user where id in (1,2,3)"
  3. if e == nil { 
  4.     errors.New("query incur error"
  5. for rows.Next(){ 
  6.     e := rows.Scan(user.sex, user.phone, user.nameuser.id,  
  7.       user.age) 
  8.     if e != nil{ 
  9.         fmt.Println(json.Marshal(user)) 
  10.     } 
  11. rows.Close() 
  12. // 单行查询操作 
  13. DB.QueryRow("select * from user where id=1").Scan(user.age,  
  14.   user.id, user.nameuser.phone, user.sex) 

执行流程 

  1. 使用db.Query()来发送查询到数据库,获取结果集Rows,并检查错误。
  2. 使用rows.Next()作为循环条件,迭代读取结果集。
  3. 使用rows.Scan从结果集中获取一行结果。
  4. 使用rows.Err()在退出迭代后检查错误。
  5. 使用rows.Close()关闭结果集,释放连接。

增删改和Exec

通常不会约束你查询必须用Query,只是Query会返回结果集,而Exec不会返回。所以如果你执行的是增删改操作一般用Exec会好一些。Exec返回的结果是Result,Result接口允许获取执行结果的元数据:

  1. type Result interface { 
  2.     // 用于返回自增ID,并不是所有的关系型数据库都有这个功能。 
  3.     LastInsertId() (int64, error) 
  4.     // 返回受影响的行数。 
  5.     RowsAffected() (int64, error) 

 准备查询

如果你现在想使用占位符的功能,where 的条件想以参数的形式传入,Go提供了db.Prepare语句来帮你绑定。准备查询的结果是一个准备好的语句(prepared statement),语句中可以包含执行时所需参数的占位符(即绑定值)。准备查询比拼字符串的方式好很多,它可以转义参数,避免SQL注入。同时,准备查询对于一些数据库也省去了解析和生成执行计划的开销,有利于性能。

占位符

PostgreSQL使用$N作为占位符,N是一个从1开始递增的整数,代表参数的位置,方便参数的重复使用。MySQL使用?作为占位符,SQLite两种占位符都可以,而Oracle则使用:param1的形式。

  1. MySQL               PostgreSQL            Oracle 
  2. =====               ==========            ====== 
  3. WHERE col = ?       WHERE col = $1        WHERE col = :col 
  4. VALUES(?, ?, ?)     VALUES($1, $2, $3)    VALUES(:val1, :val2, :val3) 
  5. stmt, e := DB.Prepare("select * from user where id=?"
  6. query, e := stmt.Query(1) 
  7. query.Scan() 

 事务的使用

通过db.Begin()来开启一个事务,Begin方法会返回一个事务对象Tx。在结果变量Tx上调用Commit()或者Rollback()方法会提交或回滚变更,并关闭事务。在底层,Tx会从连接池中获得一个连接并在事务过程中保持对它的独占。事务对象Tx上的方法与数据库对象sql.DB的方法一一对应,例如Query,Exec等。事务对象也可以准备(prepare)查询,由事务创建的准备语句会显式绑定到创建它的事务。

  1. //开启事务 
  2. tx, err := DB.Begin() 
  3. if err != nil { 
  4.     fmt.Println("tx fail"
  5. //准备sql语句 
  6. stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?"
  7. if err != nil { 
  8.     fmt.Println("Prepare fail"
  9.     return false 
  10. //设置参数以及执行sql语句 
  11. res, err := stmt.Exec(user.id) 
  12. if err != nil { 
  13.     fmt.Println("Exec fail"
  14.     return false 
  15. //提交事务 
  16. tx.Commit() 
  17.  
  18. 我们来一个完整的sql操作:package main 
  19.  
  20. import ( 
  21.     "database/sql" 
  22.     "encoding/json" 
  23.     "fmt" 
  24.     _ "github.com/go-sql-driver/mysql" 
  25.     "github.com/pkg/errors" 
  26.     "strings" 
  27.  
  28. //数据库配置 
  29. const ( 
  30.     userName = "root" 
  31.     password = "123456" 
  32.     ip       = "127.0.0.1" 
  33.     port     = "3306" 
  34.     dbName   = "test" 
  35.  
  36. //Db数据库连接池 
  37. var DB *sql.DB 
  38.  
  39. type User struct { 
  40.     id    int64 
  41.     name  string 
  42.     age   int8 
  43.     sex   int8 
  44.     phone string 
  45.  
  46. //注意方法名大写,就是public 
  47. func InitDB() { 
  48.     //构建连接:"用户名:密码@tcp(IP:端口)/数据库?charset=utf8" 
  49.     path := strings.Join([]string{userName, ":"password"@tcp(", ip, ":", port, ")/", dbName, "?charset=utf8"}, ""
  50.     //打开数据库,前者是驱动名,所以要导入:_ "github.com/go-sql-driver/mysql" 
  51.     DB, _ = sql.Open("mysql", path) 
  52.     //设置数据库最大连接数 
  53.     DB.SetConnMaxLifetime(100) 
  54.     //设置上数据库最大闲置连接数 
  55.     DB.SetMaxIdleConns(10) 
  56.     //验证连接 
  57.     if err := DB.Ping(); err != nil { 
  58.         fmt.Println("open database fail"
  59.         return 
  60.     } 
  61.     fmt.Println("connnect success"
  62.  
  63. //查询操作 
  64. func Query() { 
  65.     var user User 
  66.     rows, e := DB.Query("select * from user where id in (1,2,3)"
  67.     if e == nil { 
  68.         errors.New("query incur error"
  69.     } 
  70.     for rows.Next() { 
  71.         e := rows.Scan(user.sex, user.phone, user.nameuser.id, user.age) 
  72.         if e != nil { 
  73.             fmt.Println(json.Marshal(user)) 
  74.         } 
  75.     } 
  76.     rows.Close() 
  77.     DB.QueryRow("select * from user where id=1").Scan(user.age, user.id, user.nameuser.phone, user.sex) 
  78.  
  79.     stmt, e := DB.Prepare("select * from user where id=?"
  80.     query, e := stmt.Query(1) 
  81.     query.Scan() 
  82.  
  83. func DeleteUser(user User) bool { 
  84.     //开启事务 
  85.     tx, err := DB.Begin() 
  86.     if err != nil { 
  87.         fmt.Println("tx fail"
  88.     } 
  89.     //准备sql语句 
  90.     stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?"
  91.     if err != nil { 
  92.         fmt.Println("Prepare fail"
  93.         return false 
  94.     } 
  95.     //设置参数以及执行sql语句 
  96.     res, err := stmt.Exec(user.id) 
  97.     if err != nil { 
  98.         fmt.Println("Exec fail"
  99.         return false 
  100.     } 
  101.     //提交事务 
  102.     tx.Commit() 
  103.     //获得上一个insert的id 
  104.     fmt.Println(res.LastInsertId()) 
  105.     return true 
  106.  
  107. func InsertUser(user User) bool { 
  108.     //开启事务 
  109.     tx, err := DB.Begin() 
  110.     if err != nil { 
  111.         fmt.Println("tx fail"
  112.         return false 
  113.     } 
  114.     //准备sql语句 
  115.     stmt, err := tx.Prepare("INSERT INTO user (`name`, `phone`) VALUES (?, ?)"
  116.     if err != nil { 
  117.         fmt.Println("Prepare fail"
  118.         return false 
  119.     } 
  120.     //将参数传递到sql语句中并且执行 
  121.     res, err := stmt.Exec(user.nameuser.phone) 
  122.     if err != nil { 
  123.         fmt.Println("Exec fail"
  124.         return false 
  125.     } 
  126.     //将事务提交 
  127.     tx.Commit() 
  128.     //获得上一个插入自增的id 
  129.     fmt.Println(res.LastInsertId()) 
  130.     return true 
  131.  
  132. func main() { 
  133.     InitDB() 
  134.     Query() 
  135.     defer DB.Close() 

参考

https://www.cnblogs.com/rickiyang/p/11074180.html

 

责任编辑:姜华 来源: JavaEdge
相关推荐

2010-05-14 11:12:16

连接MySql

2009-11-24 16:48:15

PHP mysqli

2021-08-02 10:53:28

PythonMySQL数据库

2017-09-11 19:30:44

MySQLCmd命令连接数据库

2018-02-26 20:00:00

编程语言JavaMySQL

2009-07-07 14:56:33

JSP连接MySQL

2009-06-01 09:57:43

netbeans连接数netbeans数据库netbeans连接m

2020-09-22 15:56:31

Java

2011-05-26 13:42:50

MFC连接MySql数据库

2011-07-27 13:58:48

EclipseMySQL

2017-11-27 11:41:06

python数据库数据分析

2024-01-02 08:47:42

2010-06-07 15:24:34

Java连接MYSQL

2023-08-29 07:35:15

2011-07-05 10:16:16

Qt 数据库 SQLite

2010-10-12 12:00:42

MySQL连接

2010-06-04 09:33:28

连接MySQL数据库

2015-10-30 14:00:33

adosybaseodbc

2011-04-01 12:32:37

aspaccess数据库

2011-07-18 14:00:29

RailsOracle
点赞
收藏

51CTO技术栈公众号