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

remove code

Browse Source
This commit is contained in:
白茶清欢 2022-06-12 19:42:11 +08:00
parent 924aa5cbd6
commit 035aa5be51
1 changed files with 0 additions and 138 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

138
logger.go
View File

@ -1,138 +0,0 @@
// Package mysql ...
//
// Description : wrapper...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2021-11-15 6:02 下午
package mysql
import (
"github.com/pkg/errors"
"gorm.io/gorm"
)
// DBFullConfig ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 7:00 下午 2021/11/15
type DBFullConfig struct {
DBConfig *DBConfig
LogConfig *LogConfig
}
// DBWrapperConfig ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:58 下午 2021/11/15
type DBWrapperConfig struct {
Write *DBFullConfig
Read *DBFullConfig
}
// NewDBWrapper 获取包装后的数据库实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:45 下午 2021/11/15
func NewDBWrapper(cf *DBWrapperConfig) (*DBWrapper, error) {
var (
err error
)
if nil == cf {
return nil, errors.New("cf is nil")
}
if nil == cf.Write {
return nil, errors.New("write cf is nil")
}
if nil == cf.Read {
return nil, errors.New("read cf is nil")
}
wrapperInstance := &DBWrapper{}
if wrapperInstance.write, err = GetDatabaseClient(cf.Write.DBConfig, cf.Write.LogConfig); nil != err {
return nil, err
}
if wrapperInstance.read, err = GetDatabaseClient(cf.Read.DBConfig, cf.Read.LogConfig); nil != err {
return nil, err
}
return wrapperInstance, nil
}
// DBWrapper 数据库包装
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:02 下午 2021/11/15
type DBWrapper struct {
read *gorm.DB // 读库实例
write *gorm.DB // 写库实例
}
// SelectDBOption 数据库实力选择时的选项
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:33 下午 2021/11/15
type SelectDBOption struct {
Table string
ForceMaster bool
}
// SetSelectDBOption 设置选项
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:34 下午 2021/11/15
type SetSelectDBOption func(o *SelectDBOption)
// GetWriteInstance 获取主库实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:31 下午 2021/11/15
func (dw *DBWrapper) GetWriteInstance(instanceOptionList ...SetSelectDBOption) *gorm.DB {
option := dw.getOption(instanceOptionList...)
instance := dw.write
if len(option.Table) > 0 {
instance.Table(option.Table)
}
return instance
}
// GetReadInstance 获取读库实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:32 下午 2021/11/15
func (dw *DBWrapper) GetReadInstance(instanceOptionList ...SetSelectDBOption) *gorm.DB {
option := dw.getOption(instanceOptionList...)
instance := dw.read
if option.ForceMaster || nil == dw.read {
instance = dw.write
}
if len(option.Table) > 0 {
instance.Table(option.Table)
}
return instance
}
// getOption 生成option
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 6:35 下午 2021/11/15
func (dw *DBWrapper) getOption(instanceOptionList ...SetSelectDBOption) *SelectDBOption {
o := &SelectDBOption{
Table: "",
ForceMaster: false,
}
for _, f := range instanceOptionList {
f(o)
}
return o
}