版本升级遇到的问题
- 老版本 Count() 返回int，新版本 Count() 返回int64
- 老版本 Update() 的用法相当于新版本 Save() 的用法
Update
- 老版本传入结构体指针
func LogIp(admin *service.Admin) { DB.Table("admin").Where("id = ?", admin.ID).Update(admin) } 复制代码
- 新版本传入key value 更新
db.Model(&User{}).Where("active = ?", true).Update("name", "hello") 复制代码
- 新版本的 Save() 方法
db.First(&user) user.Name = "wzy" user.Age = 18 db.Save(&user) // UPDATE users SET name='wzy', age=18, birthday='2016-01-01', updated_at = '2020-11-17 21:34:10' WHERE id=111; 复制代码
钩子函数
- 钩子函数（Hook）是在创建、查询、更新、删除等操作之前、之后调用的函数。
- 常用的钩子函数有：
创建
// 开始事务 BeforeSave BeforeCreate // 关联前的 save // 插入记录至 db // 关联后的 save AfterCreate AfterSave // 提交或回滚事务 复制代码
更新
// 开始事务 BeforeSave BeforeUpdate // 关联前的 save // 更新 db // 关联后的 save AfterUpdate AfterSave // 提交或回滚事务 复制代码
删除
// 开始事务 BeforeDelete // 删除 db 中的数据 AfterDelete // 提交或回滚事务 复制代码
查询
// 从 db 中加载数据 // Preloading (eager loading) AfterFind 复制代码
查询
检查 ErrRecordNotFound 错误
// 检查 ErrRecordNotFound 错误 errors.Is(result.Error, gorm.ErrRecordNotFound) 复制代码
检索单个对象
var user User // 可以 db.First(&user) // SELECT * FROM `users` ORDER BY `users`.`id` LIMIT 1 // 可以 result := map[string]interface{}{} db.Model(&User{}).First(&result) // SELECT * FROM `users` ORDER BY `users`.`id` LIMIT 1 // 不行 result := map[string]interface{}{} db.Table("users").First(&result) // 但可以配合 Take 使用 result := map[string]interface{}{} db.Table("users").Take(&result) // 根据第一个字段排序 type Language struct { Code string Name string } db.First(&Language{}) // SELECT * FROM `languages` ORDER BY `languages`.`code` LIMIT 1 复制代码
查询注意问题
- 结构体查询会忽略零值条件，如果需要把零值作为筛选条件需要用map
db.Where(&User{Name: "xiaoming", Age: 0}).Find(&users) // SELECT * FROM users WHERE name = "jinzhu"; db.Where(map[string]interface{}{"Name": "xiaoming", "Age": 0}).Find(&users) // SELECT * FROM users WHERE name = "jinzhu" AND age = 0; 复制代码
执行原生SQL
// 原生 SQL db.Raw("SELECT name, age FROM users WHERE name = ?", "xxx").Scan(&result) 复制代码
子查询
db.Where("amount > (?)", db.Table("orders").Select("AVG(amount)")).Find(&orders) // SELECT * FROM "orders" WHERE amount > (SELECT AVG(amount) FROM "orders"); subQuery := db.Select("AVG(age)").Where("name LIKE ?", "name%").Table("users") db.Select("AVG(age) as avgage").Group("name").Having("AVG(age) > (?)", subQuery).Find(&results) // SELECT AVG(age) as avgage FROM `users` GROUP BY `name` HAVING AVG(age) > (SELECT AVG(age) FROM `users` WHERE name LIKE "name%") 复制代码
高级查询
- FirstOrInit ,获取第一条匹配的记录，或者根据给定的条件初始化一个实例（仅支持 sturct 和 map 条件,不会保存再数据库
// 未找到 user，根据条件和 Assign 属性创建记录 db.Where(User{Name: "non_existing"}).Attrs(User{Age: 20}).FirstOrCreate(&user) // SELECT * FROM users WHERE name = 'non_existing' ORDER BY id LIMIT 1; // INSERT INTO "users" (name, age) VALUES ("non_existing", 20); // user -> User{ID: 112, Name: "non_existing", Age: 20} // 找到了 `name` = `jinzhu` 的 user，则忽略 Attrs db.Where(User{Name: "jinzhu"}).Attrs(User{Age: 20}).FirstOrCreate(&user) // SELECT * FROM users WHERE name = 'jinzhu' ORDER BY id LIMIT 1; // user -> User{ID: 111, Name: "jinzhu", Age: 18} 复制代码
- FirstOrCreate ,获取第一条匹配的记录，或者根据给定的条件创建一条新纪录（仅支持 sturct 和 map 条件）,不管是否找到记录，Assign 都会将属性赋值给 struct，并将结果写回数据库
// 未找到 user，根据条件和 Assign 属性创建记录 db.Where(User{Name: "non_existing"}).Assign(User{Age: 20}).FirstOrCreate(&user) // SELECT * FROM users WHERE name = 'non_existing' ORDER BY id LIMIT 1; // INSERT INTO "users" (name, age) VALUES ("non_existing", 20); // user -> User{ID: 112, Name: "non_existing", Age: 20} // 找到了 `name` = `jinzhu` 的 user，依然会根据 Assign 更新记录 db.Where(User{Name: "jinzhu"}).Assign(User{Age: 20}).FirstOrCreate(&user) // SELECT * FROM users WHERE name = 'jinzhu' ORDER BY id LIMIT 1; // UPDATE users SET age=20 WHERE id = 111; // user -> User{ID: 111, Name: "jinzhu", Age: 20} 复制代码
- 迭代：支持通过行进行迭代
rows, err := db.Model(&User{}).Where("name = ?", "xxx").Rows() defer rows.Close() for rows.Next() { var user User // ScanRows 方法用于将一行记录扫描至结构体 db.ScanRows(rows, &user) // 业务逻辑... } 复制代码
- 查询钩子:在查询完成之后调用
func (u *User) AfterFind(tx *gorm.DB) (err error) { if u.Role == "" { u.Role = "user" } return } 复制代码
更新
使用sql表达式更新
- gorm.Expr()
// product 的 ID 是 `3` db.Model(&product).Update("price", gorm.Expr("price * ? + ?", 2, 100)) // UPDATE "products" SET "price" = price * 2 + 100, "updated_at" = '2013-11-17 21:34:10' WHERE "id" = 3; 复制代码
子查询更新
- Update("字段",子查询语句)
db.Model(&user).Update("company_name", db.Model(&Company{}).Select("name").Where("companies.id = users.company_id")) // UPDATE "users" SET "company_name" = (SELECT name FROM companies WHERE companies.id = users.company_id); db.Table("users as u").Where("name = ?", "jinzhu").Update("company_name", db.Table("companies as c").Select("name").Where("c.id = u.company_id")) db.Table("users as u").Where("name = ?", "jinzhu").Updates(map[string]interface{}{}{"company_name": db.Table("companies as c").Select("name").Where("c.id = u.company_id")}) 复制代码
通用的返回结果
- result.RowsAffected // 更新的记录数
- result.Error // 更新的错误
// 通过 `RowsAffected` 得到更新的记录数 result := db.Model(User{}).Where("role = ?", "admin").Updates(User{Name: "hello", Age: 18}) // UPDATE users SET name='hello', age=18 WHERE role = 'admin; result.RowsAffected // 更新的记录数 result.Error // 更新的错误
