com.naah69.core.handler.HyperLogLogHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-starter-redisson Show documentation
redisson for Spring Boot
The newest version!
package com.naah69.core.handler;

import org.springframework.data.redis.core.HyperLogLogOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;

import java.util.List;

/**
 * 基数助手
 *
 * @author xsx
 * @since 1.8
 */
public final class HyperLogLogHandler implements RedisHandler {
    /**
     * 对象模板
     */
    private RedisTemplate redisTemplate;
    /**
     * 字符串模板
     */
    private StringRedisTemplate stringRedisTemplate;
    /**
     * 对象模板
     */
    private HyperLogLogOperations hyperLogLogOperations;
    /**
     * 字符串模板
     */
    private HyperLogLogOperations stringHyperLogLogOperations;

    /**
     * 基数助手构造
     *
     * @param dbIndex 数据库索引
     */
    @SuppressWarnings("unchecked")
    HyperLogLogHandler(Integer dbIndex) {
        List templateList = HandlerManager.createTemplate(dbIndex);
        this.redisTemplate = templateList.get(0);
        this.stringRedisTemplate = (StringRedisTemplate) templateList.get(1);
        this.hyperLogLogOperations = redisTemplate.opsForHyperLogLog();
        this.stringHyperLogLogOperations = stringRedisTemplate.opsForHyperLogLog();
    }

    /**
     * 添加对象
     *
     * @param key    键
     * @param values 对象
     * @return 返回添加对象成功数量
     * @see Redis Documentation: PFADD
     * @since redis 2.8.9
     */
    public Long addAsObj(String key, Object... values) {
        return this.hyperLogLogOperations.add(key, values);
    }

    /**
     * 添加字符串
     *
     * @param key    键
     * @param values 字符串
     * @return 返回添加字符串成功数量
     * @see Redis Documentation: PFADD
     * @since redis 2.8.9
     */
    public Long add(String key, String... values) {
        return this.stringHyperLogLogOperations.add(key, values);
    }

    /**
     * 获取对象基数估算值
     *
     * @param keys 键
     * @return 返回对象基数估算值
     * @see Redis Documentation: PFCOUNT
     * @since redis 2.8.9
     */
    public Long sizeAsObj(String... keys) {
        return this.hyperLogLogOperations.size(keys);
    }

    /**
     * 获取字符串基数估算值
     *
     * @param keys 键
     * @return 返回字符串基数估算值
     * @see Redis Documentation: PFCOUNT
     * @since redis 2.8.9
     */
    public Long size(String... keys) {
        return this.stringHyperLogLogOperations.size(keys);
    }

    /**
     * 合并对象
     *
     * @param key       键
     * @param otherKeys 其他键
     * @return 返回合并后的对象基数估算值
     * @see Redis Documentation: PFMERGE
     * @since redis 2.8.9
     */
    public Long unionAsObj(String key, String... otherKeys) {
        return this.hyperLogLogOperations.union(key, otherKeys);
    }

    /**
     * 合并字符串
     *
     * @param key       键
     * @param otherKeys 字符串
     * @return 返回合并后的字符串基数估算值
     * @see Redis Documentation: PFMERGE
     * @since redis 2.8.9
     */
    public Long union(String key, String... otherKeys) {
        return this.stringHyperLogLogOperations.union(key, otherKeys);
    }

    /**
     * 获取spring redis模板
     *
     * @return 返回对象模板
     */
    public RedisTemplate getRedisTemplate() {
        return this.redisTemplate;
    }

    /**
     * 获取spring string redis模板
     *
     * @return 返回字符串模板
     */
    public StringRedisTemplate getStringRedisTemplate() {
        return this.stringRedisTemplate;
    }
}