RestService.java

package org.neo4j.smack.api;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;

import javax.ws.rs.core.MediaType;

import org.neo4j.server.rest.paging.LeaseManager;
import org.neo4j.server.rest.paging.RealClock;
import org.neo4j.server.rest.repr.BadInputException;
import org.neo4j.server.rest.repr.ExtensionInjector;
import org.neo4j.server.rest.repr.OutputFormat;
import org.neo4j.server.rest.repr.RepresentationFormatRepository;
import org.neo4j.smack.pipeline.database.event.Invocation;

/**
 * @author mh
 * @since 11.12.11
 */
public class RestService 
{
    private static final LeaseManager leaseManager = new LeaseManager(new RealClock());
    public static final String NODE_ID_NAME = "nodeId";
    protected static final String PATH_NODES = "node";
    protected static final String PATH_NODE = PATH_NODES + "/{" + NODE_ID_NAME + "}";
    public static final String RELATIONSHIP_ID_NAME = "relationshipId";
    protected static final String PATH_NODE_RELATIONSHIPS = PATH_NODE + "/relationships";
    protected static final String PATH_RELATIONSHIP = "relationship/{" + RELATIONSHIP_ID_NAME + "}";
    RepresentationFormatRepository repository = new RepresentationFormatRepository(new ExtensionInjector() {
        @Override
        public Map<String, List<String>> getExensionsFor(Class<?> aClass) {
            return Collections.emptyMap();
        }
    });
    
    private final String dataPath;

    public RestService(String dataPath) {
        this.dataPath = dataPath;
    }

    long extractId(Object uri) throws BadInputException {
        try {
            return Long.parseLong(uri.toString().substring(uri.toString().lastIndexOf("/") + 1));
        } catch (NumberFormatException ex) {
            throw new BadInputException(ex);
        } catch (NullPointerException ex) {
            throw new BadInputException(ex);
        }
    }

    protected OutputFormat createOutputFormat(Invocation invocation) throws URISyntaxException {
        final URI uri = new URI(dataPath);
        return repository.outputFormat(Arrays.asList(MediaType.APPLICATION_JSON_TYPE), uri);
    }

    protected Long getNodeId(Invocation invocation) {
        return getLongParameter(invocation, NODE_ID_NAME);
    }
    protected Long getRelationshipId(Invocation invocation) {
        return getLongParameter(invocation, RELATIONSHIP_ID_NAME);
    }

    protected Long getLongParameter(Invocation invocation, String key) {
        return invocation.getLongParameter(key, -1l);
    }
    protected Long getLongParameter(Invocation invocation, String key, Long defaultValue) {
        final Long value = getLongParameter(invocation, key);
        return value == null ? defaultValue : value;
    }

    protected String getParameter(Invocation invocation, String key) {
        return invocation.getPathVariables().getParameter(key);
    }

    protected Long getId(Invocation invocation) {
        return getLongParameter(invocation, "id");
    }


    protected String getKey(Invocation invocation) {
        return getParameter(invocation, "key");
    }

    @SuppressWarnings("unchecked")
    protected Map<String,Object> readMap(Invocation invocation) {
        return invocation.getContent(Map.class);
    }

    @SuppressWarnings("serial")
    public static class AmpersandSeparatedCollection extends LinkedHashSet<String> {
        public AmpersandSeparatedCollection(String path) {
            for (String e : path.split("&")) {
                if (e.trim()
                        .length() > 0) {
                    add(e);
                }
            }
        }
    }
}