org.tinygroup.jedis.impl.JedisSentinelClient2 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.Map;
import java.util.Set;

import org.tinygroup.commons.tools.StringUtil;
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 JedisSentinelClient2 {
	// private List jedisPools = new ArrayList();
	// private List poolKeys = new ArrayList();
	private JedisPool readJedisPool;
	Map jedisPoolsHostAndPortMap = new HashMap();
	private JedisSentinelPool jedisSentinelPool;
	private String masterName;
	private static final Logger logger = LoggerFactory
			.getLogger(JedisSentinelClient.class);

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

	private void initJedisPools(JedisSentinelConfig config) {
		readJedisPool = JedisUtil.createJedisPool(config.getReadJedisConfig(),
				this.getClass().getClassLoader());

	}

	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);
		destroy(readJedisPool);
	}

	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() {
		return readJedisPool;
	}

	

}