Java tutorial
/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ package org.apache.tinkerpop.gremlin.neo4j; import org.apache.commons.configuration.Configuration; import org.apache.tinkerpop.gremlin.AbstractGraphProvider; import org.apache.tinkerpop.gremlin.LoadGraphWith; import org.apache.tinkerpop.gremlin.neo4j.structure.Neo4jEdge; import org.apache.tinkerpop.gremlin.neo4j.structure.Neo4jElement; import org.apache.tinkerpop.gremlin.neo4j.structure.Neo4jGraph; import org.apache.tinkerpop.gremlin.neo4j.structure.Neo4jGraphVariables; import org.apache.tinkerpop.gremlin.neo4j.structure.Neo4jProperty; import org.apache.tinkerpop.gremlin.neo4j.structure.Neo4jVertex; import org.apache.tinkerpop.gremlin.neo4j.structure.Neo4jVertexProperty; import org.apache.tinkerpop.gremlin.structure.Graph; import java.io.File; import java.util.HashSet; import java.util.Random; import java.util.Set; /** * @author Stephen Mallette (http://stephen.genoprime.com) */ public abstract class AbstractNeo4jGraphProvider extends AbstractGraphProvider { private static final Set<Class> IMPLEMENTATIONS = new HashSet<Class>() { { add(Neo4jEdge.class); add(Neo4jElement.class); add(Neo4jGraph.class); add(Neo4jGraphVariables.class); add(Neo4jProperty.class); add(Neo4jVertex.class); add(Neo4jVertexProperty.class); } }; @Override public void clear(final Graph graph, final Configuration configuration) throws Exception { if (null != graph) { if (graph.tx().isOpen()) graph.tx().rollback(); graph.close(); } if (configuration.containsKey(Neo4jGraph.CONFIG_DIRECTORY)) { // this is a non-in-sideEffects configuration so blow away the directory final File graphDirectory = new File(configuration.getString(Neo4jGraph.CONFIG_DIRECTORY)); deleteDirectory(graphDirectory); } } @Override public void loadGraphData(final Graph graph, final LoadGraphWith loadGraphWith, final Class testClass, final String testName) { if (loadGraphWith != null) this.createIndices((Neo4jGraph) graph, loadGraphWith.value()); super.loadGraphData(graph, loadGraphWith, testClass, testName); } public static void dropIndices(final Neo4jGraph graph, final LoadGraphWith.GraphData graphData) { if (graphData.equals(LoadGraphWith.GraphData.GRATEFUL)) { graph.tx().readWrite(); graph.cypher("DROP INDEX ON :artist(name)").iterate(); graph.cypher("DROP INDEX ON :song(name)").iterate(); graph.cypher("DROP INDEX ON :song(songType)").iterate(); graph.cypher("DROP INDEX ON :song(performances)").iterate(); graph.tx().commit(); } else if (graphData.equals(LoadGraphWith.GraphData.MODERN)) { graph.tx().readWrite(); graph.cypher("DROP INDEX ON :person(name)").iterate(); graph.cypher("DROP INDEX ON :person(age)").iterate(); graph.cypher("DROP INDEX ON :software(name)").iterate(); graph.cypher("DROP INDEX ON :software(lang)").iterate(); graph.tx().commit(); } else if (graphData.equals(LoadGraphWith.GraphData.CLASSIC)) { graph.tx().readWrite(); graph.cypher("DROP INDEX ON :vertex(name)").iterate(); graph.cypher("DROP INDEX ON :vertex(age)").iterate(); graph.cypher("DROP INDEX ON :vertex(lang)").iterate(); graph.tx().commit(); } else { // TODO: add CREW work here. // TODO: add meta_property indices when meta_property graph is provided //throw new RuntimeException("Could not load graph with " + graphData); } } private void createIndices(final Neo4jGraph graph, final LoadGraphWith.GraphData graphData) { final Random random = new Random(); final boolean pick = random.nextBoolean(); if (graphData.equals(LoadGraphWith.GraphData.GRATEFUL)) { if (pick) { graph.tx().readWrite(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :artist(name)").iterate(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :song(name)").iterate(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :song(songType)").iterate(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :song(performances)").iterate(); graph.tx().commit(); } // else no indices } else if (graphData.equals(LoadGraphWith.GraphData.MODERN)) { if (pick) { graph.tx().readWrite(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :person(name)").iterate(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :person(age)").iterate(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :software(name)").iterate(); if (random.nextBoolean()) { graph.cypher("CREATE INDEX ON :software(lang)").iterate(); } graph.tx().commit(); } // else no indices } else if (graphData.equals(LoadGraphWith.GraphData.CLASSIC)) { if (pick) { graph.tx().readWrite(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :vertex(name)").iterate(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :vertex(age)").iterate(); if (random.nextBoolean()) graph.cypher("CREATE INDEX ON :vertex(lang)").iterate(); graph.tx().commit(); } // else no indices } else { // TODO: add CREW work here. // TODO: add meta_property indices when meta_property graph is provided //throw new RuntimeException("Could not load graph with " + graphData); } } @Override public Set<Class> getImplementations() { return IMPLEMENTATIONS; } }