feat(migrator.go) : add GetTables method. (#71)

* feat(migrator.go) : add GetTables method. * fix(migrator.go): use CurrentSchema method
go-gorm · Nov 12, 2021 · 0a62fb5 · 0a62fb5
1 parent 6a0adf7
commit 0a62fb5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/migrator.go b/migrator.go
@@ -163,6 +163,11 @@ func (m Migrator) DropIndex(value interface{}, name string) error {
 	})
 }
 
+func (m Migrator) GetTables() (tableList []string, err error) {
+	currentSchema, _ := m.CurrentSchema(m.DB.Statement, "")
+	return tableList, m.DB.Raw("SELECT table_name FROM information_schema.tables WHERE table_schema = ? AND table_type = ?", currentSchema, "BASE TABLE").Scan(&tableList).Error
+}
+
 func (m Migrator) CreateTable(values ...interface{}) (err error) {
 	if err = m.Migrator.CreateTable(values...); err != nil {
 		return