rik/jlibredis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor redis cluster connection handler and add JedisAskDataException handling

Browse Source
This commit is contained in:
Marcos Nils
2013-12-20 15:28:32 -03:00
parent c008070045
commit 0bc27ac3a2
6 changed files with 66 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/main/java/redis/clients/jedis/JedisCluster.java
View File

@@ -15,7 +15,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
private JedisClusterConnectionHandler connectionHandler;
public JedisCluster(Set<HostAndPort> nodes, int timeout) {
connectionHandler = new JedisClusterConnectionHandler(nodes);
connectionHandler = new JedisRandomConnectionHandler(nodes);
}
@@ -31,7 +31,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<String>(connectionHandler) {
@Override
public String execute() {
return connectionHandler.getRandomConnection().set(key, value);
return connectionHandler.getConnection().set(key, value);
}
}.run();
}
@@ -41,7 +41,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<String>(connectionHandler) {
@Override
public String execute() {
return connectionHandler.getRandomConnection().get(key);
return connectionHandler.getConnection().get(key);
}
}.run();
}
@@ -51,7 +51,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Boolean>(connectionHandler) {
@Override
public Boolean execute() {
return connectionHandler.getRandomConnection().exists(key);
return connectionHandler.getConnection().exists(key);
}
}.run();
}
@@ -61,7 +61,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Long>(connectionHandler) {
@Override
public Long execute() {
return connectionHandler.getRandomConnection().persist(key);
return connectionHandler.getConnection().persist(key);
}
}.run();
}
@@ -71,7 +71,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<String>(connectionHandler) {
@Override
public String execute() {
return connectionHandler.getRandomConnection().type(key);
return connectionHandler.getConnection().type(key);
}
}.run();
}
@@ -81,7 +81,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Long>(connectionHandler) {
@Override
public Long execute() {
return connectionHandler.getRandomConnection().expire(key, seconds);
return connectionHandler.getConnection().expire(key, seconds);
}
}.run();
}
@@ -91,7 +91,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Long>(connectionHandler) {
@Override
public Long execute() {
return connectionHandler.getRandomConnection().expireAt(key, unixTime);
return connectionHandler.getConnection().expireAt(key, unixTime);
}
}.run();
}
@@ -101,7 +101,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Long>(connectionHandler) {
@Override
public Long execute() {
return connectionHandler.getRandomConnection().ttl(key);
return connectionHandler.getConnection().ttl(key);
}
}.run();
}
@@ -111,7 +111,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Boolean>(connectionHandler) {
@Override
public Boolean execute() {
return connectionHandler.getRandomConnection().setbit(key, offset, value);
return connectionHandler.getConnection().setbit(key, offset, value);
}
}.run();
}
@@ -121,7 +121,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Boolean>(connectionHandler) {
@Override
public Boolean execute() {
return connectionHandler.getRandomConnection().setbit(key, offset, value);
return connectionHandler.getConnection().setbit(key, offset, value);
}
}.run();
}
@@ -131,7 +131,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Boolean>(connectionHandler) {
@Override
public Boolean execute() {
return connectionHandler.getRandomConnection().getbit(key, offset);
return connectionHandler.getConnection().getbit(key, offset);
}
}.run();
}
@@ -141,7 +141,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<Long>(connectionHandler) {
@Override
public Long execute() {
return connectionHandler.getRandomConnection().setrange(key, offset, value);
return connectionHandler.getConnection().setrange(key, offset, value);
}
}.run();
}
@@ -151,7 +151,7 @@ public class JedisCluster implements JedisCommands, BasicCommands {
return new JedisClusterCommand<String>(connectionHandler) {
@Override
public String execute() {
return connectionHandler.getRandomConnection().getrange(key, startOffset, endOffset);
return connectionHandler.getConnection().getrange(key, startOffset, endOffset);
}
}.run();
}

29
src/main/java/redis/clients/jedis/JedisClusterConnectionHandler.java
View File

