org.unipop.process.UniPredicatesStep Maven / Gradle / Ivy

Go to download
package org.unipop.process;

import org.apache.tinkerpop.gremlin.process.traversal.Traversal;
import org.unipop.process.properties.PropertyFetcher;
import org.unipop.structure.UniGraph;

import java.util.HashSet;
import java.util.Set;

public abstract class UniPredicatesStep extends UniBulkStep implements PropertyFetcher {

    protected Set propertyKeys;

    public UniPredicatesStep(Traversal.Admin traversal, UniGraph graph) {
        super(traversal, graph);
        this.propertyKeys = new HashSet<>();
    }

    @Override
    public void addPropertyKey(String key) {
        if (propertyKeys != null)
            propertyKeys.add(key);
    }

    @Override
    public void fetchAllKeys() {
        this.propertyKeys = null;
    }

    @Override
    public Set getKeys() {
        return propertyKeys;
    }
}