gorm update传入struct对象,零值字段不更新的解决方案
来源:脚本之家
时间:2023-01-07 11:57:24 341浏览 收藏
本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的,希望这篇《gorm update传入struct对象,零值字段不更新的解决方案》对你有很大帮助!欢迎收藏,分享给更多的需要的朋友学习~
使用gorm的update接口,出现如果字段为零值
则不会生成字段的更新语句
// Update update attributes with callbacks, refer: https://jinzhu.github.io/gorm/crud.html#update // WARNING when update with struct, GORM will not update fields that with zero value func (s *DB) Update(attrs ...interface{}) *DB { return s.Updates(toSearchableMap(attrs...), true) }
如:
type AA struct { ID int `gorm:"primary_key" json:" - "` //主键id Code1 string Code2 string }
如果
a :=AA{ ID:1, Code1:1, Code2:0 }
则产生的SQL语句将不包含Code2更新语句,
update aa set Code1 =1 where id =1;
出现该问题的原因在于
使用stuct类型对象作为参数时,struct会首先转化为map对象,然后再生成SQL语句,但是转化为map的过程中,对于零值字段是忽略的
// assignUpdatingAttributesCallback assign updating attributes to model func assignUpdatingAttributesCallback(scope *Scope) { if attrs, ok := scope.InstanceGet("gorm:update_interface"); ok { if updateMaps, hasUpdate := scope.updatedAttrsWithValues(attrs); hasUpdate { scope.InstanceSet("gorm:update_attrs", updateMaps) } else { scope.SkipLeft() } } } func (scope *Scope) updatedAttrsWithValues(value interface{}) (results map[string]interface{}, hasUpdate bool) { if scope.IndirectValue().Kind() != reflect.Struct { return convertInterfaceToMap(value, false, scope.db), true } results = map[string]interface{}{} for key, value := range convertInterfaceToMap(value, true, scope.db) { if field, ok := scope.FieldByName(key); ok && scope.changeableField(field) { if _, ok := value.(*SqlExpr); ok { hasUpdate = true results[field.DBName] = value } else { err := field.Set(value) if field.IsNormal && !field.IsIgnored { hasUpdate = true if err == ErrUnaddressable { results[field.DBName] = value } else { results[field.DBName] = field.Field.Interface() } } } } } return } func convertInterfaceToMap(values interface{}, withIgnoredField bool, db *DB) map[string]interface{} { var attrs = map[string]interface{}{} switch value := values.(type) { case map[string]interface{}: return value case []interface{}: for _, v := range value { for key, value := range convertInterfaceToMap(v, withIgnoredField, db) { attrs[key] = value } } case interface{}: reflectValue := reflect.ValueOf(values) switch reflectValue.Kind() { case reflect.Map: for _, key := range reflectValue.MapKeys() { attrs[ToColumnName(key.Interface().(string))] = reflectValue.MapIndex(key).Interface() } default: for _, field := range (&Scope{Value: values, db: db}).Fields() { if !field.IsBlank && (withIgnoredField || !field.IsIgnored) { //只有非零值才更新 attrs[field.DBName] = field.Field.Interface() } } } } return attrs }
为了更新零值字段,则需要修改gorm库,我们这里添加一个FORCE标识字段必须更新
type AA struct { ID int `gorm:"primary_key" json:" - "` //主键id Code1 string `gorm:"force"` Code2 string `gorm:"force"` }
修改Sope的Fields函数,对于有FORCE标签的字段,IsBlank直接设置为fasle
// Fields get value's fields //通过反射获取field的值 func (scope *Scope) Fields() []*Field { if scope.fields == nil { var ( fields []*Field indirectScopeValue = scope.IndirectValue() isStruct = indirectScopeValue.Kind() == reflect.Struct ) // 解析结构体 for _, structField := range scope.GetModelStruct().StructFields { if isStruct { fieldValue := indirectScopeValue // 一般只有一个字段名吧????? for _, name := range structField.Names { if fieldValue.Kind() == reflect.Ptr && fieldValue.IsNil() { // 处理数组 fieldValue.Set(reflect.New(fieldValue.Type().Elem())) } // fieldValue = reflect.Indirect(fieldValue).FieldByName(name) } //有force代表强制更新 _, ok := structField.TagSettingsGet("FORCE") fields = append(fields, &Field{StructField: structField, Field: fieldValue, IsBlank: isBlank(fieldValue) && !ok}) //如果不是struct,则返回空,因为原生没有字段这个说法 } else { fields = append(fields, &Field{StructField: structField, IsBlank: true}) } } scope.fields = &fields } return *scope.fields }
补充:Gorm 自动update操作自动过滤0和 “ “
gorm会自动过滤结构体的0 和" "
type CrmUserTableColumns struct { Id int64 `gorm:"column:ids"` Name string `gorm:"column:xxx"` Account string `gorm:"column:xxx"` Password string `gorm:"column:xxx"` State int64 `gorm:"column:state"` BusinessId int64 `gorm:"column:business_id"` DepartmentId int64 `gorm:"column:department_id"` ... } func (u *CrmUserTableColumns) TableName() string { return "crm_user" } field := map[string]interface{}{ "business_id": data.BusinessId, "state": data.State, "department_id": data.DepartmentId, } whereSql = "account = ?" updates := dbMasterClient. //Debug(). Model(&CrmUserTableColumns{}). Where(whereSql, whereCase). Updates(field) RowsAffected = updates.RowsAffected return
以上为个人经验,希望能给大家一个参考,也希望大家多多支持golang学习网。如有错误或未考虑完全的地方,望不吝赐教。
以上就是《gorm update传入struct对象,零值字段不更新的解决方案》的详细内容,更多关于golang的资料请关注golang学习网公众号!
声明:本文转载于:脚本之家 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
-
406 收藏
-
125 收藏
-
206 收藏
-
377 收藏
-
193 收藏
最新阅读
更多>
-
108 收藏
-
367 收藏
-
419 收藏
-
234 收藏
-
155 收藏
-
457 收藏
课程推荐
更多>
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 立即学习 542次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 立即学习 507次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 立即学习 497次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 立即学习 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 立即学习 484次学习