zhangdeman/database
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

完成配量读取配置文件

Browse Source
This commit is contained in:
白茶清欢
2022-06-11 18:00:47 +08:00
parent e610e02d45
commit 879c9df7cc
3 changed files with 19 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
client.go
View File

@ -77,20 +77,27 @@ func (c *client) getMysqlCfgFileList(cfgDir string) ([]cfgFile, error) {
continue
}
fileType := strings.ToLower(fileArr[len(fileArr)-1])
var (
err error
)
cfgInfo := cfgFile{
Path: filepathNames[i],
Type: FileTypeYaml,
Config: Mysql{},
}
switch fileType {
case FileTypeYaml:
fallthrough
case FileTypeYml:
cfgFileList = append(cfgFileList, cfgFile{
Path: filepathNames[i],
Type: FileTypeYaml,
})
util.File.ReadYmlConfig()
cfgInfo.Type = FileTypeYaml
if err = util.File.ReadYmlContent(filepathNames[i], &cfgInfo); nil != err {
return nil, fmt.Errorf("%s 配置文件解析失败, 原因 : %s", cfgInfo.Path, err.Error())
}
case FileTypeJson:
cfgFileList = append(cfgFileList, cfgFile{
Path: filepathNames[i],
Type: FileTypeJson,
})
cfgInfo.Type = FileTypeJson
if err = util.File.ReadJSONContent(filepathNames[i], &cfgInfo); nil != err {
return nil, fmt.Errorf("%s 配置文件解析失败, 原因 : %s", cfgInfo.Path, err.Error())
}
default:
// 不是JSON , 也不是YML, 跳过
continue
@ -104,7 +111,7 @@ func (c *client) getMysqlCfgFileList(cfgDir string) ([]cfgFile, error) {
// Author : go_developer@163.com<白茶清欢>
//
// Date : 19:32 2022/6/5
func (c *client) GetDBClient(ctx *gin.Context, dbFlag string) (*DBClient, error) {
func (c *client) GetDBClient(dbFlag string) (*DBClient, error) {
c.lock.RLock()
defer c.lock.RUnlock()
var (
@ -127,7 +134,7 @@ func (c *client) GetMasterClient(ctx *gin.Context, dbFlag string) (*gorm.DB, err
err error
dbClient *DBClient
)
if dbClient, err = c.GetDBClient(ctx, dbFlag); nil != err {
if dbClient, err = c.GetDBClient(dbFlag); nil != err {
return nil, err
}
@ -146,7 +153,7 @@ func (c *client) GetSlaveClient(ctx *gin.Context, dbFlag string) (*gorm.DB, erro
err error
dbClient *DBClient
)
if dbClient, err = c.GetDBClient(ctx, dbFlag); nil != err {
if dbClient, err = c.GetDBClient(dbFlag); nil != err {
return nil, err
}
@ -190,15 +197,6 @@ func (c *client) getGormClient() (*gorm.DB, error) {
return nil, nil
}
// BatchNewDBClient 批量设置数据库实例, 读取指定目录下的yaml文件, 文件名作为标识, 内容是数据库具体配置
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 15:43 2022/6/5
func BatchNewDBClient(dbConfigDir string) {
}
// NewDBClient ...
//
// Author : go_developer@163.com<白茶清欢>