package redis.clients.jedis;

import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Set;

import redis.clients.util.Hashing;
import redis.clients.util.ShardInfo;
import redis.clients.util.Sharded;

public class ShardedJedis extends Sharded<Jedis> {
    public ShardedJedis(List<ShardInfo> shards) {
	super(shards);
    }

    public ShardedJedis(List<ShardInfo> shards, Hashing algo) {
	super(shards, algo);
    }

    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) {
	return new Jedis(shard);
    }
}