From 722e0b0227ae3b75491a223727fd4d1b0c144ffc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A5=89=E7=9B=B4?= <jianxuan.sjx@alibaba-inc.com>
Date: Tue, 21 Apr 2020 10:30:29 +0800
Subject: [PATCH] model struct support dbTabler method TableName()

---
 model_struct.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/model_struct.go b/model_struct.go
index d9e2e90f..c26e6c80 100644
--- 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()