all

Downloads/Voltaserve/api/infra/redis.go

@@ -0,0 +1,113 @@
+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
+}