org.ff4j.store.PropertyStoreRedisLettuce Maven / Gradle / Ivy

Go to download
package org.ff4j.store;

/*-
 * #%L
 * ff4j-store-redis
 * %%
 * Copyright (C) 2013 - 2023 FF4J
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;

import org.ff4j.exception.PropertyAlreadyExistException;
import org.ff4j.property.Property;
import org.ff4j.property.store.AbstractPropertyStore;
import org.ff4j.redis.RedisKeysBuilder;
import org.ff4j.utils.Util;
import org.ff4j.utils.json.PropertyJsonParser;

import io.lettuce.core.RedisClient;
import io.lettuce.core.api.sync.RedisCommands;
import io.lettuce.core.cluster.RedisClusterClient;
import io.lettuce.core.cluster.api.sync.RedisAdvancedClusterCommands;

/**
 * Implementation of property store for REDIS.
 *
 * @author Cedrick Lunven (@clunven)
 * @author Shridhar Navanageri
 */
public class PropertyStoreRedisLettuce extends AbstractPropertyStore {

    /** Supports sentinel based redis deployment. */ 
    private RedisCommands redisCommands;
    
    /** Support the cluster based redis deployment. */
    private RedisAdvancedClusterCommands redisCommandsCluster;
    
    /** Default key builder. */
    private RedisKeysBuilder keyBuilder = new RedisKeysBuilder();
    
    /**
     * Public void.
     */
    public PropertyStoreRedisLettuce(RedisClient redisClient) {
        this(redisClient, new RedisKeysBuilder());
    }
    public PropertyStoreRedisLettuce(RedisClient redisClient, RedisKeysBuilder keyBuilder) {
        this.redisCommands = redisClient.connect().sync();
        this.keyBuilder    = keyBuilder;
    }
    public PropertyStoreRedisLettuce(RedisClusterClient redisClusterClient) {
        this(redisClusterClient, new RedisKeysBuilder());
    }
    public PropertyStoreRedisLettuce(RedisClusterClient redisClusterClient, RedisKeysBuilder keyBuilder) {
        this.redisCommandsCluster = redisClusterClient.connect().sync();
        this.keyBuilder    = keyBuilder;
    }

    /**
     * {@inheritDoc}
     */
    public boolean existProperty(String name) {
        Util.assertParamHasLength(name, "PropertyName identifier");
        String key   = keyBuilder.getKeyProperty(name);
        return 1 == ((null != redisCommands) ? 
                redisCommands.exists(key) : 
                redisCommandsCluster.exists(key));
    }

    /**
     * {@inheritDoc}
     */
    public  void createProperty(Property prop) {
        Util.assertNotNull(prop);
        if (existProperty(prop.getName())) {
            throw new PropertyAlreadyExistException(prop.getName());
        }
        if (null != redisCommands) {
            redisCommands.sadd(keyBuilder.getKeyPropertyMap(), prop.getName());
            redisCommands.set(keyBuilder.getKeyProperty(prop.getName()), prop.toJson());
            redisCommands.persist(keyBuilder.getKeyProperty(prop.getName()));
        } else {
            redisCommandsCluster.sadd(keyBuilder.getKeyPropertyMap());
            redisCommandsCluster.set(keyBuilder.getKeyProperty(prop.getName()), prop.toJson());
            redisCommandsCluster.persist(keyBuilder.getKeyProperty(prop.getName()));
        }
    }

    /**
     * {@inheritDoc}
     */
    public Property readProperty(String name) {
        assertPropertyExist(name);
        String key = keyBuilder.getKeyProperty(name);
        return PropertyJsonParser.parseProperty((null != redisCommands) ? 
                redisCommands.get(key) : 
                redisCommandsCluster.get(key));
    }

    /**
     * {@inheritDoc}
     */
    public void deleteProperty(String name) {
        assertPropertyExist(name);
        if (null != redisCommands) {
            redisCommands.srem(keyBuilder.getKeyPropertyMap(), name);
            redisCommands.del(keyBuilder.getKeyProperty(name));
        } else {
            redisCommandsCluster.srem(keyBuilder.getKeyPropertyMap(), name);
            redisCommandsCluster.del(keyBuilder.getKeyProperty(name));
        }
    }    

    /**
     * {@inheritDoc}
     */
    public Set listPropertyNames() {
        return (null != redisCommands) ? 
                redisCommands.smembers(keyBuilder.getKeyPropertyMap()) : 
                redisCommandsCluster.smembers(keyBuilder.getKeyPropertyMap());
    }

    /**
     * {@inheritDoc}
     */
    public Map> readAllProperties() {
        LinkedHashMap> mapP = new LinkedHashMap>();
        Set properties = listPropertyNames();
        if (properties != null) {
            for (String key : properties) {
                mapP.put(key, readProperty(key));
            }
         }
         return mapP;
    }
    
    /**
     * {@inheritDoc}
     */
    public void clear() {
        Set myKeys = listPropertyNames();
        for (String key : myKeys) {
            deleteProperty(key);
        }
    }
}