MYSQL驱动,通过比较原生的写法去写sql和处理事务。目前开源界也有很多封装好的orm操作框架,帮我们简省一些重复的操作,提高代码可读性。gorm
就是这样的一款作品,我们来学习一下gorm的操作流程。
安装
go get -u github.com/jinzhu/gorm
数据库连接
要连接到数据库首先要导入驱动程序。例如
import _ "github.com/go-sql-driver/mysql"
为了方便记住导入路径,GORM包装了一些驱动。
import _ "github.com/jinzhu/gorm/dialects/mysql"
// import _ "github.com/jinzhu/gorm/dialects/postgres"
// import _ "github.com/jinzhu/gorm/dialects/sqlite"
// import _ "github.com/jinzhu/gorm/dialects/mssql"
所以包名可以改为如上:
import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
)
func main() {
db, err := gorm.Open("mysql", "user:password@tcp(IP:port)/dbname?charset=utf8&parseTime=True&loc=Local")
db.DB().SetMaxIdleConns(10)
db.DB().SetMaxOpenConns(100)
defer db.Close()
}
注:为了处理time.Time
,你需要包括parseTime
作为参数。
数据模型定义
表名,列名如何对应结构体
在Gorm中,表名是结构体名的复数形式,列名是字段名的蛇形小写。
即,如果有一个user表,那么如果你定义的结构体名为:User,gorm会默认表名为users而不是user。
例如有如下表结构定义:
CREATE TABLE `areas` (
`id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主键id',
`area_id` int(11) NOT NULL COMMENT '区县id',
`area_name` varchar(45) NOT NULL COMMENT '区县名',
`city_id` int(11) NOT NULL COMMENT '城市id',
`city_name` varchar(45) NOT NULL COMMENT '城市名称',
`province_id` int(11) NOT NULL COMMENT '省份id',
`province_name` varchar(45) NOT NULL COMMENT '省份名称',
`area_status` tinyint(3) NOT NULL DEFAULT '1' COMMENT '该条区域信息是否可用 : 1:可用 2:不可用',
`created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',
`updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8mb4 COMMENT='区域表'
那么对应的结构体定义如下:
type Area struct {
Id int
AreaId int
AreaName string
CityId int
CityName string
ProvinceId int
ProvinceName string
AreaStatus int
CreatedAt time.Time
UpdatedAt time.Time
}
如何全局禁用表名复数呢?
可以在创建数据库连接的时候设置如下参数:
db.SingularTable(true)
这样的话,表名默认即为结构体的首字母小写形式。
CRUD 使用
下面我们使用一张User表来就CRUD做一些操作示例:
表结构如下:
CREATE TABLE `user` (
`id` bigint(20) NOT NULL AUTO_INCREMENT,
`name` varchar(30) NOT NULL DEFAULT '',
`age` int(3) NOT NULL DEFAULT '0',
`sex` tinyint(3) NOT NULL DEFAULT '0',
`phone` varchar(40) NOT NULL DEFAULT '',
`create_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
`updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=8 DEFAULT CHARSET=utf8mb4
首先初始化数据库连接:
package main
import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
)
var db *gorm.DB
type User struct {
Id int
Name string
Age int
Sex byte
Phone string
}
func init() {
var err error
db, err = gorm.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local")
if err != nil {
panic(err)
}
db.SingularTable(true)
}
下面所有的操作都是在上面的初始化连接上执行的操作。
插入
func (user *User) Insert() {
db.Table("user").Create(user)
}
更新
user := User{Id: 1,Name:"xiaoming"}
db.Model(&user).Update(user)
db.Model(&User{}).Where("sex = ?",1).Update("name","xiaohong")
如果你想手动将某个字段set为空值, 可以使用单独选定某些字段的方式来更新:
user := User{Id: 1}
db.Model(&user).Select("name").Update(map[string]interface{}{"name":"","age":0})
忽略掉某些字段:
当你的更新的参数为结构体,而结构体中某些字段你又不想去更新,那么可以使用Omit方法过滤掉这些不想update到库的字段:
user := User{Id: 1,Name:"xioaming",Age:12}
db.Model(&user).Omit("name").Update(&user)
删除
user := User{Id: 1}
db.Delete(&user)
db.Delete(&User{},"id > ?",11)
事务
func CreateAnimals(db *gorm.DB) err {
tx := db.Begin()
if err := tx.Create(&Animal{Name: "Giraffe"}).Error; err != nil {
tx.Rollback()
return err
}
if err := tx.Create(&Animal{Name: "Lion"}).Error; err != nil {
tx.Rollback()
return err
}
tx.Commit()
return nil
}
查询:
func (user *User) query() (u []User) {
db.Find(&u)
db.Find(&u,"id > ? and age > ?",2,12)
db.Where("id > ?", 2).Find(&u)
db.Where("name in (?)",[]string{"xiaoming","xiaohong"}).Find(&u)
db.First(&u)
db.First(&u,"where sex = ?",1)
db.Last(&u)
return u
}
注意:方法中带的&u
表示是返回值用u这个对象来接收。
上面的查询都将返回表中所有的字段,如果你想指定查询某些字段该怎么做呢?
指定查询字段-Select
db.Select("name,age").Where(map[string]interface{}{"age":12,"sex":1}).Find(&u)
使用Struct和map作为查询条件
db.Where(&User{Age:12,Sex:1}).Find(&u)
db.Where(map[string]interface{}{"age":12,"sex":1}).Find(&u)
not 条件的使用
db.Not("name","xiaoming","xiaohong").Find(&u)
db.Not("name",[]string{"xiaoming","xiaohong"}).Find(&u)
or 的使用
db.Where("age > ?",12).Or("sex = ?",1).Find(&u)
order by 的使用
db.Where("age > ?",12).Or("sex = ?",1).Order("age desc").Find(&u)
limit 的使用
db.Not("name",[]string{"xiaoming","xiaohong"}).Limit(10).Find(&u)
offset 的使用
db.Not("name",[]string{"xiaoming","xiaohong"}).Limit(10).Offset(300).Find(&u)
count(*)
var count int
db.Table("user").Where("age > ?",0).Count(&count)
注意:这里你在指定表名的情况下sql为:select count(*) from user where age > 0;
如上代码如果改为:
var count int
var user []User
db.Where("age > ?",0).Find(&user).Count(&count)
相当于你先查出来[]User,然后统计这个list的长度。跟你预期的sql不相符。
group & having
rows, _ := db.Table("user").Select("count(*),sex").Group("sex").
Having("age > ?", 10).Rows()
for rows.Next() {
fmt.Print(rows.Columns())
}
join
db.Table("user u").Select("u.name,u.age").Joins("left join user_ext ue on u.user_id = ue.user_id").Row()
如果有多个连接,用多个Join方法即可。
原生函数
db.Exec("DROP TABLE user;")
db.Exec("UPDATE user SET name=? WHERE id IN (?)", "xiaoming", []int{11,22,33})
db.Exec("select * from user where id > ?",10).Scan(&user)
一些函数
FirstOrInit 和 FirstOrCreate
获取第一个匹配的记录,若没有,则根据条件初始化一个新的记录:
//注意:where条件只能使用Struct或者map。如果这条记录不存在,那么会新增一条name=xiaoming的记录
db.FirstOrInit(&u,User{Name:"xiaoming"})
//同上
db.FirstOrCreate(&u,User{Name:"xiaoming"})
Attrs
如果没有找到记录,则使用Attrs中的数据来初始化一条记录:
db.Where(User{Name:"xiaoming"}).Attrs(User{Name:"xiaoming",Age:12}).FirstOrInit(&u)
Assign
db.Where("age > 12").Assign(User{Name:"xiaoming"}).FirstOrInit(&u)
Pluck
如果user表中你只想查询age这一列,该怎么返回呢,gorm提供了Pluck函数用于查询单列,返回数组:
var ages []int
db.Find(&u).Pluck("age",&ages)
Scan
Scan函数可以将结果转存储到另一个结构体中。
type SubUser struct{
Name string
Age int
}
db.Table("user").Select("name,age").Scan(&SubUser)
sql.Row & sql.Rows
row和rows用户获取查询结果。
row := db.Table("user").Where("name = ?", "xiaoming").Select("name, age").Row()
row.Scan(&name, &age)
rows, err := db.Model(&User{}).Where("sex = ?",1).Select("name, age, phone").Rows()
defer rows.Close()
for rows.Next() {
...
rows.Scan(&name, &age, &email)
...
}
日志
Gorm有内置的日志记录器支持,默认情况下,它会打印发生的错误。
db.LogMode(true)
db.LogMode(false)
db.Debug().Where("name = ?", "xiaoming").First(&User{})
请发表评论