@@ -1,33 +1,8 @@
package redis.clients.jedis;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
import java.util.Set;
import redis.clients.util.Pool;
public class JedisClusterConnectionHandler {
private Map<String, JedisPool> nodes = new HashMap<String, JedisPool>();
public interface JedisClusterConnectionHandler {
public JedisClusterConnectionHandler(Set<HostAndPort> nodes) {
initializeSlotsCache(nodes);
}
private void initializeSlotsCache(Set<HostAndPort> nodes) {
for (HostAndPort hostAndPort : nodes) {
JedisPool jp = new JedisPool(hostAndPort.getHost(),
hostAndPort.getPort());
this.nodes.put(hostAndPort.getHost() + hostAndPort.getPort(), jp);
}
}
@SuppressWarnings("unchecked")
public Jedis getRandomConnection() {
Object[] nodeArray = nodes.values().toArray();
return ((Pool<Jedis>) nodeArray[new Random().nextInt(nodeArray.length)]).getResource();
}
Jedis getConnection();
}

33
src/main/java/redis/clients/jedis/JedisRandomConnectionHandler.java Normal file
View File

@@ -0,0 +1,33 @@
package redis.clients.jedis;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
import java.util.Set;
import redis.clients.util.Pool;
public class JedisRandomConnectionHandler implements JedisClusterConnectionHandler {
private Map<String, JedisPool> nodes = new HashMap<String, JedisPool>();
public JedisRandomConnectionHandler(Set<HostAndPort> nodes) {
initializeSlotsCache(nodes);
}
private void initializeSlotsCache(Set<HostAndPort> nodes) {
for (HostAndPort hostAndPort : nodes) {
JedisPool jp = new JedisPool(hostAndPort.getHost(),
hostAndPort.getPort());
this.nodes.put(hostAndPort.getHost() + hostAndPort.getPort(), jp);
}
}
@SuppressWarnings("unchecked")
public Jedis getConnection() {
Object[] nodeArray = nodes.values().toArray();
return ((Pool<Jedis>) nodeArray[new Random().nextInt(nodeArray.length)]).getResource();
}
}

7
src/main/java/redis/clients/jedis/Protocol.java
View File

@@ -4,6 +4,7 @@ import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import redis.clients.jedis.exceptions.JedisAskDataException;
import redis.clients.jedis.exceptions.JedisConnectionException;
import redis.clients.jedis.exceptions.JedisDataException;
import redis.clients.jedis.exceptions.JedisMovedDataException;
@@ -13,6 +14,7 @@ import redis.clients.util.SafeEncoder;
public final class Protocol {
private static final String ASK_RESPONSE = "ASK";
private static final String MOVED_RESPONSE = "MOVED";
public static final int DEFAULT_PORT = 6379;
public static final int DEFAULT_SENTINEL_PORT = 26379;
@@ -75,9 +77,12 @@ public final class Protocol {
private static void processError(final RedisInputStream is) {
String message = is.readLine();
//TODO: Read only first 5 bytes?
//TODO: I'm not sure if this is the best way to do this.
//Maybe Read only first 5 bytes instead?
if (message.contains(MOVED_RESPONSE)) {
throw new JedisMovedDataException(message);
} else if (message.contains(ASK_RESPONSE)) {
throw new JedisAskDataException(message);
}
throw new JedisDataException(message);
}

15
src/test/java/redis/clients/jedis/tests/JedisClusterTest.java
View File

@@ -77,14 +77,13 @@ public class JedisClusterTest extends Assert {
node2.get("foo");
}
// @Test(expected=JedisAskDataException.class)
// public void ask() {
// node1.set("foo", "bar");
// int keySlot = RedisSlot.getSlot("foo");
// String node2Id = getNodeId(node2.clusterNodes());
// node1.clusterSetSlotMigrating(keySlot, node2Id);
// node1.get("foo");
// }
@Test(expected=JedisAskDataException.class)
public void ask() {
int keySlot = RedisSlot.getSlot("test");
String node2Id = getNodeId(node2.clusterNodes());
node1.clusterSetSlotMigrating(keySlot, node2Id);
node1.get("test");
}
private String getNodeId(String infoOutput) {
for (String infoLine : infoOutput.split("\n")) {

4
src/test/java/redis/clients/jedis/tests/utils/RedisSlot.java
View File

@@ -22,4 +22,8 @@ public class RedisSlot {
return crc &= 0xffff % 16384;
}
public static void main(String[] args) {
System.out.println(getSlot("test"));
}
}