增加设置数据库标识

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 获取主库连接