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

王中阳Go
1,354 阅读4分钟

版本升级遇到的问题

  • 老版本 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%")

高级查询

  1. 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}
  1. 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}
  1. 迭代：支持通过行进行迭代
rows, err := db.Model(&User{}).Where("name = ?", "xxx").Rows()
defer rows.Close()

for rows.Next() {
  var user User
  // ScanRows 方法用于将一行记录扫描至结构体
  db.ScanRows(rows, &user)

  // 业务逻辑...
}
  1. 查询钩子:在查询完成之后调用
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        // 更新的错误

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

Last But Not Least

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

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

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

欢迎关注我的公众号

avatar
文章
阅读
粉丝