com.aliyun.tair.tairzset.TairZsetCluster Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of alibabacloud-tairjedis-sdk Show documentation
There is a newer version: 3.0.9
Show newest version
package com.aliyun.tair.tairzset;

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

import com.aliyun.tair.ModuleCommand;
import com.aliyun.tair.tairzset.params.ExzaddParams;
import com.aliyun.tair.tairzset.params.ExzrangeParams;
import com.aliyun.tair.util.JoinParameters;
import redis.clients.jedis.BuilderFactory;
import redis.clients.jedis.JedisCluster;
import redis.clients.jedis.util.SafeEncoder;

import static com.aliyun.tair.tairzset.LeaderBoard.joinScoresToString;
import static redis.clients.jedis.Protocol.toByteArray;

public class TairZsetCluster {
    private JedisCluster jc;

    public TairZsetCluster(JedisCluster jc) {
        this.jc = jc;
    }

    /**
     * Adds all the specified members with the specified (multi)scores to the tairzset stored at key.
     * @param key
     * @param member
     * @param scores
     * @return
     */
    public Long exzadd(final String key, final String member, final double... scores) {
        return exzadd(SafeEncoder.encode(key), SafeEncoder.encode(joinScoresToString(scores)),
            SafeEncoder.encode(member));
    }
    public Long exzadd(final byte[] key, final byte[] member, final double... scores) {
        return exzadd(key, SafeEncoder.encode(joinScoresToString(scores)), member);
    }

    public Long exzadd(final String key, final String score, final String member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZADD, key, score, member);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzadd(final byte[] key, final byte[] score, final byte[] member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZADD, key, score, member);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzadd(final String key, final String score, final String member, final ExzaddParams params) {
        return exzadd(SafeEncoder.encode(key), SafeEncoder.encode(score), SafeEncoder.encode(member), params);
    }

