package redisConnectUtil

import (
	"context"
	"fmt"
	"log"
	"time"

	"github.com/go-redis/redis/v8"
)

var (
	RedisClient *redis.Client
	ctx         = context.Background()
)

// InitRedis 初始化Redis连接
func InitRedis(host, password string, port, db int) error {
	// 如果Redis客户端已经存在且连接正常，直接返回
	if RedisClient != nil {
		_, err := RedisClient.Ping(ctx).Result()
		if err == nil {
			log.Println("Redis连接已存在且正常")
			return nil
		}
		// 连接异常，重新初始化
		log.Println("Redis连接异常，重新初始化")
	}
	
	RedisClient = redis.NewClient(&redis.Options{
		Addr:     fmt.Sprintf("%s:%d", host, port),
		Password: password,
		DB:       db,
	})

	log.Printf("开始连接Redis.... Addr: %s:%d, password:%s, db:%d", host, port, password, db)
	// 测试连接
	_, err := RedisClient.Ping(ctx).Result()
	if err != nil {
		return fmt.Errorf("Redis连接失败: %v", err)
	}

	log.Println("Redis连接成功")
	return nil
}

// CloseRedis 关闭Redis连接
func CloseRedis() {
	if RedisClient != nil {
		RedisClient.Close()
		log.Println("Redis连接已关闭")
	}
}

// AcquireLock 获取分布式锁
func AcquireLock(lockKey string, timeout time.Duration) (bool, error) {
	// 检查Redis客户端是否已初始化
	if RedisClient == nil {
		// 初始化Redis连接
		err := InitRedis("36.212.20.113", "j8nZ4jra2E", 7963, 13)
		if err != nil {
			return false, fmt.Errorf("Redis连接初始化失败: %v", err)
		}
	}
	
	// 尝试获取锁，设置过期时间防止死锁
	result, err := RedisClient.SetNX(ctx, lockKey, "1", timeout).Result()
	if err != nil {
		return false, fmt.Errorf("获取Redis锁失败: %v", err)
	}
	return result, nil
}

// ReleaseLock 释放分布式锁
func ReleaseLock(lockKey string) error {
	// 检查Redis客户端是否已初始化
	if RedisClient == nil {
		// 初始化Redis连接
		err := InitRedis("36.212.20.113", "j8nZ4jra2E", 7963, 13)
		if err != nil {
			return fmt.Errorf("Redis连接初始化失败: %v", err)
		}
	}
	
	_, err := RedisClient.Del(ctx, lockKey).Result()
	if err != nil {
		return fmt.Errorf("释放Redis锁失败: %v", err)
	}
	return nil
}