114 lines
2.5 KiB
Go
114 lines
2.5 KiB
Go
package infra
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"voltaserve/config"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type RedisManager struct {
|
|
config config.RedisConfig
|
|
client *redis.Client
|
|
clusterClient *redis.ClusterClient
|
|
}
|
|
|
|
func NewRedisManager() *RedisManager {
|
|
mgr := new(RedisManager)
|
|
mgr.config = config.GetConfig().Redis
|
|
return mgr
|
|
}
|
|
|
|
func (mgr *RedisManager) Set(key string, value interface{}) error {
|
|
if mgr.client == nil && mgr.clusterClient == nil {
|
|
if err := mgr.connect(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if mgr.clusterClient != nil {
|
|
if _, err := mgr.clusterClient.Set(context.Background(), key, value, 0).Result(); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if _, err := mgr.client.Set(context.Background(), key, value, 0).Result(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (mgr *RedisManager) Get(key string) (string, error) {
|
|
if mgr.client == nil && mgr.clusterClient == nil {
|
|
if err := mgr.connect(); err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
if mgr.clusterClient != nil {
|
|
value, err := mgr.clusterClient.Get(context.Background(), key).Result()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return value, nil
|
|
} else {
|
|
value, err := mgr.client.Get(context.Background(), key).Result()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return value, nil
|
|
}
|
|
}
|
|
|
|
func (mgr *RedisManager) Delete(key string) error {
|
|
if mgr.client == nil && mgr.clusterClient == nil {
|
|
if err := mgr.connect(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if mgr.clusterClient != nil {
|
|
if _, err := mgr.clusterClient.Del(context.Background(), key).Result(); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if _, err := mgr.client.Del(context.Background(), key).Result(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (mgr *RedisManager) Close() error {
|
|
if mgr.client != nil {
|
|
if err := mgr.client.Close(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (mgr *RedisManager) connect() error {
|
|
if mgr.client != nil || mgr.clusterClient != nil {
|
|
return nil
|
|
}
|
|
addresses := strings.Split(mgr.config.Address, ";")
|
|
if len(addresses) > 1 {
|
|
mgr.clusterClient = redis.NewClusterClient(&redis.ClusterOptions{
|
|
Addrs: addresses,
|
|
Password: mgr.config.Password,
|
|
})
|
|
if err := mgr.clusterClient.Ping(context.Background()).Err(); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
mgr.client = redis.NewClient(&redis.Options{
|
|
Addr: mgr.config.Address,
|
|
Password: mgr.config.Password,
|
|
DB: mgr.config.DB,
|
|
})
|
|
if err := mgr.client.Ping(context.Background()).Err(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|