Update model_struct.go

处理在同时定义gorm.DefaultTableNameHandler和Model.TableName时,db.Find()与db.Where()函数获取表名结果不一致的问题
This commit is contained in:
hot1232 2018-11-20 18:05:26 +08:00 committed by GitHub
parent 472c70caa4
commit 859ebcfaa6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -13,9 +13,7 @@ import (
)
// DefaultTableNameHandler default table name handler
var DefaultTableNameHandler = func(db *DB, defaultTableName string) string {
return defaultTableName
}
var DefaultTableNameHandler = func(db *DB, defaultTableName string) string
var modelStructsMap sync.Map
@ -46,7 +44,9 @@ func (s *ModelStruct) TableName(db *DB) string {
s.defaultTableName = tableName
}
}
if nil == DefaultTableNameHandler {
return s.defaultTableName
}
return DefaultTableNameHandler(db, s.defaultTableName)
}