80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package redis
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/logger"
|
|
"github.com/fiskerinc/cloud-services/pkg/utils/envtool"
|
|
|
|
"github.com/gomodule/redigo/redis"
|
|
"github.com/pkg/errors"
|
|
redigotrace "gopkg.in/DataDog/dd-trace-go.v1/contrib/gomodule/redigo"
|
|
)
|
|
|
|
// connection vars
|
|
var (
|
|
host = envtool.GetEnv("REDIS_HOST", "localhost")
|
|
port = envtool.GetEnv("REDIS_PORT", "6379")
|
|
password = envtool.GetEnv("REDIS_PASSWORD", "")
|
|
addr = fmt.Sprintf("%v:%v", host, port)
|
|
)
|
|
|
|
func UpdateRedisConnection(redisHost string, redisPort string, redisPassword string) {
|
|
host = redisHost
|
|
port = redisPort
|
|
password = redisPassword
|
|
addr = fmt.Sprintf("%v:%v", host, port)
|
|
|
|
}
|
|
|
|
// Pool provides redis connections
|
|
type Pool interface {
|
|
Get() redis.Conn
|
|
Stats() redis.PoolStats
|
|
ActiveCount() int
|
|
}
|
|
|
|
func NewPool() Pool {
|
|
return &redis.Pool{
|
|
IdleTimeout: time.Millisecond * time.Duration(envtool.GetEnvInt("REDIS_IDLETIMEOUT_MS", 1000)),
|
|
MaxIdle: envtool.GetEnvInt("REDIS_MAXIDLECONN", 10),
|
|
MaxActive: envtool.GetEnvInt("REDIS_MAXACTIVECONN", 10),
|
|
MaxConnLifetime: time.Millisecond * time.Duration(envtool.GetEnvInt("REDIS_MAXCONNLIFETIME_MS", 1000)),
|
|
Wait: (envtool.GetEnvInt("REDIS_WAITGETCONN", 0) == 1),
|
|
Dial: func() (redis.Conn, error) {
|
|
conn, err := redigotrace.Dial("tcp", addr, redis.DialKeepAlive(time.Minute*time.Duration(envtool.GetEnvInt("REDIS_KEEPALIVE_MINS", 10))))
|
|
if password == "" {
|
|
return conn, errors.WithStack(err)
|
|
}
|
|
if err != nil {
|
|
logger.Error().Err(errors.WithStack(err)).Send()
|
|
return nil, err
|
|
}
|
|
if _, err := conn.Do("AUTH", password); err != nil {
|
|
conn.Close()
|
|
logger.Error().Err(errors.WithStack(err)).Send()
|
|
return nil, err
|
|
}
|
|
return conn, nil
|
|
},
|
|
TestOnBorrow: func(c redis.Conn, t time.Time) error {
|
|
_, err := c.Do("PING")
|
|
return err
|
|
},
|
|
}
|
|
}
|
|
|
|
type Conn interface {
|
|
Do(string, ...interface{}) (interface{}, error)
|
|
DoWithTimeout(time.Duration, string, ...interface{}) (interface{}, error)
|
|
|
|
Receive() (interface{}, error)
|
|
ReceiveWithTimeout(time.Duration) (interface{}, error)
|
|
|
|
Send(string, ...interface{}) error
|
|
Err() error
|
|
Close() error
|
|
Flush() error
|
|
}
|