mirror of
https://gitee.com/johng/gf.git
synced 2024-12-02 12:17:53 +08:00
129 lines
3.4 KiB
Go
129 lines
3.4 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"
|
||
"gitee.com/johng/gf/third/github.com/gomodule/redigo/redis"
|
||
"gitee.com/johng/gf/g/container/gmap"
|
||
"fmt"
|
||
)
|
||
|
||
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 {
|
||
pool *redis.Pool
|
||
}
|
||
|
||
// Redis服务端但节点连接配置信息
|
||
type Config struct {
|
||
Host string // IP/域名
|
||
Port int // 端口
|
||
Db int // db
|
||
Pass string // 密码
|
||
}
|
||
|
||
// Redis链接池统计信息
|
||
type PoolStats struct {
|
||
redis.PoolStats
|
||
}
|
||
|
||
// 连接池map
|
||
var pools = gmap.NewStringInterfaceMap()
|
||
|
||
// 创建redis操作对象.
|
||
func New(config Config) *Redis {
|
||
r := &Redis{}
|
||
poolKey := fmt.Sprintf("%s:%d,%d", config.Host, config.Port, config.Db)
|
||
if v := pools.Get(poolKey); v == nil {
|
||
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 : func() (redis.Conn, error) {
|
||
c, err := redis.Dial("tcp", fmt.Sprintf("%s:%d", config.Host, config.Port))
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if len(config.Pass) > 0 {
|
||
if _, err := c.Do("AUTH", config.Pass); err != nil {
|
||
return nil, err
|
||
}
|
||
}
|
||
if _, err := c.Do("SELECT", config.Db); err != nil {
|
||
return nil, err
|
||
}
|
||
return c, nil
|
||
},
|
||
// 用来测试连接是否可用
|
||
TestOnBorrow: func(c redis.Conn, t time.Time) error {
|
||
_, err := c.Do("PING")
|
||
return err
|
||
},
|
||
}
|
||
pools.Set(poolKey, pool)
|
||
r.pool = pool
|
||
} else {
|
||
r.pool = v.(*redis.Pool)
|
||
}
|
||
return r
|
||
}
|
||
|
||
// 关闭redis管理对象，将会关闭底层的
|
||
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) {
|
||
conn := r.pool.Get()
|
||
defer conn.Close()
|
||
return conn.Do(command, args...)
|
||
}
|
||
|
||
// 执行异步命令 - Send
|
||
func (r *Redis) Send(command string, args ...interface{}) error {
|
||
conn := r.pool.Get()
|
||
defer conn.Close()
|
||
return conn.Send(command, args...)
|
||
}
|
||
|