model struct support dbTabler method TableName()

2020-04-21 10:30:29 +08:00 · 2020-04-21 10:30:29 +08:00 · 722e0b0227
commit 722e0b0227
parent 7ea143b548
1 changed files with 2 additions and 0 deletions
--- a/model_struct.go
+++ b/model_struct.go
@ -42,6 +42,8 @@ func (s *ModelStruct) TableName(db *DB) string {
 		// Set default table name
 		if tabler, ok := reflect.New(s.ModelType).Interface().(tabler); ok {
 			s.defaultTableName = tabler.TableName()
+		} else if dbTabler, ok := reflect.New(s.ModelType).Interface().(dbTabler); ok {
+			s.defaultTableName = dbTabler.TableName(db)
 		} else {
 			tableName := ToTableName(s.ModelType.Name())
 			db.parent.RLock()