数据库初始化

This commit is contained in:
白茶清欢 2022-06-11 19:05:54 +08:00
parent 7cca5c5939
commit cdf075a576
2 changed files with 98 additions and 12 deletions

102
client.go
View File

@ -50,7 +50,37 @@ type client struct {
// Author : go_developer@163.com<白茶清欢> // Author : go_developer@163.com<白茶清欢>
// //
// Date : 19:19 2022/6/5 // Date : 19:19 2022/6/5
func (c *client) AddWithConfigFile(cfgFile string) error { func (c *client) AddWithConfigFile(cfgFilePath string) error {
var (
err error
cfg *cfgFile
dbClient *DBClient
)
if cfg, err = c.getCfg(cfgFilePath); nil != err {
return err
}
if nil == cfg {
// 不支持的配置文件格式
return nil
}
dbClient = &DBClient{
dbFlag: cfg.Flag,
loggerInstance: nil,
master: nil,
slave: nil,
extraFieldList: nil,
cfg: Mysql{},
}
if dbClient.master, err = c.GetDatabaseClient(cfg.Config.Master, nil); nil != err {
return err
}
if dbClient.slave, err = c.GetDatabaseClient(cfg.Config.Slave, nil); nil != err {
return err
}
c.lock.Lock()
c.clientTable[dbClient.dbFlag] = dbClient
c.lock.Unlock()
return nil return nil
} }
@ -90,20 +120,22 @@ func (c *client) getMysqlCfgFileList(cfgDir string) ([]*cfgFile, error) {
// //
// Date : 18:05 2022/6/11 // Date : 18:05 2022/6/11
func (c *client) getCfg(cfgPath string) (*cfgFile, error) { func (c *client) getCfg(cfgPath string) (*cfgFile, error) {
result := &cfgFile{
Path: cfgPath,
Type: "",
Config: Mysql{},
}
fileArr := strings.Split(cfgPath, ".") fileArr := strings.Split(cfgPath, ".")
if len(fileArr) < 2 { if len(fileArr) < 2 {
// 获取不到类型 // 获取不到类型
return nil, errors.New("文件格式必须是JSON或者YAML") return nil, errors.New("文件格式必须是JSON或者YAML")
} }
fileType := strings.ToLower(fileArr[len(fileArr)-1]) fileType := strings.ToLower(fileArr[len(fileArr)-1])
fileFlagArr := strings.Split(fileArr[0], string(filepath.Separator))
result := &cfgFile{
Path: cfgPath,
Type: "",
Flag: fileFlagArr[len(fileFlagArr)-1],
Config: Database{},
}
var ( var (
err error err error
cfgInfo Mysql cfgInfo Database
) )
switch fileType { switch fileType {
case FileTypeYaml: case FileTypeYaml:
@ -113,17 +145,22 @@ func (c *client) getCfg(cfgPath string) (*cfgFile, error) {
if err = util.File.ReadYmlContent(cfgPath, &result.Config); nil != err { if err = util.File.ReadYmlContent(cfgPath, &result.Config); nil != err {
return nil, fmt.Errorf("%s 配置文件解析失败, 原因 : %s", cfgPath, err.Error()) return nil, fmt.Errorf("%s 配置文件解析失败, 原因 : %s", cfgPath, err.Error())
} }
return result, nil
case FileTypeJson: case FileTypeJson:
result.Type = FileTypeJson result.Type = FileTypeJson
if err = util.File.ReadJSONContent(cfgPath, &cfgInfo); nil != err { if err = util.File.ReadJSONContent(cfgPath, &cfgInfo); nil != err {
return nil, fmt.Errorf("%s 配置文件解析失败, 原因 : %s", cfgPath, err.Error()) return nil, fmt.Errorf("%s 配置文件解析失败, 原因 : %s", cfgPath, err.Error())
} }
return result, nil
default: default:
// 不是JSON , 也不是YML, 跳过 // 不是JSON , 也不是YML, 跳过
return nil, nil return nil, nil
} }
if len(result.Config.Master.Timezone) == 0 {
// 默认使用本地时区
result.Config.Master.Timezone = "Local"
} else {
result.Config.Slave.Timezone = result.Config.Master.Timezone
}
return result, nil
} }
// GetDBClient 获取db client // GetDBClient 获取db client
@ -217,6 +254,52 @@ func (c *client) getGormClient() (*gorm.DB, error) {
return nil, nil return nil, nil
} }
// GetDatabaseClient 获取数据库连接
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 18:41 2022/6/11
func (c *client) GetDatabaseClient(conf *Mysql, logConf *LogConfig) (*gorm.DB, error) {
var (
instance *gorm.DB
err error
loggerInstance *zap.Logger
)
if instance, err = gorm.Open(mysql.Open(c.buildConnectionDSN(conf)), &gorm.Config{}); nil != err {
return nil, err
}
if nil != logConf {
if loggerInstance, err = getLogInstance(logConf, nil); nil != err {
return nil, err
}
instance.Logger = wrapper.NewGormLoggerWithInstance(nil, instance, loggerInstance, "", nil)
}
return instance, nil
}
// buildConnectionDSN 构建连接信息
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 18:42 2022/6/11
func (c *client) buildConnectionDSN(conf *Mysql) string {
return fmt.Sprintf(
"%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=%s",
conf.Username,
conf.Password,
conf.Host,
conf.Port,
conf.Database,
conf.Charset,
conf.Timezone,
)
}
// ========================== 以下是老版本支持
// NewDBClient ... // NewDBClient ...
// //
// Author : go_developer@163.com<白茶清欢> // Author : go_developer@163.com<白茶清欢>
@ -251,6 +334,7 @@ type DBClient struct {
master *gorm.DB // 主库 master *gorm.DB // 主库
slave *gorm.DB // 从库 slave *gorm.DB // 从库
extraFieldList []string // 提取的字段 extraFieldList []string // 提取的字段
cfg Mysql // 数据库配置
} }
// SetFlag 设置数据库标识 // SetFlag 设置数据库标识

View File

@ -48,9 +48,10 @@ type LogConfig struct {
// //
// Date : 14:47 2022/6/9 // Date : 14:47 2022/6/9
type cfgFile struct { type cfgFile struct {
Flag string `json:"flag"` // 数据库标识
Path string `json:"path"` // 配置文件路径 Path string `json:"path"` // 配置文件路径
Type string `json:"type"` // 配置文件类型 Type string `json:"type"` // 配置文件类型
Config Mysql `json:"config"` // 解析之后的配置文件 Config Database `json:"config"` // 解析之后的配置文件
} }
const ( const (
@ -86,6 +87,7 @@ type Mysql struct {
Charset string `json:"charset" yaml:"charset"` // 数据库编码 Charset string `json:"charset" yaml:"charset"` // 数据库编码
Connection *Connection `json:"connection" yaml:"connection"` // 连接配置 Connection *Connection `json:"connection" yaml:"connection"` // 连接配置
LogFileName string `json:"log_file_name" yaml:"log_file_name"` // 日志文件名 LogFileName string `json:"log_file_name" yaml:"log_file_name"` // 日志文件名
Timezone string `json:"timezone" yaml:"timezone"` // 时区
} }
// Connection 连接数配置 // Connection 连接数配置