org.tinygroup.jedis.shard.JedisCheck 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.shard;

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

import org.tinygroup.jedis.util.JedisUtil;
import org.tinygroup.logger.LogLevel;
import org.tinygroup.logger.Logger;
import org.tinygroup.logger.LoggerFactory;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisShardInfo;

public class JedisCheck {
	private final static Logger LOGGER = LoggerFactory
			.getLogger(JedisCheck.class);
	private static Map> failReadMap = new HashMap>();
	private static FailOverThread failTestThread = new FailOverThread();
	private static boolean flag = false;
	
	public static synchronized void start(){
		if(flag){
			return;
		}
		flag = true;
		failTestThread.setDaemon(true);
		failTestThread.start();
	}
	
	public static List getFailList(JedisShardInfo shardInfo){
		List failList = new ArrayList();
		String infoString = getDesString(shardInfo);
		if (failReadMap.containsKey(infoString)) {
			failList = failReadMap.get(infoString);
		} else {
			failReadMap.put(infoString, failList);
		}
		return failList;
	}
	
	private static String getDesString(JedisShardInfo shardInfo){
		return shardInfo.getName()+"_"+shardInfo.getHost()+"_"+shardInfo.getPort()+"_"+shardInfo.getDb();
	}
	
	/**
	 * 轮询失败的服务器
	 * 
	 */
	static class FailOverThread extends Thread {
		boolean stop = false;
		
		public void stopTry(){
			stop = true;
		}
		public void run() {
			while (!stop) {
				try {
					sleep(JedisUtil.getFailOverTime());
				} catch (Exception e) {
				}
				testFailJedisMap();
			}
		}
	}

	private static void testFailJedisMap() {
		LOGGER.logMessage(LogLevel.DEBUG, "开始轮询连接失败的服务器列表");
		for (List list : failReadMap.values()) {
			List newList = JedisUtil.copy(list);
			for (Jedis j : newList) {
				LOGGER.logMessage(LogLevel.DEBUG, "开始尝试连接的服务器:{}:{}", j
						.getClient().getHost(), j.getClient().getPort());
				boolean sucess = testFailJedis(j);
				if (sucess) {
					LOGGER.logMessage(LogLevel.DEBUG, "连接成功,从fail列表中删除");
					list.remove(j);
				}
				LOGGER.logMessage(LogLevel.DEBUG, "连接服务器:{}:{}结束", j
						.getClient().getHost(), j.getClient().getPort());
			}
		}
		LOGGER.logMessage(LogLevel.DEBUG, "轮询连接失败的服务器列表结束");
	}

	private static boolean testFailJedis(Jedis j) {

		try {
			j.disconnect();
			j.connect();
			if (j.ping().equals("PONG")) {
				return true;
			}
		} catch (Exception e) {
		}
		return false;
	}
	
	public static void stop(){
		failTestThread.stopTry();
	}
}