mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 12:17:53 +08:00
107 lines
2.6 KiB
Go
107 lines
2.6 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客户端.
|
||
// Redis中文手册文档请参考：http://redisdoc.com/ ，Redis官方命令请参考：https://redis.io/commands
|
||
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
|
||
}
|
||
} |