package redisClient

import (
	"errors"
	"github.com/go-redis/redis/v8"
	"sync"
)

// 保持向后兼容的全局客户端
var Client *redis.Client

// 连接管理器，用于管理多个 Redis 连接
var (
	clients    = make(map[string]*redis.Client)
	clientsMux sync.RWMutex
)

// InitRedis 初始化默认 Redis 客户端（保持向后兼容）
func InitRedis(addr, password string, db int) {
	Client = redis.NewClient(&redis.Options{
		Addr:     addr,
		Password: password,
		DB:       db,
	})
	// 同时将默认客户端添加到连接管理器
	clientsMux.Lock()
	clients["default"] = Client
	clientsMux.Unlock()
}

// GetClient 获取默认 Redis 客户端（保持向后兼容）
func GetClient() *redis.Client {
	return Client
}

// CloseRedis 关闭默认 Redis 客户端（保持向后兼容）
func CloseRedis() error {
	if Client != nil {
		return Client.Close()
	}
	return nil
}

// AddClient 添加一个新的 Redis 客户端
func AddClient(name, addr, password string, db int) {
	client := redis.NewClient(&redis.Options{
		Addr:     addr,
		Password: password,
		DB:       db,
	})
	clientsMux.Lock()
	clients[name] = client
	clientsMux.Unlock()
}

// GetClientByName 根据名称获取 Redis 客户端
func GetClientByName(name string) (*redis.Client, error) {
	clientsMux.RLock()
	defer clientsMux.RUnlock()
	client, ok := clients[name]
	if !ok {
		return nil, errors.New("redis client not found: " + name)
	}
	return client, nil
}

// CloseClient 关闭指定名称的 Redis 客户端
func CloseClient(name string) error {
	clientsMux.Lock()
	defer clientsMux.Unlock()
	client, ok := clients[name]
	if !ok {
		return errors.New("redis client not found: " + name)
	}
	err := client.Close()
	if err == nil {
		delete(clients, name)
	}
	return err
}

// CloseAllClients 关闭所有 Redis 客户端
func CloseAllClients() error {
	clientsMux.Lock()
	defer clientsMux.Unlock()
	var lastErr error
	for name, client := range clients {
		if err := client.Close(); err != nil {
			lastErr = err
		}
		delete(clients, name)
	}
	return lastErr
}

// ListClients 列出所有可用的 Redis 客户端名称
func ListClients() []string {
	clientsMux.RLock()
	defer clientsMux.RUnlock()
	names := make([]string, 0, len(clients))
	for name := range clients {
		names = append(names, name)
	}
	return names
}