120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
|
/*
|
|||
|
|
redisUtil对Redis的连接池进行了一定程度的封装
|
|||
|
|
将常用的方法进行了内部封装，对于不常见的方法，有两种处理方式：
|
|||
|
|
1、向作者提出请求，由作者添加到代码中
|
|||
|
|
2、调用GetConnection方法，然后自己实现逻辑
|
|||
|
|
在代码中，统一将conn.Do的结果和redis.Int,redis.String等类型转换合并处理
|
|||
|
|
|
|||
|
|
redis的命令请参考：https://redis.readthedocs.io/en/2.6/index.html
|
|||
|
|
*/
|
|||
|
|
package redisUtil
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gomodule/redigo/redis"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 自定义Redis连接池对象
|
|||
|
|
type RedisPool struct {
|
|||
|
|
name string
|
|||
|
|
address string
|
|||
|
|
pool *redis.Pool
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取自定义Redis连接池对象的名称
|
|||
|
|
// 返回值:
|
|||
|
|
// 自定义Redis连接池对象的名称
|
|||
|
|
func (this *RedisPool) GetName() string {
|
|||
|
|
return this.name
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取自定义Redis连接池对象的目标地址
|
|||
|
|
// 返回值:
|
|||
|
|
// 自定义Redis连接池对象的目标地址
|
|||
|
|
func (this *RedisPool) GetAddress() string {
|
|||
|
|
return this.address
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 从自定义连接池中获取连接，在使用后需要调用Close方法
|
|||
|
|
// 返回值:
|
|||
|
|
// 连接对象
|
|||
|
|
func (this *RedisPool) GetConnection() redis.Conn {
|
|||
|
|
return this.pool.Get()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 关闭自定义连接池
|
|||
|
|
func (this *RedisPool) Close() {
|
|||
|
|
this.pool.Close()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试连接情况
|
|||
|
|
// 返回值:
|
|||
|
|
// 错误对象
|
|||
|
|
func (this *RedisPool) TestConnection() error {
|
|||
|
|
conn := this.GetConnection()
|
|||
|
|
defer conn.Close()
|
|||
|
|
|
|||
|
|
_, err := conn.Do("PING")
|
|||
|
|
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建新的Redis连接池对象(obsolete，建议使用NewRedisPool2)
|
|||
|
|
// name:连接池对象名称
|
|||
|
|
// connectionString:Redis服务器连接地址
|
|||
|
|
// password：Redis服务器连接密码
|
|||
|
|
// database：Redis服务器选择的数据库
|
|||
|
|
// maxActive：Redis连接池允许的最大活跃连接数量
|
|||
|
|
// maxIdle：Redis连接池允许的最大空闲数量
|
|||
|
|
// idleTimeout：连接被回收前的空闲时间
|
|||
|
|
// dialConnectTimeout：连接Redis服务器超时时间
|
|||
|
|
// 返回值：
|
|||
|
|
// Redis连接池对象
|
|||
|
|
func NewRedisPool(name, connectionString, password string, database, maxActive, maxIdle int, idleTimeout, dialConnectTimeout time.Duration) *RedisPool {
|
|||
|
|
redisConfig := NewRedisConfig2(connectionString, password, database, maxActive, maxIdle, idleTimeout, dialConnectTimeout)
|
|||
|
|
return NewRedisPool2(name, redisConfig)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建新的Redis连接池对象
|
|||
|
|
// name:连接池对象名称
|
|||
|
|
// redisConfig:Redis配置对象
|
|||
|
|
// 返回值：
|
|||
|
|
// Redis连接池对象
|
|||
|
|
func NewRedisPool2(name string, redisConfig *RedisConfig) *RedisPool {
|
|||
|
|
poolObj := &redis.Pool{
|
|||
|
|
MaxActive: redisConfig.MaxActive,
|
|||
|
|
MaxIdle: redisConfig.MaxIdle,
|
|||
|
|
IdleTimeout: redisConfig.IdleTimeout,
|
|||
|
|
Dial: func() (redis.Conn, error) {
|
|||
|
|
options := make([]redis.DialOption, 0, 4)
|
|||
|
|
options = append(options, redis.DialConnectTimeout(redisConfig.DialConnectTimeout))
|
|||
|
|
options = append(options, redis.DialDatabase(redisConfig.Database))
|
|||
|
|
if redisConfig.Password != "" {
|
|||
|
|
options = append(options, redis.DialPassword(redisConfig.Password))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
conn, err := redis.Dial("tcp", redisConfig.ConnectionString, options...)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("Dial failed, err:%s", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return conn, err
|
|||
|
|
},
|
|||
|
|
TestOnBorrow: func(conn redis.Conn, t time.Time) error {
|
|||
|
|
if time.Since(t) < time.Minute {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
_, err := conn.Do("PING")
|
|||
|
|
return err
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &RedisPool{
|
|||
|
|
name: name,
|
|||
|
|
address: redisConfig.ConnectionString,
|
|||
|
|
pool: poolObj,
|
|||
|
|
}
|
|||
|
|
}
|