gorm中的更新

保存所有字段

　　Save 会保存所有的字段，即使字段是零值、

db.First(&user, 5)
user.Name = sql.NullString{"王八", true}
user.Age = 9
db.Debug().Save(&user)
// UPDATE `users` SET `name`='王八',`age`=9,`uuid`='c3987500-bf01-412b-aaab-685285e04037',`created_at`='2021-11-17 13:34:30' WHERE `id` = 5

更新单个列

　　当使用 Update 更新单个列时，你需要指定条件，否则会返回 ErrMissingWhereClause 错误，查看 Block Global Updates 获取详情。当使用了 Model 方法，且该对象主键有值，该值会被用于构建条件，例如：

db.Debug().Model(&User{}).Where("age = ?", 12).Update("name", "张无忌")
// UPDATE `users` SET `name`='张无忌' WHERE age = 12

var user User
db.First(&user, "age = ?", 9)
db.Debug().Model(&user).Update("name", "龟")
// UPDATE `users` SET `name`='龟' WHERE `id` = 5  // 由于user对象主键有值，所以就会被用于构建条件

更新多列

　　Updates 方法支持 struct 和 map[string]interface{} 参数。当使用 struct 更新时，默认情况下，GORM 只会更新非零值的字段

// 根据struct更新属性，只会更新非零值字段
var user User
db.First(&user, 7)
db.Debug().Model(&user).Updates(User{Name: sql.NullString{"哈哈2", true}, Age: 0})
// UPDATE `users` SET `name`='哈哈2' WHERE `id` = 7  // 结构体只会更新非零值字段

// 根据map更新属性，零值和非零值字段都会更新
var user User
db.Debug().First(&user, 7).Updates(map[string]interface{}{
   "name": "", "age": 18,
})
// SELECT * FROM `users` WHERE `users`.`id` = 7 ORDER BY `users`.`id` LIMIT 1
// UPDATE `users` SET `age`=18,`name`='' WHERE `users`.`id` = 7 AND `id` = 7 ORDER BY `users`.`id` LIMIT 1

　　注意当通过 struct 更新时，GORM 只会更新非零字段。如果您想确保指定字段被更新，你应该使用 Select 更新选定字段，或使用 map 来完成更新操作

更新选定字段

　　如果您想要在更新时选定、忽略某些字段，您可以使用 Select、Omit

// select+map更新
db.Debug().Model(new(User)).Where("id = ?", 7).Select("name").Updates(map[string]interface{}{
   "uuid": "def214f8-4b39-11ec-a337-fc3fdbb57b40",
   "created_at": time.Now(),
   "name": "哈哈哈",
})
// UPDATE `users` SET `name`='哈哈哈' WHERE id = 7

// Omit + map更新
db.Debug().Model(new(User)).Where("id = ?", 7).Omit("name").Updates(map[string]interface{}{
   "uuid": "def214f8-4b39-11ec-a337-fc3fdbb57b40",
   "created_at": time.Now(),
   "name": "哈哈哈",
})
// UPDATE `users` SET `created_at`='2021-11-22 10:31:52.602',`uuid`='def214f8-4b39-11ec-a337-fc3fdbb57b40' WHERE id = 7

// select + struct更新, 会更新零值的字段
db.Debug().Model(&User{}).Where(map[string]interface{}{"id": 7}).Select("name", "age").Updates(User{
   Name: sql.NullString{"struct2", true},
   Age: 0,
})
// UPDATE `users` SET `name`='struct2',`age`=0 WHERE `id` = 7

// Omit + struct更新, 不会更新零值的字段
db.Debug().Model(&User{}).Where(map[string]interface{}{"id": 7}).Omit("name").Updates(User{
   Name: sql.NullString{"struct2", true},
   Age: 0,
})
// 此语句不执行

// select 所有字段，（更新包括零值字段的所有字段）
db.Debug().Model(new(User)).Where("id = ?", 7).Select("*").Updates(User{
   Name: sql.NullString{"", true},
   Age: 34,
   CreatedAt: time.Now(),
})
// UPDATE `users` SET `id`=0,`name`='',`age`=34,`uuid`='00000000-0000-0000-0000-000000000000',`created_at`='2021-11-22 10:45:21.161' WHERE id = 7

// Omit 更新除name字段的所有字段，（包括零值字段的所有字段）
db.Debug().Model(new(User)).Where("id = ?", 0).Omit("name").Updates(User{
   Name: sql.NullString{"", true},
   Age: 44,
   CreatedAt: time.Now(),
   ID: 7,
})
// UPDATE `users` SET `id`=7,`age`=44,`created_at`='2021-11-22 10:51:23.298' WHERE id = 0

更新 Hook

　　对于更新操作，GORM 支持 BeforeSave、BeforeUpdate、AfterSave、AfterUpdate 钩子，这些方法将在更新记录时被调用

func (u *User) BeforeUpdate(*gorm.DB) (err error) {
	fmt.Println("User BeforeUpdate执行了")
	return
}
func (u *User) AfterUpdate(*gorm.DB) (err error) {
	fmt.Println("User AfterUpdate执行了")
	return
}
func (u *User) BeforeSave(*gorm.DB) (err error) {
	fmt.Println("User BeforeSave执行了")
	return
}
func (u *User) AfterSave(*gorm.DB) (err error) {
	fmt.Println("User AfterSave执行了")
	return
}

// 更新Hook BeforeSave、AfterSave、BeforeUpdate、AfterUpdate
// 钩子方法的函数签名是：func(*gorm.DB) error
spew.Println("更新前")
db.Debug().Model(new(User)).Where("id = ?", 7).Updates(User{
   Name: sql.NullString{"hello", true},
   Age: 36,
})
spew.Println("更新后")

