mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 12:17:53 +08:00
106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://gitee.com/johng/gf.
|
|
|
|
// Redis客户端.
|
|
package gredis
|
|
|
|
import (
|
|
"time"
|
|
"github.com/gomodule/redigo/redis"
|
|
)
|
|
|
|
const (
|
|
gDEFAULT_POOL_MAX_IDLE = 1
|
|
gDEFAULT_POOL_MAX_ACTIVE = 10
|
|
gDEFAULT_POOL_IDLE_TIMEOUT = 180 * time.Second
|
|
gDEFAULT_POOL_MAX_LIFE_TIME = 60 * time.Second
|
|
)
|
|
|
|
// Redis客户端
|
|
type Redis struct {
|
|
address string
|
|
db interface{}
|
|
pool *redis.Pool
|
|
}
|
|
|
|
// Redis链接池统计信息
|
|
type PoolStats struct {
|
|
redis.PoolStats
|
|
}
|
|
|
|
// 创建redis操作对象
|
|
// address参数格式 host:port
|
|
func New(address string, db ... interface{}) *Redis {
|
|
r := &Redis{}
|
|
r.address = address
|
|
if len(db) > 0 {
|
|
r.db = db[0]
|
|
}
|
|
r.pool = &redis.Pool {
|
|
MaxIdle : gDEFAULT_POOL_MAX_IDLE,
|
|
MaxActive : gDEFAULT_POOL_MAX_ACTIVE,
|
|
IdleTimeout : gDEFAULT_POOL_IDLE_TIMEOUT,
|
|
MaxConnLifetime : gDEFAULT_POOL_MAX_LIFE_TIME,
|
|
Dial : r.dialFunc(),
|
|
}
|
|
return r
|
|
}
|
|
|
|
// 关闭链接
|
|
func (r *Redis) Close() error {
|
|
return r.pool.Close()
|
|
}
|
|
|
|
// 设置属性 - MaxIdle
|
|
func (r *Redis) SetMaxIdle(value int) {
|
|
r.pool.MaxIdle = value
|
|
}
|
|
|
|
// 设置属性 - MaxActive
|
|
func (r *Redis) SetMaxActive(value int) {
|
|
r.pool.MaxActive = value
|
|
}
|
|
|
|
// 设置属性 - IdleTimeout
|
|
func (r *Redis) SetIdleTimeout(value time.Duration) {
|
|
r.pool.IdleTimeout = value
|
|
}
|
|
|
|
// 设置属性 - MaxConnLifetime
|
|
func (r *Redis) SetMaxConnLifetime(value time.Duration) {
|
|
r.pool.MaxConnLifetime = value
|
|
}
|
|
|
|
// 获取当前连接池统计信息
|
|
func (r *Redis) Stats() *PoolStats {
|
|
return &PoolStats{r.pool.Stats()}
|
|
}
|
|
|
|
// 执行命令 - Do
|
|
func (r *Redis) Do(command string, args ...interface{}) (interface{}, error) {
|
|
c := r.pool.Get()
|
|
defer c.Close()
|
|
return c.Do(command, args...)
|
|
}
|
|
|
|
// 执行命令 - Send
|
|
func (r *Redis) Send(command string, args ...interface{}) error {
|
|
c := r.pool.Get()
|
|
defer c.Close()
|
|
return c.Send(command, args...)
|
|
}
|
|
|
|
// 构造链接redis方法
|
|
func (r *Redis) dialFunc() func()(redis.Conn, error) {
|
|
return func() (redis.Conn, error) {
|
|
c, err := redis.Dial("tcp", r.address)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c.Do("SELECT", r.db)
|
|
return c, nil
|
|
}
|
|
} |