com.aliyun.tair.tairbloom.TairBloomPipeline Maven / Gradle / Ivy

Go to download
package com.aliyun.tair.tairbloom;

import com.aliyun.tair.tairbloom.params.BfmaddParams;
import com.aliyun.tair.ModuleCommand;
import com.aliyun.tair.tairbloom.factory.BloomBuilderFactory;
import com.aliyun.tair.tairbloom.params.BfinsertParams;
import com.aliyun.tair.tairbloom.params.BfmexistParams;
import redis.clients.jedis.BuilderFactory;
import redis.clients.jedis.Pipeline;
import redis.clients.jedis.Response;
import redis.clients.jedis.util.SafeEncoder;

import java.util.List;

import static redis.clients.jedis.Protocol.toByteArray;

public class TairBloomPipeline extends Pipeline {
    public Response bfreserve(String key, long initCapacity, double errorRate) {
        return bfreserve(SafeEncoder.encode(key), initCapacity, errorRate);
    }

    public Response bfreserve(byte[] key, long initCapacity, double errorRate) {
        getClient("").sendCommand(ModuleCommand.BFRESERVE, key, toByteArray(errorRate), toByteArray(initCapacity));
        return getResponse(BuilderFactory.STRING);
    }

    public Response bfadd(String key, String item) {
        return bfadd(SafeEncoder.encode(key), SafeEncoder.encode(item));
    }

    public Response bfadd(byte[] key, byte[] item) {
        getClient("").sendCommand(ModuleCommand.BFADD, key, item);
        return getResponse(BuilderFactory.BOOLEAN);
    }

    public Response bfmadd(String key, String... items) {
        BfmaddParams params = new BfmaddParams();
        getClient("").sendCommand(ModuleCommand.BFMADD, params.getByteParams(key, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    public Response bfmadd(byte[] key, byte[]... items) {
        BfmaddParams params = new BfmaddParams();
        getClient("").sendCommand(ModuleCommand.BFMADD, params.getByteParams(key, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    public Response bfexists(String key, String value) {
        getClient("").sendCommand(ModuleCommand.BFEXISTS, key, value);
        return getResponse(BuilderFactory.BOOLEAN);
    }

    public Response bfexists(byte[] key, byte[] value) {
        getClient("").sendCommand(ModuleCommand.BFEXISTS, key, value);
        return getResponse(BuilderFactory.BOOLEAN);
    }

    public Response bfmexists(String key, String... value) {
        BfmexistParams params = new BfmexistParams();
        getClient("").sendCommand(ModuleCommand.BFMEXISTS, params.getByteParams(key, value));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    public Response bfmexists(byte[] key, byte[]... value) {
        BfmexistParams params = new BfmexistParams();
        getClient("").sendCommand(ModuleCommand.BFMEXISTS, params.getByteParams(key, value));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    public Response bfinsert(String key, BfinsertParams params, String... items) {
        getClient("").sendCommand(ModuleCommand.BFINSERT,
            params.getByteParams(SafeEncoder.encode(key), SafeEncoder.encodeMany(items)));
        return getResponse(BloomBuilderFactory.BFINSERT_RESULT_BOOLEAN_LIST);
    }

    public Response bfinsert(byte[] key, BfinsertParams params, byte[]... items) {
        getClient("").sendCommand(ModuleCommand.BFINSERT, params.getByteParams(key, items));
        return getResponse(BloomBuilderFactory.BFINSERT_RESULT_BOOLEAN_LIST);
    }

    @Deprecated
    public Response bfinsert(String key, String initCapacityTag, long initCapacity, String errorRateTag, Double errorRate, String itemTag, String... items) {
        BfinsertParams params = new BfinsertParams();
        byte[][] metadata = params.getByteParamsMeta(key, initCapacityTag, String.valueOf(initCapacity), errorRateTag, String.valueOf(errorRate), itemTag);
        getClient("").sendCommand(ModuleCommand.BFINSERT, params.getByteParams(metadata, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    @Deprecated
    public Response bfinsert(String key, String nocreateTag, String itemTag, String... items) {
        BfinsertParams params = new BfinsertParams();
        byte[][] metadata = params.getByteParamsMeta(key, nocreateTag, itemTag);
        getClient("").sendCommand(ModuleCommand.BFINSERT, params.getByteParams(metadata, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    @Deprecated
    public Response bfinsert(String key, String itemTag, String... items) {
        BfinsertParams params = new BfinsertParams();
        byte[][] metadata = params.getByteParamsMeta(key, itemTag);
        getClient("").sendCommand(ModuleCommand.BFINSERT, params.getByteParams(metadata, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    @Deprecated
    public Response bfinsert(byte[] key, byte[] initCapacityTag, long initCapacity, byte[] errorRateTag, Double errorRate, byte[] itemTag, byte[]... items) {
        BfinsertParams params = new BfinsertParams();
        byte[][] metadata = params.getByteParamsMeta(key, initCapacityTag, toByteArray(initCapacity), errorRateTag, toByteArray(errorRate), itemTag);
        getClient("").sendCommand(ModuleCommand.BFINSERT, params.getByteParams(metadata, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    @Deprecated
    public Response bfinsert(byte[] key, byte[] nocreateTag, byte[] itemTag, byte[]... items) {
        BfinsertParams params = new BfinsertParams();
        byte[][] metadata = params.getByteParamsMeta(key, nocreateTag, itemTag);
        getClient("").sendCommand(ModuleCommand.BFINSERT, params.getByteParams(metadata, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    @Deprecated
    public Response bfinsert(byte[] key, byte[] itemTag, byte[]... items) {
        BfinsertParams params = new BfinsertParams();
        byte[][] metadata = params.getByteParamsMeta(key, itemTag);
        getClient("").sendCommand(ModuleCommand.BFINSERT, params.getByteParams(metadata, items));
        return getResponse(BloomBuilderFactory.BFMADD_RESULT_BOOLEAN_LIST);
    }

    public Response> bfdebug(String key) {
        getClient("").sendCommand(ModuleCommand.BFDEBUG, key);
        return getResponse(BuilderFactory.STRING_LIST);
    }

    public Response> bfdebug(byte[] key) {
        getClient("").sendCommand(ModuleCommand.BFDEBUG, key);
        return getResponse(BuilderFactory.STRING_LIST);
    }
}