Added sharding using ketama

2010-09-14 01:30:59 -03:00
parent f9e7887e02
commit baab2225f6
5 changed files with 610 additions and 0 deletions
--- a/src/main/java/redis/clients/jedis/ShardedJedis.java
+++ b/src/main/java/redis/clients/jedis/ShardedJedis.java
@@ -0,0 +1,356 @@
+package redis.clients.jedis;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
+import redis.clients.util.ShardInfo;
+import redis.clients.util.Sharded;
+
+public class ShardedJedis extends Sharded<Jedis> {
+    public ShardedJedis(List<ShardInfo> shards) {
+	super(shards);
+    }
+
+    public String set(String key, String value) {
+	Jedis j = getShard(key);
+	return j.set(key, value);
+    }
+
+    public String get(String key) {
+	Jedis j = getShard(key);
+	return j.get(key);
+    }
+
+    public int exists(String key) {
+	Jedis j = getShard(key);
+	return j.exists(key);
+    }
+
+    public String type(String key) {
+	Jedis j = getShard(key);
+	return j.type(key);
+    }
+
+    public int expire(String key, int seconds) {
+	Jedis j = getShard(key);
+	return j.expire(key, seconds);
+    }
+
+    public int expireAt(String key, long unixTime) {
+	Jedis j = getShard(key);
+	return j.expireAt(key, unixTime);
+    }
+
+    public int ttl(String key) {
+	Jedis j = getShard(key);
+	return j.ttl(key);
+    }
+
+    public String getSet(String key, String value) {
+	Jedis j = getShard(key);
+	return j.getSet(key, value);
+    }
+
+    public int setnx(String key, String value) {
+	Jedis j = getShard(key);
+	return j.setnx(key, value);
+    }
+
+    public String setex(String key, int seconds, String value) {
+	Jedis j = getShard(key);
+	return j.setex(key, seconds, value);
+    }
+
+    public int decrBy(String key, int integer) {
+	Jedis j = getShard(key);
+	return j.decrBy(key, integer);
+    }
+
+    public int decr(String key) {
+	Jedis j = getShard(key);
+	return j.decr(key);
+    }
+
+    public int incrBy(String key, int integer) {
+	Jedis j = getShard(key);
+	return j.incrBy(key, integer);
+    }
+
+    public int incr(String key) {
+	Jedis j = getShard(key);
+	return j.incr(key);
+    }
+
+    public int append(String key, String value) {
+	Jedis j = getShard(key);
+	return j.append(key, value);
+    }
+
+    public String substr(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.substr(key, start, end);
+    }
+
+    public int hset(String key, String field, String value) {
+	Jedis j = getShard(key);
+	return j.hset(key, field, value);
+    }
+
+    public String hget(String key, String field) {
+	Jedis j = getShard(key);
+	return j.hget(key, field);
+    }
+
+    public int hsetnx(String key, String field, String value) {
+	Jedis j = getShard(key);
+	return j.hsetnx(key, field, value);
+    }
+
+    public String hmset(String key, Map<String, String> hash) {
+	Jedis j = getShard(key);
+	return j.hmset(key, hash);
+    }
+
+    public List<String> hmget(String key, String... fields) {
+	Jedis j = getShard(key);
+	return j.hmget(key, fields);
+    }
+
+    public int hincrBy(String key, String field, int value) {
+	Jedis j = getShard(key);
+	return j.hincrBy(key, field, value);
+    }
+
+    public int hexists(String key, String field) {
+	Jedis j = getShard(key);
+	return j.hexists(key, field);
+    }
+
+    public int hdel(String key, String field) {
+	Jedis j = getShard(key);
+	return j.hdel(key, field);
+    }
+
+    public int hlen(String key) {
+	Jedis j = getShard(key);
+	return j.hlen(key);
+    }
+
+    public List<String> hkeys(String key) {
+	Jedis j = getShard(key);
+	return j.hkeys(key);
+    }
+
+    public List<String> hvals(String key) {
+	Jedis j = getShard(key);
+	return j.hvals(key);
+    }
+
+    public Map<String, String> hgetAll(String key) {
+	Jedis j = getShard(key);
+	return j.hgetAll(key);
+    }
+
+    public int rpush(String key, String string) {
+	Jedis j = getShard(key);
+	return j.rpush(key, string);
+    }
+
+    public int lpush(String key, String string) {
+	Jedis j = getShard(key);
+	return j.lpush(key, string);
+    }
+
+    public int llen(String key) {
+	Jedis j = getShard(key);
+	return j.llen(key);
+    }
+
+    public List<String> lrange(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.lrange(key, start, end);
+    }
+
+    public String ltrim(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.ltrim(key, start, end);
+    }
+
+    public String lindex(String key, int index) {
+	Jedis j = getShard(key);
+	return j.lindex(key, index);
+    }
+
+    public String lset(String key, int index, String value) {
+	Jedis j = getShard(key);
+	return j.lset(key, index, value);
+    }
+
+    public int lrem(String key, int count, String value) {
+	Jedis j = getShard(key);
+	return j.lrem(key, count, value);
+    }
+
+    public String lpop(String key) {
+	Jedis j = getShard(key);
+	return j.lpop(key);
+    }
+
+    public String rpop(String key) {
+	Jedis j = getShard(key);
+	return j.rpop(key);
+    }
+
+    public int sadd(String key, String member) {
+	Jedis j = getShard(key);
+	return j.sadd(key, member);
+    }
+
+    public Set<String> smembers(String key) {
+	Jedis j = getShard(key);
+	return j.smembers(key);
+    }
+
+    public int srem(String key, String member) {
+	Jedis j = getShard(key);
+	return j.srem(key, member);
+    }
+
+    public String spop(String key) {
+	Jedis j = getShard(key);
+	return j.spop(key);
+    }
+
+    public int scard(String key) {
+	Jedis j = getShard(key);
+	return j.scard(key);
+    }
+
+    public int sismember(String key, String member) {
+	Jedis j = getShard(key);
+	return j.sismember(key, member);
+    }
+
+    public String srandmember(String key) {
+	Jedis j = getShard(key);
+	return j.srandmember(key);
+    }
+
+    public int zadd(String key, double score, String member) {
+	Jedis j = getShard(key);
+	return j.zadd(key, score, member);
+    }
+
+    public Set<String> zrange(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.zrange(key, start, end);
+    }
+
+    public int zrem(String key, String member) {
+	Jedis j = getShard(key);
+	return j.zrem(key, member);
+    }
+
+    public double zincrby(String key, double score, String member) {
+	Jedis j = getShard(key);
+	return j.zincrby(key, score, member);
+    }
+
+    public int zrank(String key, String member) {
+	Jedis j = getShard(key);
+	return j.zrank(key, member);
+    }
+
+    public int zrevrank(String key, String member) {
+	Jedis j = getShard(key);
+	return j.zrevrank(key, member);
+    }
+
+    public Set<String> zrevrange(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.zrevrange(key, start, end);
+    }
+
+    public Set<Tuple> zrangeWithScores(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.zrangeWithScores(key, start, end);
+    }
+
+    public Set<Tuple> zrevrangeWithScores(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.zrevrangeWithScores(key, start, end);
+    }
+
+    public int zcard(String key) {
+	Jedis j = getShard(key);
+	return j.zcard(key);
+    }
+
+    public double zscore(String key, String member) {
+	Jedis j = getShard(key);
+	return j.zscore(key, member);
+    }
+
+    public List<String> sort(String key) {
+	Jedis j = getShard(key);
+	return j.sort(key);
+    }
+
+    public List<String> sort(String key, SortingParams sortingParameters) {
+	Jedis j = getShard(key);
+	return j.sort(key, sortingParameters);
+    }
+
+    public int zcount(String key, double min, double max) {
+	Jedis j = getShard(key);
+	return j.zcount(key, min, max);
+    }
+
+    public Set<String> zrangeByScore(String key, double min, double max) {
+	Jedis j = getShard(key);
+	return j.zrangeByScore(key, min, max);
+    }
+
+    public Set<String> zrangeByScore(String key, double min, double max,
+	    int offset, int count) {
+	Jedis j = getShard(key);
+	return j.zrangeByScore(key, min, max, offset, count);
+    }
+
+    public Set<Tuple> zrangeByScoreWithScores(String key, double min, double max) {
+	Jedis j = getShard(key);
+	return j.zrangeByScoreWithScores(key, min, max);
+    }
+
+    public Set<Tuple> zrangeByScoreWithScores(String key, double min,
+	    double max, int offset, int count) {
+	Jedis j = getShard(key);
+	return j.zrangeByScoreWithScores(key, min, max, offset, count);
+    }
+
+    public int zremrangeByRank(String key, int start, int end) {
+	Jedis j = getShard(key);
+	return j.zremrangeByRank(key, start, end);
+    }
+
+    public int zremrangeByScore(String key, double start, double end) {
+	Jedis j = getShard(key);
+	return j.zremrangeByScore(key, start, end);
+    }
+
+    public void disconnect() throws IOException {
+	for (Jedis jedis : getAllShards()) {
+	    jedis.disconnect();
+	}
+    }
+
+    protected Jedis create(ShardInfo shard) {
+	Jedis c = new Jedis(shard.getHost(), shard.getPort());
+	if (shard.getPassword() != null) {
+	    c.auth(shard.getPassword());
+	}
+	return c;
+    }
+}