使用GORM操作数据库
依赖安装
github.com/jinzhu/gorm
定义实体类
注意:
实体类结构体中,要映射到数据库的字段首字母必须大写,否则会被忽略
可以通过定义嵌套gorm.Model这个结构体的类型来定义实体类,gorm.Model中定义了数据库表的一些常用基本字段
type Model struct {
ID uint `gorm:"primary_key"`
CreatedAt time.Time
UpdatedAt time.Time
DeletedAt *time.Time
}
比如我们定义User实体
type User struct{
gorm.Model
Name string
}
开启连接
开启连接参数字符串格式如下
username:password@protocol(address)/dbname?param=value
其中如果protocol和address是默认值,可以直接省略,比如
import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
)
func main(){
db, err := gorm.Open("mysql", "root:root@/gorm?charset=utf8&parseTime=True&loc=Local")
defer db.Close()
}
创建表
创建表默认使用结构体类型名称的驼峰命名复数形式作为表名,比如User就是users
通过
db.SingularTable(true)
设定不使用复数形式,则User对应的表为user
CreateTable方法接收一个interface{}类型,创建对应结构体类型的表。如果创建的表已经存在,会抛出异常
比如
import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/mysql"
)
type User struct {
gorm.Model
Name string
}
func main() {
db, _ := gorm.Open("mysql", "root:root@/gorm?charset=utf8&parseTime=True&loc=Local")
defer db.Close()
db.CreateTable(&User{}) //创建表名users
}
更新表
AutoMigrate方法与CreateTable用法类似,不同的是,如果表已经存在,AutoMigrate也不会抛出异常,而是使用当前传入的最新结构体更新表结构,如果表不存在,则与CreateTable表现一样,所以刚刚的
db.CreateTable(&User{})
也可以使用下面语句替代
db.AutoMigrate(&User{})
插入/更新记录
gorm.Model的主键ID为整数类型,映射到数据库时为自动递增主键
Save方法接收一个结构体指针对象
- 如果该对象的主键没有设定,或者是默认值0,则作为插入操作,由数据库策略生成主键(比如自增)插入记录
- 如果该对象设定了主键,数据库中不存在该主键记录,则作为插入操作,使用该主键插入记录
- 如果该对象设定了主键,数据库中存在该主键记录,则作为更新操作,更新数据库记录
db, _ := gorm.Open("mysql", "root:root@/gorm?charset=utf8&parseTime=True&loc=Local")
defer db.Close()
db.AutoMigrate(&User{})
user := User{Name: "aa"}
db.Save(&user)
//新增成功后,如果主键是由数据库生成,会将主键回显到实体对象的属性
println(user.ID)
user.Name = "bb"
db.Save(&user)//更新
插入记录
Create方法用法与Save类似,不同的是Create方法只能用于插入,如果对象具备主键,并且数据库中已经存在该主键记录,则抛出异常
db, _ := gorm.Open("mysql", "root:root@/gorm?charset=utf8&parseTime=True&loc=Local")
defer db.Close()
db.AutoMigrate(&User{})
user := User{Name: "aa"}
db.Create(&user) //没有设定主键,默认由数据库自增
println(user.ID)
NewRecord方法用于判断某个对象是否可以作为新纪录插入,如果该对象主键为空或者0,或者数据库表中不存在该主键记录,返回true,否则返回false,所以可以用于辅助Create方法
...
if(db.NewRecord(&user)){
db.Create(&user)
}
修改记录
Update和Updates方法提供对记录进行更新操作,可以通过Map或者struct传递更新属性,建议通过Map
因为通过struct更新时,FORM将仅更新具有非空值的字段
// 使用`map`更新多个属性,只会更新这些更改的字段
db.Model(&user).Updates(map[string]interface{}{"name": "hello", "age": 18, "actived": false})
// 使用组合条件批量更新单个属性
db.Model(&user).Where("name= ?", "aa").Update("name", "hel