1 Go 数据库IO（一）：Mysql操作

Go使用Mysql

1.database/sql标准包

sql包提供了保证SQL或类SQL数据库的泛用接口。使用sql包时必须注入（至少）一个数据库驱动。参见http://golang.org/s/sqldrivers 获取驱动列表。如使用mysql，则需要注入github.com/go-sql-driver/mysql。

import (
    "fmt"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

//定义一个接收数据库表字段的结构体
type DbPerson struct {
    Id   int    `db:"id"`
    Name string `db:"name"`
    Age  int    `db:"age"`
    Sex  int    `db:"gender"`
}

//定义一个通用的错误处理函数
func ErrorHandler(err error, where string) {
    if err != nil {
        fmt.Println("出现错误：", err, where)
        os.Exit(1)
    }
}

func BaseDatabaseSqlCrud() {
    //1.数据库连接
    db, err := sql.Open("mysql", "username:password@tcp(localhost:3306)/go_mysql_demo?charset=utf8")
    ErrorHandler(err, "sql.Open()")
    defer func() {
        err := db.Close()
        ErrorHandler(err, "db.Close()")
    }()

    //1.增操作
    insertSql := `insert into users (name,age,gender) value (?,?,?) (?,?,?) (?,?,?)`
    stmt, err := db.Prepare(insertSql)
    //准备sql语句，预执行语句，返回*Stmt声明句柄，stmt的主要方法:Exec、Query、QueryRow、Close
    ErrorHandler(err, "db.Prepare()")
    //执行
    res, err := stmt.Exec("fun", 11, 1, "john", 12, 1, "amy", 14, 0)
    ErrorHandler(err, "stmt.Exec()")
    //获取最后一个插入执行结果
    lastInsertId, err := res.LastInsertId()
    ErrorHandler(err, "res.LastInsertId()")

    //log.Println(reflect.TypeOf(lastInsertId))    //打印变量类型
    //将int64转换为字符串
    lastid := strconv.FormatInt(lastInsertId, 10)
    log.Println("lastInsertId = " + lastid)

    //2.改操作
    updateSql := `update users set age=? where name=?`
    stmt, err = db.Prepare(updateSql)
    ErrorHandler(err, "db.Prepare()")
    //执行
    res, err = stmt.Exec(18, "fun")
    ErrorHandler(err, "stmt.Exec()")
    affectCount, _ := res.RowsAffected()
    log.Printf("%v", affectCount)

    //3.删操作
    deleteSql := "delete from users where name=?"
    stmt, err = db.Prepare(deleteSql)
    ErrorHandler(err, "db.Prepare()")
    res, err = stmt.Exec("amy")
    ErrorHandler(err, "stmt.Exec()")

    //4.查询操作
    //查询一条记录，必须使用一个接收变量
    var user *DbPerson
    findSql := "select * from users where id = ?"
    //查询一条，返回一条结果。并赋值到user这个结构体类型的变量中,就算查询到的是多条，单返回的还是一条
    err = db.QueryRow(findSql, 11).Scan(&user.Id, &user.Name, &user.Sex)
    ErrorHandler(err, "db.QueryRow")
    log.Println(user)

    //查询多条记录
    selectSql := "select * from users"
    selectRows, err := db.Query(selectSql)
    ErrorHandler(err, "db.Query")
    defer func() {
        err = selectRows.Close()
        ErrorHandler(err, "selectRows.Close")
    }()

    for selectRows.Next() {
        var person DbPerson
        if err := selectRows.Scan(&person); err != nil {
            ErrorHandler(err, "selectRows.Scan")
            return
        }
        log.Printf("person:%v", person)
    }

    //5.事务操作
    //声明一个事务的开始
    tx, err := db.Begin()
    ErrorHandler(err, "db.Begin")

    //开始事务操作
    _, err1 := tx.Exec("insert into person(name,age,gender) values (?,?,?)", "fun5", 30, 1)
    _, err2 := tx.Exec("update person1 set age=? where name=?", 31, "fun1")
    _, err3 := tx.Exec("insert into person(name,age,gender) values (?,?,?)", "fun6", 30, 2)

    if err1 != nil || err2 != nil || err3 != nil {
        fmt.Println("事务操作出错，开始回滚")
        tx.Rollback()
    } else {
        fmt.Println("事务操作成功！")
        tx.Commit()
    }

}

2.jmoiron/sqlx"第三方包

sqlx为对database/sql的封装，提高对sql支持的易用性。其使用方式和database/sql相差不大。

import (
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "github.com/jmoiron/sqlx"
    "os"
    "time"
)

//定义一个接收数据库表字段的结构体
type DbPerson struct {
    Id   int    `db:"id"`
    Name string `db:"name"`
    Age  int    `db:"age"`
    Sex  int    `db:"gender"`
}

//定义一个通用的错误处理函数
func ErrorHandler(err error, where string) {
    if err != nil {
        fmt.Println("出现错误：", err, where)
        os.Exit(1)
    }
}

//go连接mysql crud
func BaseMysqlCrud() {
    db, err := sqlx.Open("mysql", "godemouser:123456@tcp(localhost:3306)/godemodb")
    ErrorHandler(err, "sqlx.Open")
    defer db.Close()

    //执行sql操作
    //新增
    addResult, err := db.Exec("insert into person(name,age,gender) values (?,?,?)", "dun", 30, 2)
    ErrorHandler(err,"mysql insert")
    insertId, _ := addResult.LastInsertId()
    fmt.Println("插入ID：",insertId)

    //更新
    updateResult, err := db.Exec("update person set age=? where name=?", 31, "fun")
    ErrorHandler(err,"mysql update")
    rowsAffected, _ := updateResult.RowsAffected()
    fmt.Println("更新结果：",rowsAffected)

    //删除
    delResult, err := db.Exec("delete from person where name=?", "kkkk")
    ErrorHandler(err,"Mysql Delete")
    rowsAffected, _ := delResult.RowsAffected()
    fmt.Println("删除结果：",rowsAffected)

    //查询
    var ps []DbPerson
    err = db.Select(&ps, "select * from person where name=?", "fun")
    ErrorHandler(err, "Mysql Query")
    fmt.Println(ps)

}

sqlx对事务的操作非常简单：

//Mysql事务操作
func BaseMysqlTransaction() {
    db, err := sqlx.Open("mysql", "godemouser:123456@tcp(localhost:3306)/godemodb")
    ErrorHandler(err, "sqlx.Open")
    defer db.Close()

    //开启事务
    tx, err := db.Begin()
    ErrorHandler(err, "db.Begin")

    //开始事务操作
    _, err1 := tx.Exec("insert into person(name,age,gender) values (?,?,?)", "fun5", 30, 1)
    _, err2 := tx.Exec("update person1 set age=? where name=?", 31, "fun1")
    _, err3 := tx.Exec("insert into person(name,age,gender) values (?,?,?)", "fun6", 30, 2)

    if err1 != nil || err2 != nil || err3 != nil {
        fmt.Println("事务操作出错，开始回滚")
        tx.Rollback()
    } else {
        fmt.Println("事务操作成功！")
        tx.Commit()
    }

}

sqlx自带连接池管理，在项目中可直接用连接池管理对mysql的访问，以实现对IO的访问控制：

//go mysql 连接池
var db *sqlx.DB
func init() {
    //此设置一般放在init()函数内，然后把连接付给包全局变量
    db, err := sqlx.Open("mysql", "godemouser:123456@tcp(localhost:3306)/godemodb")
    ErrorHandler(err, "sqlx.Open")
    defer db.Close()

    //设置最大连接时间
    db.SetConnMaxLifetime(time.Second * 60)

    //用于设置闲置的连接数。设置闲置的连接数则当开启的一个连接使用完成后可以放在池里等候下一次使用。
    db.SetMaxIdleConns(20)

    //设置最大打开的连接数，设置最大的连接数，可以避免并发太高导致连接mysql出现too many connections的错误。默认值为0表示不限制。
    db.SetMaxOpenConns(100)

}

3.gorm第三方包

ORM（Object Relation Mapping）中文翻译为对象关系映射，这是一种把数据库记录映射为对象类型的一种技术，它把对数据的的操作变成对对象的操作，而不用在意其内部的sql语句，这对许多基于OOP开发的程序员无疑是福音。
go非常热门的orm框架 github.com/jinzhu/gorm ，已经实现了全功能的orm，使用方式请直接参考开发文档，该文档已提供中文支持：https://gorm.io/zh_CN/docs/index.html。 下面我们来简单使用一下其基本功能。

安装库：

go get -u github.com/jinzhu/gorm

导入库及设置连接池：

//导入库与mysql驱动
import(
_ "github.com/jinzhu/gorm/dialects/mysql"
"github.com/jinzhu/gorm"
)

//定义一个通用的错误处理函数
func ErrorHandler(err error, where string) {
    if err != nil {
        fmt.Println("出现错误：", err, where)
        os.Exit(1)
    }
}


//连接数据库
var db *gorm.DB
//包初始化时连接
func init() {
    var err error
    db, err = gorm.Open("mysql", "<username>:<password>/<database>?charset=utf8&parseTime=True&loc=Local")
    ErrorHandler(err,"gorm.Open()")
    
    //database/sql 已内置连接池设置
    db.DB().SetMaxIdleConns(20)
    db.DB().SetMaxOpenConns(100)
}

模型定义：

//通过标签设置 mysql 里面的约束
type User struct {
    gorm.Model //嵌入gorm基础模型，可获得模型基础能力
    Name   string `gorm:"type:varchar(12);not null;unique_index"`
    Email  string `gorm:"type:varchar(24);unique_index"`
    Phone  string `gorm:"type:varchar(11);unique_index"`
    Avatar string `gorm:"type:varchar(64)"`
    Title  string `gorm:"type:varchar(20)"`
    Gender int32  `gorm:"type:varchar(2)"`
}

表操作：

//创建表
if !db.HasTable(&User{}) {
    err := db.Set("gorm:table_options", "ENGINE=InnoDB DEFAULT CHARSET=utf8").CreateTable(&User{}).Error
    ErrorHandler(err,"CreateTable()")
}


//1.插入操作
user := &User{
    Name: "fun",
    Email:"xxx@xxx.com",
    Phone:"13813812138",
    Avatar:"link/to/your/avatar.jpg",
    Title:"Gopher",
    Gender:1
    CreatedAt: time.Now(),
}

if err := db.Create(user).Error; err != nil {
    log.Println("Create data fail!")
}

//2.更新操作
// UPDATE users SET name='joker', updated_at='2016-04-20 22:12:52' WHERE id=1;
db.Model(&user).Update("name", "joker")
// UPDATE users SET name='joker',title='Coder', updated_at='2016-04-20 22:12:52' WHERE id=1;
db.Model(&user).Updates(User{Name: "joker", Title: "Coder"})

//3.查询操作
//通过主键查询第一条记录
db.First(&user)
//// SELECT * FROM users ORDER BY id LIMIT 1;
// 随机取一条记录
db.Take(&user)
//// SELECT * FROM users LIMIT 1;
// 通过主键查询最后一条记录
db.Last(&user)
//// SELECT * FROM users ORDER BY id DESC LIMIT 1;
// 拿到所有的记录
db.Find(&users)
//// SELECT * FROM users;
// 查询指定的某条记录(只可在主键为整数型时使用)
db.First(&user, 10)
//// SELECT * FROM users WHERE id = 10;

//条件查询
// 获取单条记录
db.Where("name = ?", "fun").First(&user)
// SELECT * FROM users WHERE name = 'fun' limit 1;
// 获取多条记录
db.Where("name = ?", "fun").Find(&users)
//// SELECT * FROM users WHERE name = 'fun';
// <>
db.Where("name <> ?", "fun").Find(&users)
// IN
db.Where("name IN (?)", []string{"fun", "joker"}).Find(&users)
// LIKE
db.Where("name LIKE ?", "%f%").Find(&users)
// AND
db.Where("name = ? AND age >= ?", "fun", "18").Find(&users)
// Time
db.Where("updated_at > ?", lastWeek).Find(&users)
// BETWEEN
db.Where("created_at BETWEEN ? AND ?", lastWeek, today).Find(&users)


//4.删除操作
// 删除一条
db.Delete(&user)
// DELETE from users where id=1;
//条件删除多条
db.Where("name LIKE ?", "%fun%").Delete(User{})
// DELETE from users where name LIKE "%f%";
//条件删除多条
db.Delete(User{}, "name LIKE ?", "%fun%")
// DELETE from users where name LIKE "%fun%";