org.neo4j.rest.graphdb.traversal.RestPathParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-data-neo4j-rest Show documentation
pring Data Neo4j Wrapper for the Neo4j REST API, provides a Graph Database proxy for the remote invocation.
There is a newer version: 3.4.6.RELEASE
Show newest version
/**
 * Copyright (c) 2002-2013 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This file is part of Neo4j.
 *
 * Neo4j is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */
package org.neo4j.rest.graphdb.traversal;

import java.util.Collection;
import java.util.List;
import java.util.Map;

import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.Path;
import org.neo4j.graphdb.Relationship;
import org.neo4j.helpers.collection.IterableWrapper;
import org.neo4j.helpers.collection.IteratorUtil;
import org.neo4j.rest.graphdb.RequestResult;
import org.neo4j.rest.graphdb.RestAPI;
import org.neo4j.rest.graphdb.converter.RestResultConverter;
import org.neo4j.rest.graphdb.converter.TypeInformation;
import org.neo4j.rest.graphdb.entity.RestNode;
import org.neo4j.rest.graphdb.entity.RestRelationship;

/**
 * @author Michael Hunger
 * @since 03.02.11
 */
public class RestPathParser implements RestResultConverter {

     private RestAPI restAPI;

    public RestPathParser(RestAPI restAPI) {
        this.restAPI = restAPI;
    }

    public static Path parse(Map path, final RestAPI restApi) {
         TypeInformation typeInfo = new TypeInformation(path.get("nodes"));
        RestPathParser restPathParser = new RestPathParser(restApi);
        if (restPathParser.isFullPath(typeInfo)){
            return restPathParser.parseFullPath(path, restApi);
         }
         if (restPathParser.isPath(typeInfo)){
             return restPathParser.parsePath(path, restApi);
         }

         throw new IllegalArgumentException("params map contained illegal type "+typeInfo.getGenericArguments()[0]);
    }

    private  boolean isPath(TypeInformation typeInfo) {
        return typeInfo.getGenericArguments()[0].equals(String.class);
    }

    private  boolean isFullPath(TypeInformation typeInfo) {
        return Map.class.isAssignableFrom(typeInfo.getGenericArguments()[0]);
    }


    private Path parseFullPath(Map path, final RestAPI restApi) {
        final Collection> nodesData = (Collection>) path.get("nodes");
        final Collection> relationshipsData = (Collection>) path.get("relationships");
        final Map lastRelationshipData = lastElementMap(relationshipsData);
        final Map startData = (Map) path.get("start");
        final Map endData = (Map) path.get("end");
        final Integer length = (Integer) path.get("length");

        RestRelationship lastRelationship = lastRelationshipData == null ? null : new RestRelationship(lastRelationshipData, restApi);
        return new SimplePath(
                new RestNode(startData,restApi),
                new RestNode(endData,restApi),
                lastRelationship,
                length,
                new IterableWrapper>(nodesData) {
                    @Override
                    protected Node underlyingObjectToObject(Map data) {
                        return new RestNode(data,restApi);
                    }
                },
                new IterableWrapper>(relationshipsData) {
                    @Override
                    protected Relationship underlyingObjectToObject(Map data) {
                        return new RestRelationship(data,restApi);
                    }
                });
    }

    private Path parsePath(Map path, final RestAPI restApi){
        final Collection nodesData = (Collection) path.get("nodes");
        final Collection relationshipsData = (Collection) path.get("relationships");
        final String lastRelationshipData = lastElement(relationshipsData);
        final String startData = (String) path.get("start");
        final String endData = (String) path.get("end");
        final Integer length = (Integer) path.get("length");
        RestRelationship lastRelationship = lastRelationshipData == null ? null : new RestRelationship(lastRelationshipData, restApi);
        return new SimplePath(
                new RestNode(startData,restApi),
                new RestNode(endData,restApi),
                lastRelationship,
                length,
                new IterableWrapper(nodesData) {
                    @Override
                    protected Node underlyingObjectToObject(String data) {
                        return new RestNode(data,restApi);
                    }
                },
                new IterableWrapper(relationshipsData) {
                    @Override
                    protected Relationship underlyingObjectToObject(String data) {
                        return new RestRelationship(data,restApi);
                    }
                });
    }

    private String lastElement(Collection collection){
       return IteratorUtil.lastOrNull(collection);
    }

    private Map lastElementMap(Collection> collection) {
        if (collection.isEmpty()) return null;
        if (collection instanceof List) {
            List> list = (List>) collection;
            return list.get(list.size()-1);
        }
        Map result = null;
        for (Map value : collection) {
            result=value;
        }
        return result;
    }

    @Override
    public Object convertFromRepresentation(RequestResult value) {
        return parse(value.toMap(), this.restAPI);
    }
}