问题描述:
最近在使用gorm时,发现db文件在某些情况下会多出来几行重复的数据。
type Sip struct {
SipPoolID uint
Sip string `json:"-" gorm:"primary_key"`
NetSegment string `json:"-" gorm:"primary_key"`
}
使用以上结构体,通过gorm创建出来的表会出现如下情况:
诊断:
经过一番操作和观察,发现在某行数据主键为空值的时候,这行数据会重复出现。然后去看源码,发现了这么一段代码:
// Save update value in database, if the value doesn't have primary key, will insert it
func (s *DB) Save(value interface{}) *DB {
scope := s.NewScope(value)
if !scope.PrimaryKeyZero() {
newDB := scope.callCallbacks(s.parent.callbacks.updates).db
if newDB.Error == nil && newDB.RowsAffected == 0 {
return s.New().FirstOrCreate(value)
}
return newDB
}
return scope.callCallbacks(s.parent.callbacks.creates).db
}
原因:
在保存数据时,如果发现主键为空值时,会去insert这条数据,但其实这条数据已经存在,于是就多出来了一条。对于这个问题作者给出的解释是,正常情况下主键不应该为空,那如果像我这样需要复合主键,且有一个主键是会有空值的情况怎么办呢?作者给出一个解决方法:使用指针!比如我这个可能为空的主键现在是string,那么可以设置为*string,这样当键值为空字符串时,指针也不是nil,即指针不是一个空值,就不会出现这样的问题啦。