org.tinygroup.jedis.impl.JedisManagerImpl 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 org.tinygroup.jedis.JedisManager;
import org.tinygroup.jedis.config.JedisConfig;
import org.tinygroup.jedis.config.JedisConfigs;
import org.tinygroup.jedis.util.JedisUtil;

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

public class JedisManagerImpl implements JedisManager {

	private Map jedisConfigMap = new HashMap();
	private Map jedisPoolMap = new HashMap();

	public void addJedisConfigs(JedisConfigs configs) {
		for (JedisConfig config : configs.getJedisConfigList()) {
			addJedisConfig(config);
		}
	}

	public void removeJedisConfigs(JedisConfigs configs) {
		for (JedisConfig config : configs.getJedisConfigList()) {
			removeJedisConfig(config);
		}
	}

	public void addJedisConfig(JedisConfig config) {
		jedisConfigMap.put(config.getId(), config);
		jedisPoolMap.put(config.getId(), JedisUtil.createJedisPool(config,
				this.getClass().getClassLoader()));
	}

	public void removeJedisConfig(JedisConfig config) {
		jedisConfigMap.remove(config.getId());
		removeJedisPool(config.getId());
	}

	public JedisConfig getJedisConfig(String id) {
		return id == null ? null : jedisConfigMap.get(id);
	}

	public JedisPoolConfig getJedisPoolConfig(String id) {
		JedisConfig jedisConfig = getJedisConfig(id);
		return JedisUtil.getJedisPoolConfig(jedisConfig.getPoolConfig(), this
				.getClass().getClassLoader());
	}

	public JedisPool getJedisPool(String jedisId) {
		if (jedisPoolMap.containsKey(jedisId)) {
			return jedisPoolMap.get(jedisId);
		}
		return createJedisPool(jedisId);
	}

	private JedisPool removeJedisPool(String jedisId, JedisConfig config) {
		return jedisPoolMap.remove(jedisId);
	}

	public JedisPool removeJedisPool(String jedisId) {
		if (!jedisConfigMap.containsKey(jedisId)) {
			throw new RuntimeException("不存在JedisConfig:" + jedisId);
		}
		return removeJedisPool(jedisId, jedisConfigMap.get(jedisId));
	}

	public String removeJedisPool(JedisPool pool) {
		String jedisId = null;
		for (String id : jedisPoolMap.keySet()) {
			if (pool == jedisPoolMap.get(jedisId)) {
				jedisId = id;
				break;
			}
		}
		if (jedisId != null) {
			removeJedisPool(jedisId);
		}
		return jedisId;
	}

	/*
	 * private JedisPool createJedisPool(JedisConfig jedisConfig) { // 设置默认参数
	 * String host = StringUtil.isBlank(jedisConfig.getHost()) ?
	 * Protocol.DEFAULT_HOST : jedisConfig.getHost(); int port =
	 * jedisConfig.getPort() <= 0 ? Protocol.DEFAULT_PORT :
	 * jedisConfig.getPort(); int timeout = jedisConfig.getTimeout() < 0 ?
	 * Protocol.DEFAULT_TIMEOUT : jedisConfig.getTimeout(); int database =
	 * jedisConfig.getDatabase() < 0 ? Protocol.DEFAULT_DATABASE :
	 * jedisConfig.getDatabase(); // 实例化jedis连接池 JedisPool jedisPool = new
	 * JedisPool( getJedisPoolConfig(jedisConfig.getPoolConfig()), host, port,
	 * timeout, jedisConfig.getPassword(), database,
	 * jedisConfig.getClientName()); jedisPoolMap.put(jedisConfig.getId(),
	 * jedisPool); return jedisPool; }
	 */

	private JedisPool createJedisPool(String jedisId) {
		JedisConfig jedisConfig = getJedisConfig(jedisId);
		if (jedisConfig == null) {
			throw new NullPointerException(String.format(
					"根据Id:[%s]没有找到匹配的JedisConfig配置对象", jedisId));
		}
		return JedisUtil.createJedisPool(jedisConfig, this.getClass()
				.getClassLoader());

	}

}