update
parent
e807b9393c
commit
9c83b42ccd
|
@ -10,6 +10,8 @@ import (
|
|||
"time"
|
||||
)
|
||||
|
||||
const DefaultKey = "default"
|
||||
|
||||
var (
|
||||
DefaultConfig = &Config{
|
||||
Dsn: "root:root@tcp(127.0.0.1:3306)/?loc=Local&charset=utf8mb4&parseTime=true",
|
||||
|
@ -20,12 +22,14 @@ var (
|
|||
Debug: true,
|
||||
GormLogger: gormLogger.Default.LogMode(gormLogger.Info),
|
||||
}
|
||||
|
||||
instanceMap = make(map[string]*MyDB)
|
||||
)
|
||||
|
||||
type (
|
||||
MyDB struct {
|
||||
DB *gorm.DB
|
||||
SqlDB *sql.DB
|
||||
db *gorm.DB
|
||||
sqlDb *sql.DB
|
||||
}
|
||||
Config struct {
|
||||
Dsn string
|
||||
|
@ -38,7 +42,27 @@ type (
|
|||
}
|
||||
)
|
||||
|
||||
func New(config *Config) (*MyDB, error) {
|
||||
func Instance(key ...string) *MyDB {
|
||||
var key0 string
|
||||
|
||||
if len(key) > 0 {
|
||||
key0 = key[0]
|
||||
} else {
|
||||
key0 = DefaultKey
|
||||
}
|
||||
|
||||
instance, ok := instanceMap[key0]
|
||||
if !ok {
|
||||
panic(fmt.Errorf("%s not config", key))
|
||||
}
|
||||
return instance
|
||||
}
|
||||
|
||||
func NewDefault(config *Config) (*MyDB, error) {
|
||||
return New(DefaultKey, config)
|
||||
}
|
||||
|
||||
func New(key string, config *Config) (*MyDB, error) {
|
||||
var (
|
||||
maxLifeTime, _ = time.ParseDuration(DefaultConfig.MaxLifeTime)
|
||||
maxIdleTime, _ = time.ParseDuration(DefaultConfig.MaxIdleTime)
|
||||
|
@ -92,10 +116,13 @@ func New(config *Config) (*MyDB, error) {
|
|||
sqlDb.SetMaxIdleConns(config.MaxIdleConn)
|
||||
sqlDb.SetConnMaxLifetime(maxLifeTime)
|
||||
sqlDb.SetConnMaxIdleTime(maxIdleTime)
|
||||
return &MyDB{
|
||||
DB: db,
|
||||
SqlDB: sqlDb,
|
||||
}, nil
|
||||
|
||||
myDb := &MyDB{
|
||||
db: db,
|
||||
sqlDb: sqlDb,
|
||||
}
|
||||
instanceMap[key] = myDb
|
||||
return myDb, nil
|
||||
}
|
||||
|
||||
func DefaultGormLogger(level gormLogger.LogLevel) gormLogger.Interface {
|
||||
|
@ -113,7 +140,15 @@ func NewGormLogger(writer gormLogger.Writer, gormLoggerConfig gormLogger.Config)
|
|||
}
|
||||
|
||||
func (m *MyDB) Close() {
|
||||
if m.SqlDB != nil {
|
||||
m.SqlDB.Close()
|
||||
if m.sqlDb != nil {
|
||||
m.sqlDb.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MyDB) DB() *gorm.DB {
|
||||
return m.db
|
||||
}
|
||||
|
||||
func (m *MyDB) SqlDB() *sql.DB {
|
||||
return m.sqlDb
|
||||
}
|
||||
|
|
|
@ -6,7 +6,7 @@ import (
|
|||
)
|
||||
|
||||
func TestMysql(t *testing.T) {
|
||||
myDB, err := New(&Config{
|
||||
myDB, err := NewDefault(&Config{
|
||||
Dsn: "root:root@tcp(127.0.0.1:3306)/site?loc=Local&charset=utf8mb4&writeTimeout=3s&readTimeout=3s&timeout=2s&parseTime=true",
|
||||
MaxOpenConn: 16,
|
||||
MaxIdleConn: 4,
|
||||
|
@ -23,7 +23,7 @@ func TestMysql(t *testing.T) {
|
|||
defer myDB.Close()
|
||||
|
||||
var res = make(map[string]interface{})
|
||||
err = myDB.DB.Table("image").Limit(1).Take(&res).Error
|
||||
err = myDB.db.Table("image").Limit(1).Take(&res).Error
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
|
|
|
@ -8,6 +8,8 @@ import (
|
|||
"time"
|
||||
)
|
||||
|
||||
const DefaultKey = "default"
|
||||
|
||||
var (
|
||||
DefaultConfig = &Config{
|
||||
Addr: "127.0.0.1:6379",
|
||||
|
@ -18,11 +20,13 @@ var (
|
|||
MaxConnAge: "1h",
|
||||
IdleTimeout: "10m",
|
||||
}
|
||||
|
||||
instanceMap = make(map[string]*MyRedis)
|
||||
)
|
||||
|
||||
type (
|
||||
MyRedis struct {
|
||||
Redis *redis.Client
|
||||
redis *redis.Client
|
||||
ctx context.Context
|
||||
}
|
||||
|
||||
|
@ -37,7 +41,27 @@ type (
|
|||
}
|
||||
)
|
||||
|
||||
func New(config *Config) (*MyRedis, error) {
|
||||
func Instance(key ...string) *MyRedis {
|
||||
var key0 string
|
||||
|
||||
if len(key) > 0 {
|
||||
key0 = key[0]
|
||||
} else {
|
||||
key0 = DefaultKey
|
||||
}
|
||||
|
||||
instance, ok := instanceMap[key0]
|
||||
if !ok {
|
||||
panic(fmt.Errorf("%s not config", key))
|
||||
}
|
||||
return instance
|
||||
}
|
||||
|
||||
func NewDefault(config *Config) (*MyRedis, error) {
|
||||
return New(DefaultKey, config)
|
||||
}
|
||||
|
||||
func New(key string, config *Config) (*MyRedis, error) {
|
||||
var (
|
||||
maxConnAge, _ = time.ParseDuration(DefaultConfig.MaxConnAge)
|
||||
idleTimeout, _ = time.ParseDuration(DefaultConfig.IdleTimeout)
|
||||
|
@ -82,18 +106,19 @@ func New(config *Config) (*MyRedis, error) {
|
|||
ctx := context.Background()
|
||||
rd := &MyRedis{}
|
||||
rd.ctx = ctx
|
||||
rd.Redis = client
|
||||
ping := rd.Redis.Ping(ctx)
|
||||
rd.redis = client
|
||||
ping := rd.redis.Ping(ctx)
|
||||
if ping.Err() != nil {
|
||||
return nil, fmt.Errorf("connet redis err: %s", ping.Err())
|
||||
}
|
||||
|
||||
instanceMap[key] = rd
|
||||
return rd, nil
|
||||
}
|
||||
|
||||
// Get 通用get
|
||||
func (r *MyRedis) Get(key string) (string, error) {
|
||||
return r.Redis.Get(r.ctx, key).Result()
|
||||
return r.redis.Get(r.ctx, key).Result()
|
||||
}
|
||||
|
||||
// Set 通用set
|
||||
|
@ -102,12 +127,12 @@ func (r *MyRedis) Set(key string, value interface{}, t ...time.Duration) (string
|
|||
if len(t) > 0 {
|
||||
t2 = t[0]
|
||||
}
|
||||
return r.Redis.Set(r.ctx, key, value, t2).Result()
|
||||
return r.redis.Set(r.ctx, key, value, t2).Result()
|
||||
}
|
||||
|
||||
// GetJson json序列化
|
||||
func (r *MyRedis) GetJson(key string) (interface{}, error) {
|
||||
res := r.Redis.Get(r.ctx, key)
|
||||
res := r.redis.Get(r.ctx, key)
|
||||
if res.Err() != nil {
|
||||
return nil, res.Err()
|
||||
}
|
||||
|
@ -134,11 +159,15 @@ func (r *MyRedis) SetJson(key string, value interface{}, t ...time.Duration) (st
|
|||
if err != nil {
|
||||
return "", fmt.Errorf("set key:%s 序列化json失败", key)
|
||||
}
|
||||
return r.Redis.Set(r.ctx, key, v, t2).Result()
|
||||
return r.redis.Set(r.ctx, key, v, t2).Result()
|
||||
}
|
||||
|
||||
func (r *MyRedis) Close() {
|
||||
if r.Redis != nil {
|
||||
r.Redis.Close()
|
||||
if r.redis != nil {
|
||||
r.redis.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func (r *MyRedis) Redis() *redis.Client {
|
||||
return r.redis
|
||||
}
|
||||
|
|
|
@ -7,7 +7,7 @@ import (
|
|||
)
|
||||
|
||||
func TestRedis(t *testing.T) {
|
||||
redis, err := New(&Config{
|
||||
redis, err := NewDefault(&Config{
|
||||
Addr: "127.0.0.1:6379",
|
||||
Password: "",
|
||||
DB: 15,
|
||||
|
|
Loading…
Reference in New Issue