diff --git a/model_struct.go b/model_struct.go index 8c27e209..2585988a 100644 --- a/model_struct.go +++ b/model_struct.go @@ -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) }