diff --git a/client.go b/client.go index 43ad8f3..225e96a 100644 --- a/client.go +++ b/client.go @@ -22,6 +22,15 @@ import ( "gorm.io/gorm" ) +// BatchNewDBClient 批量设置数据库实例, 读取指定目录下的yaml文件, 文件名作为标识, 内容是数据库具体配置 +// +// Author : go_developer@163.com<白茶清欢> +// +// Date : 15:43 2022/6/5 +func BatchNewDBClient(dbConfigDir string) { + +} + // NewDBClient ... // // Author : go_developer@163.com<白茶清欢> @@ -51,10 +60,20 @@ func NewDBClient(masterConf *DBConfig, slaveConf *DBConfig, logConf *LogConfig, // // Date : 3:09 PM 2021/12/24 type DBClient struct { - loggerInstance *zap.Logger - master *gorm.DB - slave *gorm.DB - extraFieldList []string + dbFlag string // 数据库标识 + loggerInstance *zap.Logger // 日志实例 + master *gorm.DB // 主库 + slave *gorm.DB // 从库 + extraFieldList []string // 提取的字段 +} + +// SetFlag 设置数据库标识 +// +// Author : go_developer@163.com<白茶清欢> +// +// Date : 16:18 2022/6/5 +func (dc *DBClient) SetFlag(dbFlag string) { + dc.dbFlag = dbFlag } // GetMaster 获取主库连接