org.tinygroup.jedis.impl.JedisSentinelClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.jedis Show documentation
There is a newer version: 2.2.3
package org.tinygroup.jedis.impl;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.tinygroup.commons.tools.StringUtil;
import org.tinygroup.jedis.config.JedisConfig;
import org.tinygroup.jedis.config.JedisSentinelConfig;
import org.tinygroup.jedis.util.JedisUtil;
import org.tinygroup.logger.Logger;
import org.tinygroup.logger.LoggerFactory;

import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.JedisSentinelPool;
import redis.clients.util.Pool;

public class JedisSentinelClient {
	// private List jedisPools = new ArrayList();
	// private List poolKeys = new ArrayList();
	private Map jedisPoolsMap = new HashMap();
	Map jedisPoolsHostAndPortMap = new HashMap();
	private JedisSentinelPool jedisSentinelPool;
	private String masterName;
	private static final Logger logger = LoggerFactory
			.getLogger(JedisSentinelClient2.class);

	public JedisSentinelClient(JedisSentinelConfig config) {
		masterName = config.getMasterName();
		initJedisSentinelPool(config);
		initJedisPools(config);
	}

	private void initJedisPools(JedisSentinelConfig config) {
		List configs = config.getJedisConfigList();
		for (JedisConfig jedisConfig : configs) {
			if (!jedisPoolsMap.containsKey(jedisConfig.getId())) {
				JedisPool pool = JedisUtil.createJedisPool(jedisConfig, this
						.getClass().getClassLoader());
				jedisPoolsMap.put(jedisConfig.getId(), pool);

				jedisPoolsHostAndPortMap.put(jedisConfig.getId(),
						getHostAndPort(jedisConfig));
			}

		}
	}

	private HostAndPort getHostAndPort(JedisConfig config) {
		return new HostAndPort(config.getHost(), config.getPort());
	}

	private void initJedisSentinelPool(JedisSentinelConfig config) {
		String masterName = config.getMasterName();
		if (StringUtil.isBlank(masterName)) {
			throw new RuntimeException("JedisSentinelPool配置的masterName不可为空."
					+ config.toString());
		}
		String sentinels = config.getSentinels();
		if (StringUtil.isBlank(sentinels)) {
			throw new RuntimeException("JedisSentinelPool配置的sentinels不可为空."
					+ config.toString());
		}
		Set sentinelSet = getSentinelSet(sentinels);
		int database = config.getDatabase();
		int timeout = config.getTimeout();
		String password = config.getPassword();
		String poolConfig = config.getPoolConfig();
		JedisPoolConfig jedisPoolConfig = JedisUtil.getJedisPoolConfig(
				poolConfig, this.getClass().getClassLoader());
		jedisSentinelPool = new JedisSentinelPool(masterName, sentinelSet,
				jedisPoolConfig, timeout, password, database);
	}

	private Set getSentinelSet(String sentinels) {
		String[] sentinelArray = sentinels.split(",");
		Map map = new HashMap();
		for (String sentinel : sentinelArray) {
			map.put(sentinel, sentinel);
		}
		return map.keySet();
	}

	public void destroy() {
		destroy(jedisSentinelPool);
		for (JedisPool pool : jedisPoolsMap.values()) {
			destroy(pool);
		}
		jedisPoolsMap.clear();
	}

	private void destroy(Pool pool) {
		try {
			pool.destroy();
		} catch (Exception e) {
			logger.errorMessage("销毁{}的连接池{}时出错", e, masterName,
					pool.toString());
		}

	}

	public JedisSentinelPool getJedisSentinelPool() {
		return jedisSentinelPool;
	}

	public Jedis getWriteJedis() {
		return getJedisSentinelPool().getResource();
	}

	public Jedis getReadJedis() {
		return getReadJedisPool().getResource();
	}

	public JedisPool getReadJedisPool() {
		// get from jedisPoolsMap.values() except
		// jedisSentinelPool.getCurrentHostMaster()
		return getReadJedisPool(jedisSentinelPool.getCurrentHostMaster());
	}

	private JedisPool getReadJedisPool(HostAndPort except) {
		String key = getJedisPool(jedisPoolsMap);
		if(except.toString().equals(jedisPoolsHostAndPortMap.get(key).toString())){
			return getReadJedisPool(except);//这里要考虑读写共用一个的情况，即判断列表可用性是否只有1个
		}
		
		return jedisPoolsMap.get(key);
	}

	private String getJedisPool(Map pools) {
		for(String s : pools.keySet()){
			return s;
		}
		throw new RuntimeException("未找到合适的jedisPool");
	}

}