/* 打印结果
　　更新前 
　　User BeforeSave执行了
　　User BeforeUpdate执行了
　　User AfterSave执行了
　　User AfterUpdate执行了
　　2021/11/22 11:26:26 C:/Users/mayanan/Desktop/pro_go/topgoer_gorm/main.go:114 SLOW SQL >= 200ms
　　[260.657ms] [rows:1] UPDATE `users` SET `name`='hello',`age`=36 WHERE id = 7
　　更新后
*/

批量更新

　　如果您尚未通过 Model 指定记录的主键，则 GORM 会执行批量更新

// 根据struct批量更新
db.Debug().Model(new(User)).Where("name <> ?", "").Updates(User{
   Name: sql.NullString{"mayanan", true},
   Age: 28,
})
//  UPDATE `users` SET `name`='mayanan',`age`=28 WHERE name <> ''

// 根据map批量更新
db.Debug().Model(new(User)).Where("name <> ?", "").Updates(map[string]interface{}{
   "name": "王五", "age": 39,
})
// UPDATE `users` SET `age`=39,`name`='王五' WHERE name <> ''

阻止全局更新

如果在没有任何条件的情况下执行批量更新，默认情况下，GORM 不会执行该操作，并返回 ErrMissingWhereClause 错误

对此，你必须加一些条件，或者使用原生 SQL，或者启用 AllowGlobalUpdate 模式，例如：

// gorm 默认阻止全局更新
db.Debug().Model(&User{}).Updates(map[string]interface{}{
   "name": "lisi", "age": 33,
})
// 报错：WHERE conditions required

db.Debug().Model(&User{}).Where("1 = 1").Updates(map[string]interface{}{
   "name": "lisi", "age": 33,
})
// UPDATE `users` SET `age`=33,`name`='lisi' WHERE 1 = 1

db.Debug().Exec("update users set name = ?", "田七")
// update users set name = '田七';

db.Debug().Session(&gorm.Session{AllowGlobalUpdate: true}).Updates(User{
   Age: 35,
})
// UPDATE `users` SET `age`=33

更新的记录数

　　获取受更新影响的行数

// 获取受更新影响的行数
ret := db.Debug().Session(&gorm.Session{AllowGlobalUpdate: true}).Updates(User{
   Age: 37,
})
spew.Dump(ret.RowsAffected)  // 5  更新的记录数
spew.Dump(ret.Error)  // nil 更新产生的错误

高级选项

使用 SQL 表达式更新

　　GORM 允许使用 SQL 表达式更新列，例如：

var user User
db.First(&user)
db.Debug().Model(&user).Update("age", gorm.Expr("age * ? + ?", 2, 3))
// UPDATE `users` SET `age`=age * 2 + 3 WHERE `id` = 1

// map里面使用sql表达式
var user []User
db.Find(&user)
db.Debug().Model(&user).Updates(map[string]interface{}{
   "age": gorm.Expr("(age - ?) / ?", 2, 2),
})
// UPDATE `users` SET `age`=(age - 2) / 2 WHERE (`id` = 1 OR `id` = 2 OR `id` = 5 OR `id` = 6 OR `id` = 7)

// UpdateColumn里面使用sql表达式
var user []User
db.Find(&user)
db.Debug().Model(&user).UpdateColumn("age", gorm.Expr("age / ? - ?", 2, 2))
// UPDATE `users` SET `age`=age / 2 - 2 WHERE (`id` = 1 OR `id` = 2 OR `id` = 5 OR `id` = 6 OR `id` = 7)

根据子查询进行更新

　　使用子查询更新表

// 根据子查询更新表
var user User
db.First(&user, 6)
subQuery := db.Model(new(Email)).Select("user_id").Where("email = ?", "abc@qq.com")
db.Debug().Model(&user).Update("age", subQuery)
// UPDATE `users` SET `age`=(SELECT `user_id` FROM `emails` WHERE email = 'abc@qq.com') WHERE `id` = 6

subQuery := db.Table("emails as e").Select("email").Where("user_id = ?", 10)
db.Debug().Table("users as u").Where("id = ?", 7).Update("name", subQuery)
// UPDATE users as u SET `name`=(SELECT email FROM emails as e WHERE e.user_id = 11) WHERE id = 5

subQuery := db.Table("emails as e").Select("email").Where("user_id = ?", 2)
db.Debug().Table("users as u").Where("id = ?", 2).Updates(map[string]interface{}{
   "name": subQuery,
})
// UPDATE users as u SET `name`=(SELECT email FROM emails as e WHERE user_id = 2) WHERE id = 2

不使用 Hook 和时间追踪

如果您想在更新时跳过 Hook 方法且不追踪更新时间，可以使用 UpdateColumn、UpdateColumns，其用法类似于 Update、Updates

// 更新单个列
db.Debug().Table("users").Where("id = ?", 1).UpdateColumn("name", "李四")
// UPDATE `users` SET `name`='李四' WHERE id = 1

// 更新多个列
db.Debug().Table("users").Where("id = ?", 2).UpdateColumns(map[string]interface{}{
   "name": "王五", "age": 31,
})
// UPDATE `users` SET `age`=31,`name`='王五' WHERE id = 2

db.Debug().Table("users").Where("id = ?", 2).Select("name").UpdateColumns(User{
   Name: sql.NullString{"王十六", true}, Age: 36,
})
// UPDATE `users` SET `name`='王十六' WHERE id = 2

posted @ 2021-11-19 18:09 专职阅读(1843) 评论(0) 编辑收藏举报

刷新页面返回顶部

专职