com.github.bingoohuang.voucherno.RedisBitSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of voucher-no Show documentation

voucher NO. generator

There is a newer version: 0.0.7

package com.github.bingoohuang.voucherno;


import lombok.val;
import redis.clients.jedis.BinaryJedis;
import redis.clients.jedis.Protocol;

import java.util.Arrays;

public class RedisBitSet {
    private final BinaryJedis jedis;
    private final byte[] name;

    public RedisBitSet(BinaryJedis jedis, String name) {
        this.jedis = jedis;
        this.name = name.getBytes(MessageDigestUtils.UTF8);
    }

    public boolean add(int[] hashes) {
        val multi = jedis.multi();
        Arrays.stream(hashes).forEach(x -> multi.setbit(name, x, Protocol.BYTES_TRUE));
        // 只要有一个比特位之前没被设置过，说明曾经不存在，返回为true
        return multi.exec().stream().filter(x -> x == Boolean.FALSE).count() > 0;
    }

    public boolean maybeContains(int[] hashes) {
        val multi = jedis.multi();
        Arrays.stream(hashes).forEach(x -> multi.getbit(name, x));
        // 所有比特位，都被设置了，那就是可能被包含了。
        return multi.exec().stream().filter(x -> x == Boolean.FALSE).count() == 0;
    }
}