com.aliyun.tair.mcommamd.AliyunRedisCommand 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.mcommamd;

import java.util.ArrayList;
import java.util.List;

import com.aliyun.tair.ModuleCommand;
import com.aliyun.tair.mcommamd.factory.AliyunRedisCommandBuilderFactory;
import com.aliyun.tair.mcommamd.results.SlotAndNodeIndex;
import redis.clients.jedis.BuilderFactory;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisMonitor;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.Protocol.Command;
import redis.clients.jedis.ScanParams;
import redis.clients.jedis.ScanResult;
import redis.clients.jedis.util.SafeEncoder;

public class AliyunRedisCommand {
    private Jedis jedis;
    private JedisPool jedisPool;

    public AliyunRedisCommand(Jedis jedis) {
        this.jedis = jedis;
    }

    public AliyunRedisCommand(JedisPool jedisPool) {
        this.jedisPool = jedisPool;
    }

    private Jedis getJedis() {
        if (jedisPool != null) {
            return jedisPool.getResource();
        }
        return jedis;
    }

    private void releaseJedis(Jedis jedis) {
        if (jedisPool != null) {
            jedis.close();
        }
    }

    /**
     * Query the slot and db to which the key belongs. Redis' native info command can take at most one optional section
     * (info [section]). Currently, in ApsaraDB for Redis cluster instances, some commands restrict all keys to be in
     * the same slot. The info key command is convenient for users to query whether certain keys are in the same slot
     * or db node.
     * @param key the key
     * @return SlotAndNodeIndex
     */
    public SlotAndNodeIndex infoKey(String key) {
        Jedis jedis = getJedis();
        try {
            Object obj = jedis.sendCommand(Command.INFO, "KEY", key);
            return AliyunRedisCommandBuilderFactory.SlotAndNodeIndexResult.build(obj);
        } finally {
            releaseJedis(jedis);
        }
    }

    public SlotAndNodeIndex infoKey(byte[] key) {
        Jedis jedis = getJedis();
        try {
            Object obj = jedis.sendCommand(Command.INFO, "KEY".getBytes(), key);
            return AliyunRedisCommandBuilderFactory.SlotAndNodeIndexResult.build(obj);
        } finally {
            releaseJedis(jedis);
        }
    }

    /**
     * Execute the info command on the specified Redis node.
     * @param index the node index, see {@link AliyunRedisCommand#infoKey(String)}.
     * @return the info content.
     */
    public String iInfo(int index) {
        Jedis jedis = getJedis();
        try {
            Object obj = jedis.sendCommand(ModuleCommand.IINFO, String.valueOf(index));
            return BuilderFactory.STRING.build(obj);
        } finally {
            releaseJedis(jedis);
        }
    }

    public String iInfo(int index, String section) {
        Jedis jedis = getJedis();
        try {
            Object obj = jedis.sendCommand(ModuleCommand.IINFO, String.valueOf(index), section);
            return BuilderFactory.STRING.build(obj);
        } finally {
            releaseJedis(jedis);
        }
    }

    /**
     * Similar to the iinfo command, but can only be used in read-write separation mode. The idx of a readonly slave
     * is added to the usage, which is used to specify the number of readonly slaves to execute the info command.
     * It can be used to execute the info command on the specified readonly slave in a read-write separation cluster.
     * If used in a non-read-write split cluster, an error will be returned.
     * @param index the node index, see {@link AliyunRedisCommand#infoKey(String)}.
     * @param roIndex the read-only node index.
     * @return the info content.
     */
    public String rIInfo(int index, int roIndex) {
        Jedis jedis = getJedis();
        try {
            Object obj = jedis.sendCommand(ModuleCommand.RIINFO, String.valueOf(index), String.valueOf(roIndex));
            return BuilderFactory.STRING.build(obj);
        } finally {
            releaseJedis(jedis);
        }
    }

    public String rIInfo(int index, int roIndex, String section) {
        Jedis jedis = getJedis();
        try {
            Object obj = jedis.sendCommand(ModuleCommand.RIINFO, String.valueOf(index), String.valueOf(roIndex), section);
            return BuilderFactory.STRING.build(obj);
        } finally {
            releaseJedis(jedis);
        }
    }

    /**
     * In cluster mode, the scan command can be executed on the specified db node. Based on the scan command,
     * a parameter is extended to specify db_idx. The range of db_idx is [0, nodecount]. The nodecount can be obtained
     * through the info command or viewed from the console.
     * @param index
     * @param cursor
     * @return
     */
    public ScanResult iScan(int index, String cursor) {
        return iScan(index, cursor, new ScanParams());
    }

    public ScanResult iScan(int index, String cursor, ScanParams params) {
        final List args = new ArrayList<>();
        args.add(String.valueOf(index).getBytes());
        args.add(cursor.getBytes());
        args.addAll(params.getParams());

        Jedis jedis = getJedis();
        try {
            Object obj = jedis.sendCommand(ModuleCommand.ISCAN, args.toArray(new byte[args.size()][]));
            List