- 优化

memcached.go

@@ -4,23 +4,28 @@ import (
 	"github.com/bradfitz/gomemcache/memcache"
 )
 
+// MemcachedConfig 配置
+type MemcachedConfig struct {
+	Dns string           // 连接地址，可选
+	Db  *memcache.Client // 驱动，可选
+}
+
 // Memcached https://github.com/bradfitz/gomemcache
 type Memcached struct {
 	db *memcache.Client // 驱动
 }
 
 // NewMemcached 实例化
-func NewMemcached(dns string) *Memcached {
-	mc := memcache.New(dns)
-	if mc == nil {
-		panic("连接失败")
+func NewMemcached(config *MemcachedConfig) *Memcached {
+	if config.Dns == "" {
+		return &Memcached{db: config.Db}
+	} else {
+		mc := memcache.New(config.Dns)
+		if mc == nil {
+			panic("连接失败")
+		}
+		return &Memcached{db: mc}
 	}
-	return &Memcached{db: mc}
-}
-
-// NewMemcachedDb 实例化
-func NewMemcachedDb(memcached *memcache.Client) *Memcached {
-	return &Memcached{db: memcached}
 }
 
 // Set 插入数据

redis.go

@@ -9,36 +9,48 @@ import (
 	"time"
 )
 
+// RedisConfig 配置
+type RedisConfig struct {
+	Addr              string        // 地址，可选
+	Password          string        // 密码，可选
+	DbName            int           // 数据库，可选
+	DefaultExpiration time.Duration // 默认过期时间
+	Db                *redis.Client // 驱动，可选
+}
+
 // Redis https://github.com/go-redis/redis
 type Redis struct {
-	db         *redis.Client   // 驱动
-	ctx        context.Context // 上下文内容
-	expiration time.Duration   // 默认过期时间
+	RedisConfig
+	db  *redis.Client   // 驱动
+	ctx context.Context // 上下文内容
 }
 
 // NewRedis 实例化
-func NewRedis(addr, password string, dbName int, expiration time.Duration) *Redis {
-	db := redis.NewClient(&redis.Options{
-		Addr:     addr,     // 地址
-		Password: password, // 密码
-		DB:       dbName,   // 数据库
-		PoolSize: 100,      // 连接池大小
-	})
-
-	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-	defer cancel()
-
-	_, err := db.Ping(ctx).Result()
-	if err != nil {
-		panic(errors.New(fmt.Sprintf("连接失败，%s", err.Error())))
+func NewRedis(config *RedisConfig) *Redis {
+	app := &Redis{}
+	app.DefaultExpiration = config.DefaultExpiration
+	app.ctx = context.Background()
+	if config.Addr == "" {
+		app.db = config.Db
+		return app
+	} else {
+		app.db = redis.NewClient(&redis.Options{
+			Addr:     config.Addr,     // 地址
+			Password: config.Password, // 密码
+			DB:       config.DbName,   // 数据库
+			PoolSize: 100,             // 连接池大小
+		})
+
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+
+		_, err := app.db.Ping(ctx).Result()
+		if err != nil {
+			panic(errors.New(fmt.Sprintf("连接失败，%s", err.Error())))
+		}
+
+		return app
 	}
-
-	return &Redis{db: db, ctx: context.Background(), expiration: expiration}
-}
-
-// NewRedisDb 实例化
-func NewRedisDb(db *redis.Client, expiration time.Duration) *Redis {
-	return &Redis{db: db, ctx: context.Background(), expiration: expiration}
 }
 
 // Set 设置一个key的值
@@ -54,13 +66,13 @@ func (r *Redis) SetInterface(key string, value interface{}, expiration time.Dura
 
 // SetDefaultExpiration 设置一个key的值，使用全局默认过期时间
 func (r *Redis) SetDefaultExpiration(key string, value interface{}) (string, error) {
-	return r.db.Set(r.ctx, key, value, r.expiration).Result()
+	return r.db.Set(r.ctx, key, value, r.DefaultExpiration).Result()
 }
 
 // SetInterfaceDefaultExpiration 设置一个key的值，使用全局默认过期时间
 func (r *Redis) SetInterfaceDefaultExpiration(key string, value interface{}) (string, error) {
 	marshal, _ := json.Marshal(value)
-	return r.db.Set(r.ctx, key, marshal, r.expiration).Result()
+	return r.db.Set(r.ctx, key, marshal, r.DefaultExpiration).Result()
 }
 
 // Get 查询key的值
@@ -90,7 +102,7 @@ func (r *Redis) SetNX(key string, value interface{}, expiration time.Duration) e
 
 // SetNXDefaultExpiration 如果key不存在，则设置这个key的值，使用全局默认过期时间
 func (r *Redis) SetNXDefaultExpiration(key string, value interface{}) error {
-	return r.db.SetNX(r.ctx, key, value, r.expiration).Err()
+	return r.db.SetNX(r.ctx, key, value, r.DefaultExpiration).Err()
 }
 
 // MGet 批量查询key的值

redis_cache.go

@@ -5,22 +5,33 @@ import (
 	"time"
 )
 
+// RedisCacheConfig 配置
+type RedisCacheConfig struct {
+	expiration time.Duration // 过期时间
+}
+
 // RedisCache https://github.com/go-redis/redis
 type RedisCache struct {
+	RedisCacheConfig
 	db              *Redis           // 驱动
-	expiration      time.Duration    // 默认过期时间
 	GetterString    GttStringFunc    // 不存在的操作
 	GetterInterface GttInterfaceFunc // 不存在的操作
 }
 
 // NewCache 实例化
-func (r *Redis) NewCache(expiration time.Duration) *RedisCache {
-	return &RedisCache{db: r, expiration: expiration}
+func (r *Redis) NewCache(config *RedisCacheConfig) *RedisCache {
+	app := &RedisCache{}
+	app.expiration = config.expiration
+	app.db = r
+	return app
 }
 
 // NewCacheDefaultExpiration 实例化
 func (r *Redis) NewCacheDefaultExpiration() *RedisCache {
-	return &RedisCache{db: r, expiration: r.expiration}
+	app := &RedisCache{}
+	app.expiration = r.DefaultExpiration
+	app.db = r
+	return app
 }
 
 // GetString 缓存操作