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

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

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

除database/sql包，还需引入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访问。

 func init() {
   sql.Register(name:"mysql", &MySQLDriver{})
 }

案例用数据表

 CREATE TABLE user (
   id bigint(20) NOT NULL AUTO_INCREMENT,
   name varchar(45) DEFAULT '',
   age int(11) NOT NULL DEFAULT '0',
   sex varchar(45) NOT NULL DEFAULT '0',
   phone varchar(45) NOT NULL DEFAULT '',
   PRIMARY KEY (id)
 ) ENGINE=InnoDB
 AUTO_INCREMENT = 3
 DEFAULT CHARSET=utf8;

初始化数据库连接

sql.Open()中的数据库连接串格式为：

 用户名:密码@tcp(IP:端口)/数据库?charset=utf8

DB的类型为:*sql.DB，有DB后即可执行CRUD。

Go将数据库操作分为两类：

• Query表示查询，会从数据库获取查询结果（一系列行，可能为空）

• Exec表示执行语句，不会返回行

常见数据库操作模式：

• QueryRow只返回一行的查询，作为Query的一个常见特例

• Prepare准备一个需要多次使用的语句，供后续执行用

1 查询操作

 Explainvar user User
 rows, e := DB.Query("select * from user where id in (1,2,3)")
 if e == nil {
     errors.New("query incur error")
 }
 for rows.Next(){
     e := rows.Scan(user.sex, user.phone, user.name, user.id, 
       user.age)
     if e != nil{
         fmt.Println(json.Marshal(user))
     }
 }
 rows.Close()
 // 单行查询操作
 DB.QueryRow("select * from user where id=1").Scan(user.age, 
   user.id, user.name, user.phone, user.sex)

2 执行流程

1. 使用db.Query()来发送查询到数据库，获取结果集Rows，并检查错误。

2. 使用rows.Next()作为循环条件，迭代读取结果集。

3. 使用rows.Scan从结果集中获取一行结果。

4. 使用rows.Err()在退出迭代后检查错误。

5. 使用rows.Close()关闭结果集，释放连接。

3 增删改和Exec

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

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

4 准备查询

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

占位符

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

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

5 事务的使用

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

 Explain//开启事务
 tx, err := DB.Begin()
 if err != nil {
     fmt.Println("tx fail")
 }
 //准备sql语句
 stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?")
 if err != nil {
     fmt.Println("Prepare fail")
     return false
 }
 //设置参数以及执行sql语句
 res, err := stmt.Exec(user.id)
 if err != nil {
     fmt.Println("Exec fail")
     return false
 }
 //提交事务
 tx.Commit()
 ​
 ​
 完整sql操作：
 ​
 package main
 import (
     "database/sql"
     "encoding/json"
     "fmt"
     _ "github.com/go-sql-driver/mysql"
     "github.com/pkg/errors"
     "strings"
 )
 ​
 ​
 //数据库配置
 const (
     userName = "root"
     password = "123456"
     ip       = "127.0.0.1"
     port     = "3306"
     dbName   = "test"
 )
 ​
 ​
 //Db数据库连接池
 var DB *sql.DB
 ​
 ​
 type User struct {
     id    int64
     name  string
     age   int8
     sex   int8
     phone string
 }
 ​
 ​
 //注意方法名大写，就是public
 func InitDB() {
     //构建连接："用户名:密码@tcp(IP:端口)/数据库?charset=utf8"
     path := strings.Join([]string{userName, ":", password, "@tcp(", ip, ":", port, ")/", dbName, "?charset=utf8"}, "")
     //打开数据库,前者是驱动名，所以要导入：_ "github.com/go-sql-driver/mysql"
     DB, _ = sql.Open("mysql", path)
     //设置数据库最大连接数
     DB.SetConnMaxLifetime(100)
     //设置上数据库最大闲置连接数
     DB.SetMaxIdleConns(10)
     //验证连接
     if err := DB.Ping(); err != nil {
         fmt.Println("open database fail")
         return
     }
     fmt.Println("connnect success")
 }
 ​
 // 查询操作
 func Query() {
     var user User
     rows, e := DB.Query("select * from user where id in (1,2,3)")
     if e == nil {
         errors.New("query incur error")
     }
     for rows.Next() {
         e := rows.Scan(user.sex, user.phone, user.name, user.id, user.age)
         if e != nil {
             fmt.Println(json.Marshal(user))
         }
     }
     rows.Close()
     DB.QueryRow("select * from user where id=1").Scan(user.age, user.id, user.name, user.phone, user.sex)
 ​
 ​
     stmt, e := DB.Prepare("select * from user where id=?")
     query, e := stmt.Query(1)
     query.Scan()
 }
 ​
 ​
 func DeleteUser(user User) bool {
     //开启事务
     tx, err := DB.Begin()
     if err != nil {
         fmt.Println("tx fail")
     }
     //准备sql语句
     stmt, err := tx.Prepare("DELETE FROM user WHERE id = ?")
     if err != nil {
         fmt.Println("Prepare fail")
         return false
     }
     //设置参数以及执行sql语句
     res, err := stmt.Exec(user.id)
     if err != nil {
         fmt.Println("Exec fail")
         return false
     }
     //提交事务
     tx.Commit()
     //获得上一个insert的id
     fmt.Println(res.LastInsertId())
     return true
 }
 ​
 ​
 func InsertUser(user User) bool {
     //开启事务
     tx, err := DB.Begin()
     if err != nil {
         fmt.Println("tx fail")
         return false
     }
     //准备sql语句
     stmt, err := tx.Prepare("INSERT INTO user (`name`, `phone`) VALUES (?, ?)")
     if err != nil {
         fmt.Println("Prepare fail")
         return false
     }
     //将参数传递到sql语句中并且执行
     res, err := stmt.Exec(user.name, user.phone)
     if err != nil {
         fmt.Println("Exec fail")
         return false
     }
     //将事务提交
     tx.Commit()
     //获得上一个插入自增的id
     fmt.Println(res.LastInsertId())
     return true
 }
 ​
 ​
 func main() {
     InitDB()
     Query()
     defer DB.Close()
 }

参考

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