Java实现Redis哨兵的示例代码

package com.middleware.redis.sentinels;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

import java.util.*;


public class JavaSentinel {

    // 主节点ip:端口    127.0.0.1:6379
    static String masterAddress = "127.0.0.1:6379";
    // 所有 slave
    static final Vector<String> slaveRedisServers = new Vector<String>();
    // 坏掉的实例
    static final Vector<String> badRedisServers = new Vector<String>();

    // 连接池对象
    static JedisPool jedisPool ;

    // 连接池配置信息对象
    private static JedisPoolConfig config = new JedisPoolConfig();

    
    static {

        // 最大连接数10
        config.setMaxTotal(10);
        //最大空闲连接数5
        config.setMaxIdle(5);

    }

    
    public Jedis newJedisInstance() {
        return jedisPool.getResource() ;
    }

    volatile static JavaSentinel javaSentinel;

    

    public static synchronized JavaSentinel getInstance(boolean isOpenSentinel){

        // 是否开启java哨兵
        if(isOpenSentinel){

            // 定时任务
            new Timer().schedule(new TimerTask() {
                @Override
                public void run() {
                    // 检测 master是否可以
                    checkMaster();
                    // 更新slave列表
                    updateSlaves();
                    // 检测坏掉的实例是否恢复正常
                    checkBadServer();

                }
            }, 1000, 3000);
        }

        if(null == javaSentinel){

            
            String[] serverInfo = masterAddress.split(":");
            String masterHost = serverInfo[0] ;
            int masterPort = Integer.parseInt(serverInfo[1]) ;
            jedisPool = new JedisPool(config, masterHost, masterPort, 100000);

            //初始化当前类对象
            javaSentinel = new JavaSentinel();
        }

        return javaSentinel;

    }


    
    private static void checkMaster() {
        // 主从切换
        // 检查状态
        System.out.println("检查master状态:" + masterAddress);
        String masterHost = masterAddress.split(":")[0];
        int masterPort = Integer.parseInt(masterAddress.split(":")[1]);
        try {
            Jedis jedis = new Jedis(masterHost, masterPort);
            jedis.ping();
            jedis.close();
        } catch (Exception e) {
            // master挂掉啦
            badRedisServers.add(masterAddress);
            // 切换master
            changeMaster();
        }
    }

    
    private static void changeMaster() {
        Iterator<String> iterator = slaveRedisServers.iterator();
        while (iterator.hasNext()) {
            String slaveAddress = iterator.next();
            try {
                String slaveHost = slaveAddress.split(":")[0];
                int slavePort = Integer.parseInt(slaveAddress.split(":")[1]);
                Jedis jedis = new Jedis(slaveHost, slavePort);

                
                jedis.ping();

                
                jedis.slaveofNoOne();
                jedis.close();
                masterAddress = slaveAddress;
                System.out.println("产生新的master:" + masterAddress);
                break;
            } catch (Exception e) {
                badRedisServers.add(slaveAddress);
            } finally {
                iterator.remove();
            }
        }

        // 所有slave切到新的master
        for (String slave : slaveRedisServers) {
            String slaveHost = slave.split(":")[0];
            int slavePort = Integer.parseInt(slave.split(":")[1]);
            Jedis jedis = new Jedis(slaveHost, slavePort);
            jedis.slaveof(masterAddress.split(":")[0], Integer.parseInt(masterAddress.split(":")[1]));
            jedis.close();
        }
    }

    
    private static void updateSlaves() {
        // 获取所有slave
        try {
            String masterHost = masterAddress.split(":")[0];
            int masterPort = Integer.parseInt(masterAddress.split(":")[1]);
            Jedis jedis = new Jedis(masterHost, masterPort);
            String info_replication = jedis.info("replication");
            // 解析info replication
            String[] lines = info_replication.split("\r\n");
            int slaveCount = Integer.parseInt(lines[2].split(":")[1]);
            if (slaveCount > 0) {
                slaveRedisServers.clear();
                for (int i = 0; i < slaveCount; i++) {
                    String host = lines[3 + i].split(",")[0].split("=")[1];
                    String port = lines[3 + i].split(",")[1].split("=")[1];
                    slaveRedisServers.add(host + ":" + port);
                }
            }
            System.out.println("更新slave列表:" + Arrays.toString(slaveRedisServers.toArray(new String[] {})));
            jedis.close();
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("更新slave失败:" + e.getMessage());
        }
    }


    
    private static void checkBadServer() {
        // 获取所有slave
        Iterator<String> iterator = badRedisServers.iterator();
        while (iterator.hasNext()) {
            String bad = iterator.next();
            try {
                String badHost = bad.split(":")[0];
                int badPort = Integer.parseInt(bad.split(":")[1]);
                Jedis badServer = new Jedis(badHost, badPort);
                badServer.ping();

                // 如果ping没有问题，则挂在当前的master
                badServer.slaveof(masterAddress.split(":")[0], Integer.parseInt(masterAddress.split(":")[1]));
                badServer.close();

                slaveRedisServers.add(bad);
                iterator.remove();
                System.out.println(bad + " 恢复正常，当前master:" + masterAddress);
            } catch (Exception e) {
            }
        }
    }
}