gorm update传入struct对象,零值字段不更新的解决方案

 更新时间:2021年04月26日 09:10:37   作者:qauzy  
这篇文章主要介绍了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

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。

相关文章

  • Go语言格式化动词使用详解

    Go语言格式化动词使用详解

    这篇文章主要介绍了Go语言格式化动词使用详解的相关资料,需要的朋友可以参考下
    2023-08-08
  • Go Excelize API源码阅读GetPageLayout及SetPageMargins

    Go Excelize API源码阅读GetPageLayout及SetPageMargins

    这篇文章主要为大家介绍了Go Excelize API源码阅读GetPageLayout及SetPageMargins的方法示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-08-08
  • GoLang中的timer定时器实现原理分析

    GoLang中的timer定时器实现原理分析

    Timer中对外暴露的只有一个channel,这个 channel也是定时器的核心。当计时结束时,Timer会发送值到channel中,外部环境在这个 channel 收到值的时候,就代表计时器超时了,可与select搭配执行一些超时逻辑
    2023-02-02
  • Go语言中基本数据类型的相互转换详解

    Go语言中基本数据类型的相互转换详解

    Go在不同类型的变量之间赋值时需要显示转换,不能自动转换。这篇文章主要和大家介绍了Go语言中基本数据类型的相互转换,感兴趣的小伙伴可以了解一下
    2022-10-10
  • Go秒爬博客园100页新闻

    Go秒爬博客园100页新闻

    利用go语言的协程并发优势爬取网页速度相当之快,博客园100页新闻标题只需一秒即可全部爬取,跟着小编一起去看看如何实现的,希望大家可以从中受益
    2018-09-09
  • Goland项目使用gomod配置的详细步骤

    Goland项目使用gomod配置的详细步骤

    Goland是一个用于Go语言开发的IDE,Goland的项目结构与Go语言的项目结构相似,下面这篇文章主要给大家介绍了关于Goland项目使用gomod配置的详细步骤,文中通过图文介绍的非常详细,需要的朋友可以参考下
    2023-04-04
  • Go垃圾回收提升内存管理效率优化最佳实践

    Go垃圾回收提升内存管理效率优化最佳实践

    这篇文章主要为大家介绍了Go垃圾回收提升内存管理效率优化最佳实践,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-12-12
  • golang gorm框架数据库的连接操作示例

    golang gorm框架数据库的连接操作示例

    这篇文章主要为大家介绍了golang gorm框架数据库操作示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步早日升职加薪
    2022-04-04
  • Go语言接口的嵌套的具体使用

    Go语言接口的嵌套的具体使用

    在Go语言中,不仅结构体与结构体之间可以嵌套,接口与接口间也可以通过嵌套创造出新的接口,本文主要介绍了Go语言接口的嵌套的具体使用,感兴趣的可以了解一下
    2023-04-04
  • 一文搞懂Golang 时间和日期相关函数

    一文搞懂Golang 时间和日期相关函数

    这篇文章主要介绍了Golang 时间和日期相关函数,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-12-12

最新评论