From 14e0bb177fa4d7d352ab7d4572bd448278cf54ff Mon Sep 17 00:00:00 2001
From: pengnanfang <1064977639@qq.com>
Date: Thu, 14 Jan 2021 23:05:39 +0800
Subject: [PATCH] tablePrefix

---
 data/view/model/genmysql/genmysql.go |  5 +++++
 data/view/model/model.go             | 11 ++++++++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/data/view/model/genmysql/genmysql.go b/data/view/model/genmysql/genmysql.go
index cc44f2d..c9245d4 100644
--- a/data/view/model/genmysql/genmysql.go
+++ b/data/view/model/genmysql/genmysql.go
@@ -218,6 +218,11 @@ func (m *mysqlModel) getTables(orm *mysqldb.MySqlDB) map[string]string {
 	for rows.Next() {
 		var table string
 		rows.Scan(&table)
+		tablePrefix := config.GetTablePrefix()
+		if tablePrefix != "" && !strings.Contains(table, tablePrefix) {
+			// 不包含前缀则跳过
+			continue
+		}
 		tables = append(tables, table)
 		tbDesc[table] = ""
 	}
diff --git a/data/view/model/model.go b/data/view/model/model.go
index 75e1638..9344352 100644
--- a/data/view/model/model.go
+++ b/data/view/model/model.go
@@ -57,7 +57,11 @@ func (m *_Model) GetPackage() genstruct.GenPackage {
 
 			//如果设置了表前缀
 			if tablePrefix != "" {
-				tab.Name = strings.TrimLeft(tab.Name, tablePrefix)
+				var hasPrefix = strings.Contains(tab.Name, tablePrefix)
+				if !hasPrefix {
+					// 不包含前缀则跳过
+					continue
+				}
 			}
 
 			sct.SetStructName(getCamelName(tab.Name)) // Big hump.大驼峰
@@ -248,6 +252,11 @@ func (m *_Model) generateFunc() (genOut []GenOutInfo) {
 	// -------end------
 
 	for _, tab := range m.info.TabList {
+		tablePrefix := config.GetTablePrefix()
+		if tablePrefix != "" && !strings.Contains(tab.Name, tablePrefix) {
+			// 不包含前缀则跳过
+			continue
+		}
 		var pkg genstruct.GenPackage
 		pkg.SetPackage(m.info.PackageName) //package name
 		pkg.AddImport(`"fmt"`)