org.neo4j.ogm.drivers.embedded.response.EmbeddedEntityAdapter Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2002-2016 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This product is licensed to you under the Apache License, Version 2.0 (the "License").
 * You may not use this product except in compliance with the License.
 *
 * This product may include a number of subcomponents with
 * separate copyright notices and license terms. Your use of the source
 * code for these subcomponents is subject to the terms and
 *  conditions of the subcomponent's license, as noted in the LICENSE file.
 */

package org.neo4j.ogm.drivers.embedded.response;

import org.neo4j.graphdb.*;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Helper methods for embedded graph entities
 *
 * @author Luanne Misquitta
 */
public class EmbeddedEntityAdapter {

	public boolean isPath(Object value) {
		return value instanceof Path;
	}

	public boolean isNode(Object value) {
		return value instanceof Node;
	}

	public boolean isRelationship(Object value) {
		return value instanceof Relationship;
	}

	public long nodeId(Object node) {
		return ((Node) node).getId();
	}

	public List labels(Object node) {
		List labels = new ArrayList<>();
		for (Label label : ((Node) node).getLabels()) {
			labels.add(label.name());
		}
		return labels;
	}

	public long relationshipId(Object relationship) {
		return ((Relationship) relationship).getId();
	}

	public String relationshipType(Object relationship) {
		return ((Relationship) relationship).getType().name();
	}

	public Long startNodeId(Object relationship) {
		return ((Relationship) relationship).getStartNode().getId();
	}

	public Long endNodeId(Object relationship) {
		return ((Relationship) relationship).getEndNode().getId();
	}

	public Map properties(Object container) {
		return getAllProperties((PropertyContainer) container);
	}

	public List