tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(migrator.go and migrator/migrator.go) : remove Table Struct replace with []string

Browse Source
This commit is contained in:
dino.ma 2021-11-09 20:09:19 +08:00
parent f676d504f3
commit acde65b965
2 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
migrator.go
View File

@ -36,11 +36,6 @@ type ViewOption struct {
// ALLTables get all database tables // ALLTables get all database tables
const ALLTables = "migrator:all_tables" const ALLTables = "migrator:all_tables"
// Table Database table list info
type Table struct {
TableName string `gorm:"column:TABLE_NAME"`
}
type ColumnType interface { type ColumnType interface {
Name() string Name() string
DatabaseTypeName() string DatabaseTypeName() string
@ -62,7 +57,8 @@ type Migrator interface {
DropTable(dst ...interface{}) error DropTable(dst ...interface{}) error
HasTable(dst interface{}) bool HasTable(dst interface{}) bool
RenameTable(oldName, newName interface{}) error RenameTable(oldName, newName interface{}) error
GetTables(tables ...string) (tableList []Table, err error) GetTables(tables ...string) (tableList []string, err error)
// Columns // Columns
AddColumn(dst interface{}, field string) error AddColumn(dst interface{}, field string) error
DropColumn(dst interface{}, field string) error DropColumn(dst interface{}, field string) error

2
migrator/migrator.go
View File

@ -162,7 +162,7 @@ func (m Migrator) AutoMigrate(values ...interface{}) error {
return nil return nil
} }
func (m Migrator) GetTables(tables ...string) (tableList []gorm.Table, err error) { func (m Migrator) GetTables(tables ...string) (tableList []string, err error) {
if len(tables) == 1 && tables[0] == gorm.ALLTables { if len(tables) == 1 && tables[0] == gorm.ALLTables {
return tableList, m.DB.Raw(allTableQuery, m.CurrentDatabase()).Scan(&tableList).Error return tableList, m.DB.Raw(allTableQuery, m.CurrentDatabase()).Scan(&tableList).Error
} }