diff --git a/system.go b/system.go index b656c16..c1151d6 100644 --- a/system.go +++ b/system.go @@ -21,6 +21,32 @@ import ( type SystemDao struct { } +// SelectDatabase 切换数据库 +// +// Author : go_developer@163.com<白茶清欢> +// +// Date : 17:44 2023/4/9 +func (sd *SystemDao) SelectDatabase(dbInstance *gorm.DB, database string) error { + return dbInstance.Raw("USE " + database).Error +} + +// GetDatabaseList 获取数据库列表 +// +// Author : go_developer@163.com<白茶清欢> +// +// Date : 17:46 2023/4/9 +func (sd *SystemDao) GetDatabaseList(dbInstance *gorm.DB) ([]string, error) { + var ( + err error + result []string + ) + + if err = dbInstance.Raw("SHOW DATABASES").Scan(&result).Error; nil != err { + return nil, err + } + return result, nil +} + // GetTableList 查询数据表 // // Author : go_developer@163.com<白茶清欢>