    public Long exzadd(final byte[] key, final byte[] score, final byte[] member, final ExzaddParams params) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZADD, params.getByteParams(key, score, member));
        return BuilderFactory.LONG.build(obj);
    }

    /** see {@link TairZsetCluster#exzaddMembers(String, Map)} */
    @Deprecated
    public Long exzadd(final String key, final Map scoreMembers) {
        final List bparams = new ArrayList();
        bparams.add(SafeEncoder.encode(key));

        for (final Entry entry : scoreMembers.entrySet()) {
            bparams.add(SafeEncoder.encode(entry.getKey()));
            bparams.add(SafeEncoder.encode(entry.getValue()));
        }
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZADD,
            bparams.toArray(new byte[bparams.size()][]));
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzaddMembers(final String key, final Map members) {
        final List bparams = new ArrayList();
        bparams.add(SafeEncoder.encode(key));

        for (final Entry entry : members.entrySet()) {
            bparams.add(SafeEncoder.encode(entry.getValue()));
            bparams.add(SafeEncoder.encode(entry.getKey()));
        }
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZADD,
            bparams.toArray(new byte[bparams.size()][]));
        return BuilderFactory.LONG.build(obj);
    }

    /** see {@link TairZsetCluster#exzaddMembers(byte[], Map)} */
    @Deprecated
    public Long exzadd(final byte[] key, final Map scoreMembers) {
        final List bparams = new ArrayList();
        bparams.add(key);

        for (final Entry entry : scoreMembers.entrySet()) {
            bparams.add(entry.getKey());
            bparams.add(entry.getValue());
        }
        Object obj = jc.sendCommand(key, ModuleCommand.EXZADD, bparams.toArray(new byte[bparams.size()][]));
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzaddMembers(final byte[] key, final Map members) {
        final List bparams = new ArrayList();
        bparams.add(key);

        for (final Entry entry : members.entrySet()) {
            bparams.add(entry.getValue());
            bparams.add(entry.getKey());
        }
        Object obj = jc.sendCommand(key, ModuleCommand.EXZADD, bparams.toArray(new byte[bparams.size()][]));
        return BuilderFactory.LONG.build(obj);
    }

    /** see {@link TairZsetCluster#exzaddMembers(String, Map, ExzaddParams)} */
    @Deprecated
    public Long exzadd(final String key, final Map scoreMembers, final ExzaddParams params) {
        final List bparams = new ArrayList();
        for (final Entry entry : scoreMembers.entrySet()) {
            bparams.add(SafeEncoder.encode(entry.getKey()));
            bparams.add(SafeEncoder.encode(entry.getValue()));
        }
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZADD,
            params.getByteParams(SafeEncoder.encode(key), bparams.toArray(new byte[bparams.size()][])));
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzaddMembers(final String key, final Map members, final ExzaddParams params) {
        final List bparams = new ArrayList();
        for (final Entry entry : members.entrySet()) {
            bparams.add(SafeEncoder.encode(entry.getValue()));
            bparams.add(SafeEncoder.encode(entry.getKey()));
        }
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZADD,
            params.getByteParams(SafeEncoder.encode(key), bparams.toArray(new byte[bparams.size()][])));
        return BuilderFactory.LONG.build(obj);
    }
    
    /** see {@link TairZsetCluster#exzaddMembers(byte[], Map, ExzaddParams)} */
    @Deprecated
    public Long exzadd(final byte[] key, final Map scoreMembers, final ExzaddParams params) {
        final List bparams = new ArrayList();
        for (final Entry entry : scoreMembers.entrySet()) {
            bparams.add(entry.getKey());
            bparams.add(entry.getValue());
        }
        Object obj = jc.sendCommand(key, ModuleCommand.EXZADD,
            params.getByteParams(key, bparams.toArray(new byte[bparams.size()][])));
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzaddMembers(final byte[] key, final Map members, final ExzaddParams params) {
        final List bparams = new ArrayList();
        for (final Entry entry : members.entrySet()) {
            bparams.add(entry.getValue());
            bparams.add(entry.getKey());
        }
        Object obj = jc.sendCommand(key, ModuleCommand.EXZADD,
            params.getByteParams(key, bparams.toArray(new byte[bparams.size()][])));
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * Increments the score of member in the tairzset stored at key by increment.
     * @param key
     * @param increment
     * @param member
     * @return
     */
    public String exzincrBy(final String key, final String increment, final String member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZINCRBY, key, increment, member);
        return BuilderFactory.STRING.build(obj);
    }

    public byte[] exzincrBy(final byte[] key, final byte[] increment, final byte[] member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZINCRBY, key, increment, member);
        return BuilderFactory.BYTE_ARRAY.build(obj);
    }

    public String exzincrBy(final String key, final String member, final double... scores) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZINCRBY, key, joinScoresToString(scores), member);
        return BuilderFactory.STRING.build(obj);
    }

    public byte[] exzincrBy(final byte[] key, final byte[] member, final double... scores) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZINCRBY, key,
            SafeEncoder.encode(joinScoresToString(scores)), member);
        return BuilderFactory.BYTE_ARRAY.build(obj);
    }

    /**
     * Removes the specified members from the tairzset stored at key. Non existing members are ignored.
     * @param key
     * @param member
     * @return
     */
    public Long exzrem(final String key, final String... member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREM, JoinParameters.joinParameters(key, member));
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzrem(final byte[] key, final byte[]... member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREM, JoinParameters.joinParameters(key, member));
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * Removes all elements in the tairzset stored at key with a score between min and max (inclusive).
     * @param key
     * @param min
     * @param max
     * @return
     */
    public Long exzremrangeByScore(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREMRANGEBYSCORE, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzremrangeByScore(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREMRANGEBYSCORE, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * Removes all elements in the tairzset stored at key with rank between start and stop.
     * @param key
     * @param start
     * @param stop
     * @return
     */
    public Long exzremrangeByRank(final String key, final long start, final long stop) {
        return exzremrangeByRank(SafeEncoder.encode(key), start, stop);
    }

    public Long exzremrangeByRank(final byte[] key, final long start, final long stop) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREMRANGEBYRANK, key, toByteArray(start),
            toByteArray(stop));
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * When all the elements in a sorted set are inserted with the same score, in order to force lexicographical
     * ordering, this command removes all elements in the sorted set stored at key between the lexicographical range
     * specified by min and max.
     *
     * The meaning of min and max are the same of the ZRANGEBYLEX command. Similarly, this command actually removes
     * the same elements that ZRANGEBYLEX would return if called with the same min and max arguments.
     * @param key
     * @param min
     * @param max
     * @return
     */
    public Long exzremrangeByLex(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREMRANGEBYLEX, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzremrangeByLex(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREMRANGEBYLEX, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * Returns the score of member in the tairzset at key.
     * @param key
     * @param member
     * @return
     */
    public String exzscore(final String key, final String member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZSCORE, key, member);
        return BuilderFactory.STRING.build(obj);
    }

    public byte[] exzscore(final byte[] key, final byte[] member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZSCORE, key, member);
        return BuilderFactory.BYTE_ARRAY.build(obj);
    }

    /**
     * Returns the specified range of elements in the sorted set stored at key.
     * @param key
     * @param min
     * @param max
     * @return
     */
    public List exzrange(final String key, final long min, final long max) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZRANGE, SafeEncoder.encode(key),
            toByteArray(min), toByteArray(max));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrange(final byte[] key, final long min, final long max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGE, key, toByteArray(min), toByteArray(max));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    public List exzrangeWithScores(final String key, final long min, final long max) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZRANGE, SafeEncoder.encode(key),
            toByteArray(min), toByteArray(max), SafeEncoder.encode("WITHSCORES"));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrangeWithScores(final byte[] key, final long min, final long max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGE, key, toByteArray(min), toByteArray(max),
            SafeEncoder.encode("WITHSCORES"));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    /**
     * Returns the specified range of elements in the sorted set stored at key. The elements are considered to be
     * ordered from the highest to the lowest score.
     * @param key
     * @param min
     * @param max
     * @return
     */
    public List exzrevrange(final String key, final long min, final long max) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZREVRANGE, SafeEncoder.encode(key),
            toByteArray(min), toByteArray(max));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrevrange(final byte[] key, final long min, final long max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGE, key, toByteArray(min), toByteArray(max));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    public List exzrevrangeWithScores(final String key, final long min, final long max) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZREVRANGE, SafeEncoder.encode(key),
            toByteArray(min), toByteArray(max), SafeEncoder.encode("WITHSCORES"));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrevrangeWithScores(final byte[] key, final long min, final long max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGE, key, toByteArray(min), toByteArray(max),
            SafeEncoder.encode("WITHSCORES"));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    /**
     * Returns all the elements in the tairzset at key with a score between min and max (including elements with
     * score equal to min or max). The elements are considered to be ordered from low to high scores.
     * @param key
     * @param min
     * @param max
     * @return
     */
    public List exzrangeByScore(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGEBYSCORE, key, min, max);
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrangeByScore(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGEBYSCORE, key, min, max);
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    public List exzrangeByScore(final String key, final String min, final String max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZRANGEBYSCORE, params.getByteParams(SafeEncoder.encode(key),
            SafeEncoder.encode(min), SafeEncoder.encode(max)));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrangeByScore(final byte[] key, final byte[] min, final byte[] max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGEBYSCORE, params.getByteParams(key, min, max));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    /**
     * Returns all the elements in the tairzset at key with a score between max and min (including elements with score
     * equal to max or min). In contrary to the default ordering of tairzsets, for this command the elements are
     * considered to be ordered from high to low scores.
     * @param key
     * @param min
     * @param max
     * @return
     */
    public List exzrevrangeByScore(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGEBYSCORE, key, min, max);
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrevrangeByScore(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGEBYSCORE, key, min, max);
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    public List exzrevrangeByScore(final String key, final String min, final String max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZREVRANGEBYSCORE, params.getByteParams(SafeEncoder.encode(key),
            SafeEncoder.encode(min), SafeEncoder.encode(max)));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrevrangeByScore(final byte[] key, final byte[] min, final byte[] max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGEBYSCORE, params.getByteParams(key, min, max));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    /**
     *
     * @param key
     * @param min
     * @param max
     * @return
     */
    public List exzrangeByLex(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGEBYLEX, key, min, max);
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrangeByLex(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGEBYLEX, key, min, max);
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    public List exzrangeByLex(final String key, final String min, final String max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZRANGEBYLEX, params.getByteParams(SafeEncoder.encode(key),
            SafeEncoder.encode(min), SafeEncoder.encode(max)));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrangeByLex(final byte[] key, final byte[] min, final byte[] max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANGEBYLEX, params.getByteParams(key, min, max));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    /**
     *
     * @param key
     * @param min
     * @param max
     * @return
     */
    public List exzrevrangeByLex(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGEBYLEX, key, min, max);
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrevrangeByLex(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGEBYLEX, key, min, max);
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    public List exzrevrangeByLex(final String key, final String min, final String max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(SafeEncoder.encode(key), ModuleCommand.EXZREVRANGEBYLEX, params.getByteParams(SafeEncoder.encode(key),
            SafeEncoder.encode(min), SafeEncoder.encode(max)));
        return BuilderFactory.STRING_LIST.build(obj);
    }

    public List exzrevrangeByLex(final byte[] key, final byte[] min, final byte[] max, final ExzrangeParams params) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANGEBYLEX, params.getByteParams(key, min, max));
        return BuilderFactory.BYTE_ARRAY_LIST.build(obj);
    }

    /**
     * Returns the tairzset cardinality (number of elements) of the tairzset stored at key.
     * @param key
     * @return
     */
    public Long exzcard(final String key) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZCARD, key);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzcard(final byte[] key) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZCARD, key);
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * Returns the rank of member in the tairzset stored at key, with the scores ordered from low to high.
     * The rank (or index) is 0-based, which means that the member with the lowest score has rank 0.
     * @param key
     * @param member
     * @return
     */
    public Long exzrank(final String key, final String member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANK, key, member);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzrank(final byte[] key, final byte[] member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANK, key, member);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzrevrank(final String key, final String member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANK, key, member);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzrevrank(final byte[] key, final byte[] member) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANK, key, member);
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * Same with zrank, but use score to get rank, when the field corresponding to score does not exist,
     * an estimate is used.
     * @param key
     * @param score
     * @return
     */
    public Long exzrankByScore(final String key, final String score) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANKBYSCORE, key, score);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzrankByScore(final byte[] key, final byte[] score) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZRANKBYSCORE, key, score);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzrevrankByScore(final String key, final String score) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANKBYSCORE, key, score);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzrevrankByScore(final byte[] key, final byte[] score) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZREVRANKBYSCORE, key, score);
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * Returns the number of elements in the tairzset at key with a score between min and max.
     * @param key
     * @param min
     * @param max
     * @return
     */
    public Long exzcount(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZCOUNT, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzcount(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZCOUNT, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }

    /**
     * When all the elements in a tairzset are inserted with the same score, in order to force lexicographical ordering,
     * this command returns the number of elements in the tairzset at key with a value between min and max.
     * @param key
     * @param min
     * @param max
     * @return
     */
    public Long exzlexcount(final String key, final String min, final String max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZLEXCOUNT, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }

    public Long exzlexcount(final byte[] key, final byte[] min, final byte[] max) {
        Object obj = jc.sendCommand(key, ModuleCommand.EXZLEXCOUNT, key, min, max);
        return BuilderFactory.LONG.build(obj);
    }
}