From 0a62fb50eb4e3e2cdb9e0c518c71174f4cb367dd Mon Sep 17 00:00:00 2001 From: "dino.ma" Date: Fri, 12 Nov 2021 17:50:49 +0800 Subject: [PATCH] feat(migrator.go) : add GetTables method. (#71) * feat(migrator.go) : add GetTables method. * fix(migrator.go): use CurrentSchema method --- migrator.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/migrator.go b/migrator.go index 392b47a..70bab96 100644 --- 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