org.jnosql.diana.redis.key.RedisSet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of redis-driver
The Eclipse JNoSQL communication layer, Diana, implementation Redis
There is a newer version: 0.0.9
/*
 *  Copyright (c) 2017 Otávio Santana and others
 *   All rights reserved. This program and the accompanying materials
 *   are made available under the terms of the Eclipse Public License v1.0
 *   and Apache License v2.0 which accompanies this distribution.
 *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
 *
 *   You may elect to redistribute this code under either of these licenses.
 *
 *   Contributors:
 *
 *   Otavio Santana
 */

package org.jnosql.diana.redis.key;

import redis.clients.jedis.Jedis;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Set;

class RedisSet extends RedisCollection implements Set {

    RedisSet(Jedis jedis, Class clazz, String keyWithNameSpace) {
        super(jedis, clazz, keyWithNameSpace);
    }

    @Override
    public boolean add(T e) {
        Objects.requireNonNull(e);
        if (isString) {
            jedis.sadd(keyWithNameSpace, e.toString());
        } else {
            jedis.sadd(keyWithNameSpace, JSONB.toJson(e));
        }
        return true;
    }

    @Override
    public void clear() {
        jedis.del(keyWithNameSpace);
    }

    @Override
    public int size() {
        return jedis.scard(keyWithNameSpace).intValue();
    }

    @Override
    protected int indexOf(Object o) {
        Objects.requireNonNull(o);

        String find = serialize(o);
        Set values = jedis.smembers(keyWithNameSpace);
        int index = 0;
        for (String value : values) {
            if (value.contains(find)) {
                return index;
            }
            index++;
        }
        return -1;
    }

    @Override
    protected T remove(int index) {
        T element = toArrayList().get(index);
        if (element == null) {
            return null;
        }
        remove(element);
        return element;
    }

    @Override
    public boolean remove(Object o) {
        if (!clazz.isInstance(o)) {
            throw new ClassCastException("The object required is " + clazz.getName());
        }
        String find = serialize(o);
        Set values = jedis.smembers(keyWithNameSpace);
        for (String value : values) {
            if (value.contains(find)) {
                jedis.srem(keyWithNameSpace, value);
                return true;
            }
        }
        return false;
    }

    @Override
    protected List toArrayList() {
        Set redisValues = jedis.smembers(keyWithNameSpace);
        List list = new ArrayList<>();
        for (String redisValue : redisValues) {
            if (isString) {
                list.add((T) redisValue);
            } else {
                list.add(JSONB.fromJson(redisValue, clazz));
            }

        }
        return list;
    }

}