com.datastax.driver.dse.graph.DefaultPath Maven / Gradle / Ivy

Go to download
/*
 *      Copyright (C) 2012-2016 DataStax Inc.
 *
 *      This software can be used solely with DataStax Enterprise. Please consult the license at
 *      http://www.datastax.com/terms/datastax-dse-driver-license-terms
 */
package com.datastax.driver.dse.graph;

import com.google.common.base.Function;
import com.google.common.base.Objects;
import com.google.common.collect.Lists;

import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

class DefaultPath implements Path {

    private static final Function, Set> TO_UNMODIFIABLE_SET = new Function, Set>() {
        @Override
        public Set apply(Set input) {
            return Collections.unmodifiableSet(input);
        }
    };

    List> labels;

    List objects;

    DefaultPath() {
    }

    @Override
    public List> getLabels() {
        return labels == null || labels.isEmpty()
                ? Collections.>emptyList()
                : Collections.unmodifiableList(Lists.transform(labels, TO_UNMODIFIABLE_SET));
    }

    @Override
    public List getObjects() {
        return objects == null || objects.isEmpty()
                ? Collections.emptyList()
                : Collections.unmodifiableList(objects);
    }

    @Override
    public int size() {
        return objects == null ? 0 : objects.size();
    }

    @Override
    public GraphNode getObject(int index) {
        if (index < 0 || index >= size())
            return null;
        return getObjects().get(index);
    }

    @Override
    public GraphNode getObject(String label) {
        List objects = getObjects(label);
        return objects.isEmpty() ? null : objects.get(0);
    }

    @Override
    public List getObjects(String label) {
        List objects = Lists.newArrayListWithExpectedSize(1);
        List> labels = getLabels();
        for (int i = 0; i < labels.size(); i++) {
            if (labels.get(i).contains(label)) {
                objects.add(getObject(i));
            }
        }
        return Collections.unmodifiableList(objects);
    }

    @Override
    public boolean hasLabel(String label) {
        for (Set labelGroup : getLabels()) {
            if (labelGroup.contains(label))
                return true;
        }
        return false;
    }

    @Override
    public Iterator iterator() {
        return getObjects().iterator();
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof Path)) return false;
        Path that = (Path) o;
        return Objects.equal(getLabels(), that.getLabels()) &&
                Objects.equal(getObjects(), that.getObjects());
    }

    @Override
    public int hashCode() {
        // getLabels() deliberately left out for faster hashcodes
        return Objects.hashCode(getObjects());
    }

    @Override
    public String toString() {
        return Objects.toStringHelper(this)
                .add("labels", getLabels())
                .add("objects", getObjects())
                .toString();
    }
}