common.framework.redis.RedisReactiveUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-framework-util Show documentation
There is a newer version: 1.0.2-RELEASE
Show newest version
package common.framework.redis;

import io.lettuce.core.RedisClient;
import io.lettuce.core.RedisURI;
import io.lettuce.core.SetArgs;
import io.lettuce.core.api.StatefulRedisConnection;
import io.lettuce.core.api.reactive.RedisReactiveCommands;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.time.Duration;
import java.util.List;

/**
 * Description: 反应式Redis工具
 *
 * @author linan
 * @date 2021-01-15
 */
@Component
public class RedisReactiveUtil {

    private RedisReactiveCommands redisReactiveCommands;

    private int timeout;

    /**
     *  init redis
     * @param url  url
     * @param port port
     * @param timeout  expire time in seconds
     */
    public RedisReactiveUtil (@Value("${spring.redis.hosturl}")String url,
                           @Value("${spring.redis.ports}")int port,
                           @Value("${spring.redis.expire}")int timeout){
        RedisURI redisUri = RedisURI.builder()
                .withHost(url)
                .withPort(port)
                .build();
        //判断是否有过期时间
        if(0==timeout){
            redisUri.setTimeout(Duration.ZERO);
        }else {
            redisUri.setTimeout(Duration.ofSeconds(timeout, 0));
            this.timeout = timeout;
        }
        RedisClient redisClient = RedisClient.create(redisUri);
        StatefulRedisConnection connection = redisClient.connect();
        this.redisReactiveCommands = connection.reactive();
    }

    /**
     *  get key value
     * @param key
     * @return
     */
    public Mono get(String key){
        return redisReactiveCommands.get(key);
    }

    /**
     * save with timeout
     * @param key   key
     * @param value values
     * @param timeout expire time in seconds
     */
    public Mono set(String key,Object value,int timeout){
        SetArgs setArgs = SetArgs.Builder.nx().ex(timeout);
        return redisReactiveCommands.set(key, value, setArgs);
    }

    /**
     *  save without outtime
     * @param key   key
     * @param value values
     */
    public Mono set(String key,Object value){
        if (0 != timeout){
            SetArgs setArgs = SetArgs.Builder.nx().ex(timeout);
           return redisReactiveCommands.set(key, value, setArgs);
        }else {
            return redisReactiveCommands.set(key, value);
        }

    }

    /**
     * delete key
     * @param keys keys
     */
    public Mono delete(Object... keys){
        return redisReactiveCommands.del(keys);
    }

    /**
     * update key timeout
     * @param key key
     * @param timeout timeout in second
     */
    public Mono expire(Object key,long timeout){
        return redisReactiveCommands.expire(key,timeout);
    }

    /**
     *  get all keys with pattern
     * @param pattern 正则表达式
     * @return flux
     */
    public Flux getKeyList(String pattern){
        return redisReactiveCommands.keys(pattern);
    }

}