package cn.ac.iie.utils.redis;

import org.apache.log4j.Logger;
import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.JedisCluster;
import redis.clients.jedis.JedisPoolConfig;

import java.io.IOException;
import java.util.LinkedHashSet;
import java.util.Properties;
import java.util.Set;

/**
 * Redis连接池
 *
 * @author my
 * @date 2018-07-04
 */
public final class RedisPollUtils {
    private static final Logger logger = Logger.getLogger(RedisPollUtils.class);
    private static JedisCluster jedisCluster;
    private static Properties props = new Properties();

    /**
     * 不允许通过new创建该类的实例
     */
    private RedisPollUtils() {
    }

    /**
     * 初始化Redis连接池
     */
    public static JedisCluster getJedisCluster() {
        try {
            String redisConfigFile = "redis_aaa_config.properties";
            props.load(RedisPollUtils.class.getClassLoader().getResourceAsStream(redisConfigFile));
        } catch (IOException e) {
            logger.error("Properties Initialization Failed!!!!");
            e.printStackTrace();
        }
        JedisPoolConfig poolConfig = new JedisPoolConfig();
        poolConfig.setMaxTotal(Integer.valueOf(props.getProperty("jedis.pool.maxActive")));
        poolConfig.setMaxIdle(Integer.valueOf(props.getProperty("jedis.pool.maxIdle")));
        poolConfig.setMaxWaitMillis(Long.valueOf(props.getProperty("jedis.pool.maxWait")));
        poolConfig.setTestOnReturn(Boolean.valueOf(props.getProperty("jedis.pool.testOnReturn")));
        poolConfig.setTestOnBorrow(Boolean.valueOf(props.getProperty("jedis.pool.testOnBorrow")));
        Set<HostAndPort> nodes = new LinkedHashSet<HostAndPort>();
        for (String port : props.getProperty("redis.port").split(",")) {
            nodes.add(new HostAndPort(props.getProperty("redis.ip1"), Integer.parseInt(port)));
            nodes.add(new HostAndPort(props.getProperty("redis.ip2"), Integer.parseInt(port)));
            nodes.add(new HostAndPort(props.getProperty("redis.ip3"), Integer.parseInt(port)));
            nodes.add(new HostAndPort(props.getProperty("redis.ip4"), Integer.parseInt(port)));
            nodes.add(new HostAndPort(props.getProperty("redis.ip5"), Integer.parseInt(port)));
        }
        if (jedisCluster == null) {
            jedisCluster = new JedisCluster(nodes, poolConfig);
        }
        return jedisCluster;
    }

}