com.datastax.driver.dse.graph.ObjectGraphNode 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 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.Objects;
import com.google.common.reflect.TypeToken;

import java.util.Iterator;
import java.util.List;
import java.util.Map;

@SuppressWarnings({"unchecked", "rawtypes"})
public class ObjectGraphNode implements GraphNode {

    private final Object delegate;

    public ObjectGraphNode(Object delegate) {
        this.delegate = delegate;
    }

    @Override
    public boolean isNull() {
        return delegate == null;
    }

    @Override
    public boolean isObject() {
        return delegate instanceof Map;
    }

    @Override
    public boolean isArray() {
        return delegate instanceof List;
    }

    @Override
    public boolean isValue() {
        return !(isArray() || isObject());
    }

    @Override
    public Iterator fieldNames() {
        return ((Map)delegate).keySet().iterator();
    }

    @Override
    public int size() {
        return isArray() ? ((List) delegate).size() : 0;
    }

    @Override
    public int asInt() {
        return (Integer) delegate;
    }

    @Override
    public boolean asBoolean() {
        return (Boolean) delegate;
    }

    @Override
    public long asLong() {
        return (Long) delegate;
    }

    @Override
    public double asDouble() {
        return (Double) delegate;
    }

    @Override
    public String asString() {
        return (String) delegate;
    }

    @Override
    public Map asMap() {
        return (Map) delegate;
    }

    @Override
    public  T as(Class clazz) {
        return (T) delegate;
    }

    @Override
    public  T as(TypeToken type) {
        return (T) delegate;
    }

    @Override
    public GraphNode get(String fieldName) {
        if (((Map) delegate).containsKey(fieldName)) {
            return new ObjectGraphNode(((Map) delegate).get(fieldName));
        }
        return null;
    }

    @Override
    public GraphNode get(int index) {
        Object indexValue = ((List) delegate).get(index);
        if (indexValue == null) {
            return null;
        } else {
            return new ObjectGraphNode(indexValue);
        }
    }
    @Override
    public boolean isVertex() {
        return delegate instanceof Vertex;
    }

    @Override
    public boolean isEdge() {
        return delegate instanceof Edge;
    }

    @Override
    public Vertex asVertex() {
        return (Vertex) delegate;
    }

    @Override
    public Edge asEdge() {
        return (Edge) delegate;
    }

    @Override
    public Path asPath() {
        return (Path) delegate;
    }

    @Override
    public Property asProperty() {
        return (Property) delegate;
    }

    @Override
    public VertexProperty asVertexProperty() {
        return (VertexProperty) delegate;
    }

    @Override
    public String toString() {
        return this.delegate.toString();
    }

    @Override
    public boolean equals(Object other) {
        if (this == other) return true;
        // Compare each others' delegates.
        return other instanceof ObjectGraphNode &&
                Objects.equal(this.delegate, ((ObjectGraphNode) other).delegate);
    }

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

}