236 lines
7.0 KiB
Java
236 lines
7.0 KiB
Java
/**
|
||
* Copyright: Copyright (c) 2015
|
||
*
|
||
* @author youaremoon
|
||
* @date 2016年6月25日
|
||
* @version V1.0
|
||
*/
|
||
package com.nis.util.redis;
|
||
|
||
import java.io.Closeable;
|
||
import java.lang.reflect.Field;
|
||
import java.util.ArrayList;
|
||
import java.util.HashMap;
|
||
import java.util.HashSet;
|
||
import java.util.LinkedList;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
import java.util.Queue;
|
||
|
||
import redis.clients.jedis.BinaryJedisCluster;
|
||
import redis.clients.jedis.Client;
|
||
import redis.clients.jedis.Jedis;
|
||
import redis.clients.jedis.JedisCluster;
|
||
import redis.clients.jedis.JedisClusterConnectionHandler;
|
||
import redis.clients.jedis.JedisClusterInfoCache;
|
||
import redis.clients.jedis.JedisPool;
|
||
import redis.clients.jedis.JedisSlotBasedConnectionHandler;
|
||
import redis.clients.jedis.PipelineBase;
|
||
import redis.clients.jedis.exceptions.JedisMovedDataException;
|
||
import redis.clients.jedis.exceptions.JedisRedirectionException;
|
||
import redis.clients.util.JedisClusterCRC16;
|
||
import redis.clients.util.SafeEncoder;
|
||
|
||
/**
|
||
* 在集群模式下提供批量操作的功能。 <br/>
|
||
* 由于集群模式存在节点的动态添加删除，且client不能实时感知（只有在执行命令时才可能知道集群发生变更），
|
||
* 因此，该实现不保证一定成功，建议在批量操作之前调用 refreshCluster() 方法重新获取集群信息。<br />
|
||
* 应用需要保证不论成功还是失败都会调用close() 方法，否则可能会造成泄露。<br/>
|
||
* 如果失败需要应用自己去重试，因此每个批次执行的命令数量需要控制。防止失败后重试的数量过多。<br />
|
||
* 基于以上说明，建议在集群环境较稳定（增减节点不会过于频繁）的情况下使用，且允许失败或有对应的重试策略。<br />
|
||
*
|
||
* 该类非线程安全
|
||
*
|
||
* @author youaremoon
|
||
* @version
|
||
* @since Ver 1.1
|
||
*/
|
||
public class JedisClusterPipeline extends PipelineBase implements Closeable {
|
||
// private static final Logger LOGGER = LoggerFactory.getLogger(JedisClusterPipeline.class);
|
||
|
||
// 部分字段没有对应的获取方法，只能采用反射来做
|
||
// 你也可以去继承JedisCluster和JedisSlotBasedConnectionHandler来提供访问接口
|
||
private static final Field FIELD_CONNECTION_HANDLER;
|
||
private static final Field FIELD_CACHE;
|
||
static {
|
||
FIELD_CONNECTION_HANDLER = getField(BinaryJedisCluster.class, "connectionHandler");
|
||
FIELD_CACHE = getField(JedisClusterConnectionHandler.class, "cache");
|
||
}
|
||
|
||
private JedisSlotBasedConnectionHandler connectionHandler;
|
||
private JedisClusterInfoCache clusterInfoCache;
|
||
private Queue<Client> clients = new LinkedList<Client>(); // 根据顺序存储每个命令对应的Client
|
||
private Map<JedisPool, Jedis> jedisMap = new HashMap<>(); // 用于缓存连接
|
||
private boolean hasDataInBuf = false; // 是否有数据在缓存区
|
||
|
||
/**
|
||
* 根据jedisCluster实例生成对应的JedisClusterPipeline
|
||
* @param
|
||
* @return
|
||
*/
|
||
public static JedisClusterPipeline pipelined(JedisCluster jedisCluster) {
|
||
JedisClusterPipeline pipeline = new JedisClusterPipeline();
|
||
pipeline.setJedisCluster(jedisCluster);
|
||
return pipeline;
|
||
}
|
||
|
||
public JedisClusterPipeline() {
|
||
}
|
||
|
||
public void setJedisCluster(JedisCluster jedis) {
|
||
connectionHandler = getValue(jedis, FIELD_CONNECTION_HANDLER);
|
||
clusterInfoCache = getValue(connectionHandler, FIELD_CACHE);
|
||
}
|
||
|
||
/**
|
||
* 刷新集群信息，当集群信息发生变更时调用
|
||
* @param
|
||
* @return
|
||
*/
|
||
public void refreshCluster() {
|
||
connectionHandler.renewSlotCache();
|
||
}
|
||
|
||
/**
|
||
* 同步读取所有数据. 与syncAndReturnAll()相比，sync()只是没有对数据做反序列化
|
||
*/
|
||
public void sync() {
|
||
innerSync(null);
|
||
}
|
||
|
||
/**
|
||
* 同步读取所有数据 并按命令顺序返回一个列表
|
||
*
|
||
* @return 按照命令的顺序返回所有的数据
|
||
*/
|
||
public List<Object> syncAndReturnAll() {
|
||
List<Object> responseList = new ArrayList<Object>();
|
||
|
||
innerSync(responseList);
|
||
|
||
return responseList;
|
||
}
|
||
|
||
private void innerSync(List<Object> formatted) {
|
||
HashSet<Client> clientSet = new HashSet<Client>();
|
||
|
||
try {
|
||
for (Client client : clients) {
|
||
// 在sync()调用时其实是不需要解析结果数据的，但是如果不调用get方法，发生了JedisMovedDataException这样的错误应用是不知道的，因此需要调用get()来触发错误。
|
||
// 其实如果Response的data属性可以直接获取，可以省掉解析数据的时间，然而它并没有提供对应方法，要获取data属性就得用反射，不想再反射了，所以就这样了
|
||
Object data = generateResponse(client.getOne()).get();
|
||
if (null != formatted) {
|
||
formatted.add(data);
|
||
}
|
||
|
||
// size相同说明所有的client都已经添加，就不用再调用add方法了
|
||
if (clientSet.size() != jedisMap.size()) {
|
||
clientSet.add(client);
|
||
}
|
||
}
|
||
} catch (JedisRedirectionException jre) {
|
||
if (jre instanceof JedisMovedDataException) {
|
||
// if MOVED redirection occurred, rebuilds cluster's slot cache,
|
||
// recommended by Redis cluster specification
|
||
refreshCluster();
|
||
}
|
||
|
||
throw jre;
|
||
} finally {
|
||
if (clientSet.size() != jedisMap.size()) {
|
||
// 所有还没有执行过的client要保证执行(flush)，防止放回连接池后后面的命令被污染
|
||
for (Jedis jedis : jedisMap.values()) {
|
||
if (clientSet.contains(jedis.getClient())) {
|
||
continue;
|
||
}
|
||
|
||
flushCachedData(jedis);
|
||
}
|
||
}
|
||
|
||
hasDataInBuf = false;
|
||
close();
|
||
}
|
||
}
|
||
|
||
@Override
|
||
public void close() {
|
||
clean();
|
||
|
||
clients.clear();
|
||
|
||
for (Jedis jedis : jedisMap.values()) {
|
||
if (hasDataInBuf) {
|
||
flushCachedData(jedis);
|
||
}
|
||
|
||
jedis.close();
|
||
}
|
||
|
||
jedisMap.clear();
|
||
|
||
hasDataInBuf = false;
|
||
}
|
||
|
||
private void flushCachedData(Jedis jedis) {
|
||
try {
|
||
jedis.getClient().getAll();
|
||
} catch (RuntimeException ex) {
|
||
// 其中一个client出问题，后面出问题的几率较大
|
||
}
|
||
}
|
||
|
||
@Override
|
||
protected Client getClient(String key) {
|
||
byte[] bKey = SafeEncoder.encode(key);
|
||
|
||
return getClient(bKey);
|
||
}
|
||
|
||
@Override
|
||
protected Client getClient(byte[] key) {
|
||
Jedis jedis = getJedis(JedisClusterCRC16.getSlot(key));
|
||
|
||
Client client = jedis.getClient();
|
||
clients.add(client);
|
||
|
||
return client;
|
||
}
|
||
|
||
private Jedis getJedis(int slot) {
|
||
JedisPool pool = clusterInfoCache.getSlotPool(slot);
|
||
|
||
// 根据pool从缓存中获取Jedis
|
||
Jedis jedis = jedisMap.get(pool);
|
||
if (null == jedis) {
|
||
jedis = pool.getResource();
|
||
jedisMap.put(pool, jedis);
|
||
}
|
||
|
||
hasDataInBuf = true;
|
||
return jedis;
|
||
}
|
||
|
||
private static Field getField(Class<?> cls, String fieldName) {
|
||
try {
|
||
Field field = cls.getDeclaredField(fieldName);
|
||
field.setAccessible(true);
|
||
|
||
return field;
|
||
} catch (NoSuchFieldException | SecurityException e) {
|
||
throw new RuntimeException("cannot find or access field '" + fieldName + "' from " + cls.getName(), e);
|
||
}
|
||
}
|
||
|
||
@SuppressWarnings({"unchecked" })
|
||
private static <T> T getValue(Object obj, Field field) {
|
||
try {
|
||
return (T)field.get(obj);
|
||
} catch (IllegalArgumentException | IllegalAccessException e) {
|
||
// LOGGER.error("get value fail", e);
|
||
|
||
throw new RuntimeException(e);
|
||
}
|
||
}
|
||
}
|