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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dse-java-driver-core Show documentation

A driver for DataStax Enterprise (DSE) and Apache Cassandra 1.2+ clusters that works exclusively with the Cassandra Query Language version 3 (CQL3) and Cassandra's binary protocol, supporting DSE-specific features such as geospatial types, DSE Graph and DSE authentication.

There is a newer version: 2.4.0

Show newest version

/*
 *      Copyright (C) 2012-2017 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.ImmutableSet;
import com.google.common.collect.Iterators;
import com.google.common.collect.Multimap;

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

/**
 * Base class for elements (vertices, edges and vertex properties).
 */
abstract class DefaultElement implements Element {

    private static final Function GRAPH_NODE_TO_PROPERTY = new Function() {
        @Override
        public Property apply(GraphNode input) {
            return input.asProperty();
        }
    };

    GraphNode id;

    String label;

    Multimap properties;

    DefaultElement() {
    }

    @Override
    public GraphNode getId() {
        return id;
    }

    @Override
    public String getLabel() {
        return label;
    }

    @Override
    public Set getPropertyNames() {
        if (properties == null || properties.isEmpty())
            return Collections.emptySet();
        return Collections.unmodifiableSet(properties.keySet());
    }

    @Override
    public Property getProperty(String name) {
        if (properties == null || properties.isEmpty() || !properties.containsKey(name))
            return null;
        Iterator iterator = getProperties(name);
        if (!iterator.hasNext())
            return null;
        return iterator.next();
    }

    @Override
    public Iterator getProperties(String name) {
        if (properties == null || properties.isEmpty() || !properties.containsKey(name))
            return ImmutableSet.of().iterator();
        return Iterators.unmodifiableIterator(Iterators.transform(properties.get(name).iterator(), GRAPH_NODE_TO_PROPERTY));
    }

    @Override
    public Iterator getProperties() {
        if (properties == null || properties.isEmpty())
            return ImmutableSet.of().iterator();
        return Iterators.unmodifiableIterator(Iterators.transform(properties.values().iterator(), GRAPH_NODE_TO_PROPERTY));
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof Element)) return false;
        Element that = (Element) o;
        // we compare only ids to be consistent with Tinkerpop
        return Objects.equal(getId(), that.getId());
    }

    @Override
    public int hashCode() {
        return Objects.hashCode(getId());
    }

    @Override
    public String toString() {
        return "DefaultElement{" +
                "id=" + id +
                ", label='" + label + '\'' +
                ", properties=" + properties +
                '}';
    }

}