ai.grakn.redismock.commands.RO_pfcount Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of redis-mock Show documentation

In memory Redis mock for testing

There is a newer version: 0.1.6

package ai.grakn.redismock.commands;

import ai.grakn.redismock.RedisBase;
import ai.grakn.redismock.Response;
import ai.grakn.redismock.Slice;
import com.google.common.collect.Sets;

import java.util.List;
import java.util.Set;

import static ai.grakn.redismock.Utils.deserializeObject;

class RO_pfcount extends AbstractRedisOperation {
    RO_pfcount(RedisBase base, List params) {
        super(base, params, null, 0, null);
    }

    Slice response() {
        Set set = Sets.newHashSet();
        for (Slice key : params()) {
            Slice data = base().rawGet(key);
            if (data == null) {
                continue;
            }

            Set s = deserializeObject(data);
            set.addAll(s);
        }
        return Response.integer((long) set.size());
    }
}