From 0f50b07a20b25be899e573dbb6b50d464f9001d1 Mon Sep 17 00:00:00 2001
From: Ming Deng <mingflycash@gmail.com>
Date: Mon, 7 Sep 2020 21:40:20 +0800
Subject: [PATCH] allow users to ignore some table when run orm commands

---
 pkg/client/orm/cmd.go               |  6 +++++
 pkg/client/orm/models.go            |  2 +-
 pkg/client/orm/models_utils.go      | 12 ++++++++++
 pkg/client/orm/models_utils_test.go | 35 +++++++++++++++++++++++++++++
 pkg/client/orm/types.go             |  5 +++++
 5 files changed, 59 insertions(+), 1 deletion(-)
 create mode 100644 pkg/client/orm/models_utils_test.go

diff --git a/pkg/client/orm/cmd.go b/pkg/client/orm/cmd.go
index e03fc0ee..b0661971 100644
--- a/pkg/client/orm/cmd.go
+++ b/pkg/client/orm/cmd.go
@@ -142,6 +142,12 @@ func (d *commandSyncDb) Run() error {
 	}
 
 	for i, mi := range modelCache.allOrdered() {
+
+		if !isApplicableTableForDB(mi.addrField, d.al.Name) {
+			fmt.Printf("table `%s` is not applicable to database '%s'\n", mi.table, d.al.Name)
+			continue
+		}
+
 		if tables[mi.table] {
 			if !d.noInfo {
 				fmt.Printf("table `%s` already exists, skip\n", mi.table)
diff --git a/pkg/client/orm/models.go b/pkg/client/orm/models.go
index a7de10f7..24f564ab 100644
--- a/pkg/client/orm/models.go
+++ b/pkg/client/orm/models.go
@@ -414,7 +414,7 @@ func (mc *_modelCache) getDbDropSQL(al *alias) (queries []string, err error) {
 	for _, mi := range modelCache.allOrdered() {
 		queries = append(queries, fmt.Sprintf(`DROP TABLE IF EXISTS %s%s%s`, Q, mi.table, Q))
 	}
-	return queries,nil
+	return queries, nil
 }
 
 //getDbCreateSQL get database scheme creation sql queries
diff --git a/pkg/client/orm/models_utils.go b/pkg/client/orm/models_utils.go
index 6fca59a9..950ca243 100644
--- a/pkg/client/orm/models_utils.go
+++ b/pkg/client/orm/models_utils.go
@@ -107,6 +107,18 @@ func getTableUnique(val reflect.Value) [][]string {
 	return nil
 }
 
+// get whether the table needs to be created for the database alias
+func isApplicableTableForDB(val reflect.Value, db string) bool {
+	fun := val.MethodByName("IsApplicableTableForDB")
+	if fun.IsValid() {
+		vals := fun.Call([]reflect.Value{reflect.ValueOf(db)})
+		if len(vals) > 0 && vals[0].Kind() == reflect.Bool {
+			return vals[0].Bool()
+		}
+	}
+	return true
+}
+
 // get snaked column name
 func getColumnName(ft int, addrField reflect.Value, sf reflect.StructField, col string) string {
 	column := col
diff --git a/pkg/client/orm/models_utils_test.go b/pkg/client/orm/models_utils_test.go
new file mode 100644
index 00000000..0a6995b3
--- /dev/null
+++ b/pkg/client/orm/models_utils_test.go
@@ -0,0 +1,35 @@
+// Copyright 2020
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package orm
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+type NotApplicableModel struct {
+	Id int
+}
+
+func (n *NotApplicableModel) IsApplicableTableForDB(db string) bool {
+	return db == "default"
+}
+
+func Test_IsApplicableTableForDB(t *testing.T) {
+	assert.False(t, isApplicableTableForDB(reflect.ValueOf(&NotApplicableModel{}), "defa"))
+	assert.True(t, isApplicableTableForDB(reflect.ValueOf(&NotApplicableModel{}), "default"))
+}
diff --git a/pkg/client/orm/types.go b/pkg/client/orm/types.go
index eb34e759..b0c793b7 100644
--- a/pkg/client/orm/types.go
+++ b/pkg/client/orm/types.go
@@ -75,6 +75,11 @@ type TableUniqueI interface {
 	TableUnique() [][]string
 }
 
+// IsApplicableTableForDB if return false, we won't create table to this db
+type IsApplicableTableForDB interface {
+	IsApplicableTableForDB(db string) bool
+}
+
 // Driver define database driver
 type Driver interface {
 	Name() string