com.jeesuite.cache.redis.sentinel.JedisSentinelProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jeesuite-cache Show documentation
There is a newer version: 1.4.0
/**
 * 
 */
package com.jeesuite.cache.redis.sentinel;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.jeesuite.cache.redis.JedisProvider;

import redis.clients.jedis.BinaryJedis;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.JedisSentinelPool;
import redis.clients.jedis.exceptions.JedisDataException;
import redis.clients.jedis.exceptions.JedisException;

/**
 * redis哨兵主从模式服务提供者
 * @description 

 * @author vakin
 * @date 2015年04月26日
 */
public class JedisSentinelProvider implements JedisProvider{
	

	private static final String SLAVE_CHEKER_KEY = "_slave_cheker";
	private static final String SLAVE_CHEKER_VAL = "1";

	protected static final Logger logger = LoggerFactory.getLogger(JedisSentinelProvider.class);

	
	public static final String MODE = "sentinel";

	private ThreadLocal context = new ThreadLocal<>();
	
	private JedisSentinelPool jedisPool;
	
	private String groupName;
	
	private ScheduledExecutorService failoverCheker;
	

	public JedisSentinelProvider(final String groupName,final JedisPoolConfig jedisPoolConfig, String[] servers, final int timeout, final String password, final int database, final String clientName, final String masterName) {
		super();
		this.groupName = groupName;
		final Set sentinels = new HashSet(Arrays.asList(servers));
		
		jedisPool = new JedisSentinelPool(masterName, sentinels, jedisPoolConfig, timeout, password, database,clientName);
	
		failoverCheker = Executors.newScheduledThreadPool(1);
		failoverCheker.scheduleWithFixedDelay(new Runnable() {
			@Override
			public void run() {
				Jedis jedis = null;
				try {					
					jedis = jedisPool.getResource();
					jedis.set(SLAVE_CHEKER_KEY,SLAVE_CHEKER_VAL);
				} catch (Exception e) {
					if(e instanceof JedisDataException && e.getMessage().contains("READONLY")){
						logger.warn("JedisDataException happend error:{} and will re-init jedisPool" ,e.getMessage());
						//重新初始化jedisPool
						synchronized (jedisPool) {							
							jedisPool.destroy();
							jedisPool = new JedisSentinelPool(masterName, sentinels, jedisPoolConfig, timeout, password, database,clientName);
							logger.info("jedisPool re-init ok,currentHostMaster is:{}:{}" ,jedisPool.getCurrentHostMaster().getHost(),jedisPool.getCurrentHostMaster().getPort());
						}
					}
				}finally {
					try {jedis.close();} catch (Exception e2) {}
				}
			}
		}, 1, 1, TimeUnit.MINUTES);
	}

	public Jedis get() throws JedisException {
        Jedis jedis = context.get();
        if(jedis != null)return jedis;
        try {
            jedis = jedisPool.getResource();
        } catch (JedisException e) {
            if(jedis!=null){
            	jedis.close();
            }
            throw e;
        }
        context.set(jedis);
        if(logger.isTraceEnabled()){
        	logger.trace(">>get a redis conn[{}],Host:{}",jedis.toString(),jedis.getClient().getHost());
        }
        return jedis;
    }
 
	@Override
	public BinaryJedis getBinary() {	
		return get();
	}
	
	public void release() {
		Jedis jedis = context.get();
        if (jedis != null) {
        	context.remove();
        	jedis.close();
        	if(logger.isTraceEnabled()){
            	logger.trace("<