Skip to content

Go GORM是时候升级新版本了 2.0新特性介绍(2)| Go主题月

版本升级遇到的问题

  • 老版本 Count() 返回int,新版本 Count() 返回int64
  • 老版本 Update() 的用法相当于新版本 Save() 的用法

Update

  • 老版本传入结构体指针
go
func LogIp(admin *service.Admin) { 	DB.Table("admin").Where("id = ?", admin.ID).Update(admin) }
  • 新版本传入key value 更新
go
db.Model(&User{}).Where("active = ?", true).Update("name", "hello")
  • 新版本的 Save() 方法
go
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)是在创建、查询、更新、删除等操作之前、之后调用的函数。
  • 常用的钩子函数有:

创建

go
// 开始事务 BeforeSave BeforeCreate // 关联前的 save // 插入记录至 db // 关联后的 save AfterCreate AfterSave // 提交或回滚事务

更新

go
// 开始事务 BeforeSave BeforeUpdate // 关联前的 save // 更新 db // 关联后的 save AfterUpdate AfterSave // 提交或回滚事务

删除

arduino
// 开始事务 BeforeDelete // 删除 db 中的数据 AfterDelete // 提交或回滚事务

查询

go
// 从 db 中加载数据 // Preloading (eager loading) AfterFind

查询

检查 ErrRecordNotFound 错误

scss
// 检查 ErrRecordNotFound 错误 errors.Is(result.Error, gorm.ErrRecordNotFound)

检索单个对象

go

```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

``` go
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

go
// 原生 SQL db.Raw("SELECT name, age FROM users WHERE name = ?", "xxx").Scan(&result)

子查询

go

```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%")``

### 高级查询

1.  FirstOrInit ,获取第一条匹配的记录,或者根据给定的条件初始化一个实例(仅支持 sturct 和 map 条件,不会保存再数据库

``` go

```// 未找到 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}``

2.  FirstOrCreate ,获取第一条匹配的记录,或者根据给定的条件创建一条新纪录(仅支持 sturct 和 map 条件),不管是否找到记录,Assign 都会将属性赋值给 struct,并将结果写回数据库

``` go

```// 未找到 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}``

3.  迭代:支持通过行进行迭代

``` go
rows, err := db.Model(&User{}).Where("name = ?", "xxx").Rows() defer rows.Close() for rows.Next() {   var user User   // ScanRows 方法用于将一行记录扫描至结构体   db.ScanRows(rows, &user)   // 业务逻辑... }
  1. 查询钩子:在查询完成之后调用
go
func (u *User) AfterFind(tx *gorm.DB) (err error) {   if u.Role == "" {     u.Role = "user"   }   return }

更新

使用sql表达式更新

  • gorm.Expr()
go

```// 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("字段",子查询语句)

``` go
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 // 更新的错误
go

```// 通过 `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        // 更新的错误``

> 添加和删除的知识点下一篇整理

## Last But Not Least

真心感谢帅哥美女们能看到这里,如果觉得这篇文章写得还不错😄

求点赞👍 ,求关注❤️ ,求分享👥 ,对想成为海贼王的我来说**非常有用**!!!

如果能关注我公众号的话,那我比彩票中奖还高兴 ❤️❤️❤️❤️

![欢迎关注我的公众号](https://p3-juejin.byteimg.com/tos-cn-i-k3u1fbpfcp/3ae5ee8a4a4b4dd397b23b92d806951f~tplv-k3u1fbpfcp-zoom-in-crop-mark:1512:0:0:0.awebp)

🚀 学习遇到瓶颈?想进大厂?

看完这篇技术文章,如果还是觉得不够系统,或者想在实战中快速提升?
王中阳的就业陪跑训练营,提供定制化学习路线 + 企业级实战项目 + 简历优化 + 模拟面试。

了解训练营详情