【gorm】updateはstructオブジェクトに転送され、ゼロ値フィールドは問題を更新しません

4451 ワード

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文にはCode 2更新文は含まれません.
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
}