Initial TitanGods example

669b8d4d · Jeffrey Phillips Freeman · 669b8d4d · 669b8d4d · 669b8d4d
Commit 669b8d4d authored 10 years ago by Jeffrey Phillips Freeman
--- a/pom.xml
+++ b/pom.xml
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+
+    <modelVersion>4.0.0</modelVersion>
+
+    <groupId>com.syncleus.titandbexamples</groupId>
+    <artifactId>titandbexamples</artifactId>
+    <version>1.0-SNAPSHOT</version>
+    <packaging>jar</packaging>
+
+    <name>TitanDbExamples</name>
+    <inceptionYear>2014</inceptionYear>
+
+    <licenses>
+        <license>
+            <name>Open Source Community License - Type C version 1.0 or later</name>
+            <url>http://wiki.syncleus.com/images/3/3b/Oscl-c.txt</url>
+            <distribution>repo</distribution>
+        </license>
+    </licenses>
+
+    <properties>
+        <project.build.encoding>UTF-8</project.build.encoding>
+        <project.build.sourceEncoding>${project.build.encoding}</project.build.sourceEncoding>
+        <project.reporting.outputEncoding>${project.build.encoding}</project.reporting.outputEncoding>
+        <project.build.resourceEncoding>${project.build.encoding}</project.build.resourceEncoding>
+    </properties>
+
+    <reporting>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-jxr-plugin</artifactId>
+                <version>2.3</version>
+                <configuration>
+                    <linkJavadoc>true</linkJavadoc>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.codehaus.mojo</groupId>
+                <artifactId>cobertura-maven-plugin</artifactId>
+                <version>2.6</version>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-javadoc-plugin</artifactId>
+                <version>2.10.1</version>
+            </plugin>
+        </plugins>
+    </reporting>
+
+    <build>
+        <defaultGoal>package</defaultGoal>
+
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <version>3.1</version>
+                <configuration>
+                    <source>1.7</source>
+                    <target>1.7</target>
+                    <encoding>${project.build.sourceEncoding}</encoding>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-resources-plugin</artifactId>
+                <version>2.6</version>
+                <configuration>
+                    <encoding>${project.build.resourceEncoding}</encoding>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-surefire-plugin</artifactId>
+                <version>2.16</version>
+                <configuration>
+                    <argLine>-enableassertions</argLine>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.codehaus.mojo</groupId>
+                <artifactId>cobertura-maven-plugin</artifactId>
+                <version>2.6</version>
+                <executions>
+                    <execution>
+                        <phase>verify</phase>
+                        <goals>
+                            <goal>cobertura</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
+        </plugins>
+    </build>
+
+    <dependencies>
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>4.8.1</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>com.thinkaurelius.titan</groupId>
+            <artifactId>titan-core</artifactId>
+            <version>0.4.4</version>
+        </dependency>
+        <dependency>
+            <groupId>com.thinkaurelius.titan</groupId>
+            <artifactId>titan-berkeleyje</artifactId>
+            <version>0.4.4</version>
+        </dependency>
+        <dependency>
+            <groupId>com.thinkaurelius.titan</groupId>
+            <artifactId>titan-es</artifactId>
+            <version>0.4.4</version>
+        </dependency>
+    </dependencies>
+</project>
--- a/src/main/java/com/syncleus/titangraph/example/TitanGods.java
+++ b/src/main/java/com/syncleus/titangraph/example/TitanGods.java
+package com.syncleus.titangraph.example;
+
+import com.thinkaurelius.titan.core.TitanFactory;
+import com.thinkaurelius.titan.core.TitanGraph;
+import com.thinkaurelius.titan.core.TitanKey;
+import com.thinkaurelius.titan.core.attribute.Geoshape;
+import com.thinkaurelius.titan.graphdb.configuration.GraphDatabaseConfiguration;
+import com.tinkerpop.blueprints.Edge;
+import com.tinkerpop.blueprints.Vertex;
+import com.tinkerpop.blueprints.util.ElementHelper;
+import org.apache.commons.configuration.BaseConfiguration;
+import org.apache.commons.configuration.Configuration;
+
+import java.io.File;
+
+import static com.thinkaurelius.titan.graphdb.configuration.GraphDatabaseConfiguration.INDEX_BACKEND_KEY;
+import static com.thinkaurelius.titan.graphdb.configuration.GraphDatabaseConfiguration.STORAGE_DIRECTORY_KEY;
+
+
+/**
+ * Example Graph factory that creates a {@link TitanGraph} based on roman mythology.
+ * Used in the documentation examples and tutorials.
+ *
+ * @author Marko A. Rodriguez (http://markorodriguez.com)
+ */
+public class TitanGods {
+
+    public static final String INDEX_NAME = "search";
+
+
+    public static TitanGraph create(final String directory) {
+        BaseConfiguration config = new BaseConfiguration();
+        Configuration storage = config.subset(GraphDatabaseConfiguration.STORAGE_NAMESPACE);
+        // configuring local backend
+        storage.setProperty(GraphDatabaseConfiguration.STORAGE_BACKEND_KEY, "local");
+        storage.setProperty(GraphDatabaseConfiguration.STORAGE_DIRECTORY_KEY, directory);
+        // configuring elastic search index
+        Configuration index = storage.subset(GraphDatabaseConfiguration.INDEX_NAMESPACE).subset(INDEX_NAME);
+        index.setProperty(INDEX_BACKEND_KEY, "elasticsearch");
+        index.setProperty("local-mode", true);
+        index.setProperty("client-only", false);
+        index.setProperty(STORAGE_DIRECTORY_KEY, directory + File.separator + "es");
+
+        TitanGraph graph = TitanFactory.open(config);
+        if( !graph.getVertices().iterator().hasNext() )
+            TitanGods.load(graph);
+        return graph;
+    }
+
+    public static void load(final TitanGraph graph) {
+
+        graph.makeKey("name").dataType(String.class).indexed(Vertex.class).unique().make();
+        graph.makeKey("age").dataType(Integer.class).indexed(INDEX_NAME, Vertex.class).make();
+        graph.makeKey("type").dataType(String.class).make();
+
+        final TitanKey time = graph.makeKey("time").dataType(Integer.class).make();
+        final TitanKey reason = graph.makeKey("reason").dataType(String.class).indexed(INDEX_NAME, Edge.class).make();
+        graph.makeKey("place").dataType(Geoshape.class).indexed(INDEX_NAME, Edge.class).make();
+
+        graph.makeLabel("father").manyToOne().make();
+        graph.makeLabel("mother").manyToOne().make();
+        graph.makeLabel("battled").sortKey(time).make();
+        graph.makeLabel("lives").signature(reason).make();
+        graph.makeLabel("pet").make();
+        graph.makeLabel("brother").make();
+
+        graph.commit();
+
+        // vertices
+
+        Vertex saturn = graph.addVertex(null);
+        saturn.setProperty("name", "saturn");
+        saturn.setProperty("age", 10000);
+        saturn.setProperty("type", "titan");
+
+        Vertex sky = graph.addVertex(null);
+        ElementHelper.setProperties(sky, "name", "sky", "type", "location");
+
+        Vertex sea = graph.addVertex(null);
+        ElementHelper.setProperties(sea, "name", "sea", "type", "location");
+
+        Vertex jupiter = graph.addVertex(null);
+        ElementHelper.setProperties(jupiter, "name", "jupiter", "age", 5000, "type", "god");
+
+        Vertex neptune = graph.addVertex(null);
+        ElementHelper.setProperties(neptune, "name", "neptune", "age", 4500, "type", "god");
+
+        Vertex hercules = graph.addVertex(null);
+        ElementHelper.setProperties(hercules, "name", "hercules", "age", 30, "type", "demigod");
+
+        Vertex alcmene = graph.addVertex(null);
+        ElementHelper.setProperties(alcmene, "name", "alcmene", "age", 45, "type", "human");
+
+        Vertex pluto = graph.addVertex(null);
+        ElementHelper.setProperties(pluto, "name", "pluto", "age", 4000, "type", "god");
+
+        Vertex nemean = graph.addVertex(null);
+        ElementHelper.setProperties(nemean, "name", "nemean", "type", "monster");
+
+        Vertex hydra = graph.addVertex(null);
+        ElementHelper.setProperties(hydra, "name", "hydra", "type", "monster");
+
+        Vertex cerberus = graph.addVertex(null);
+        ElementHelper.setProperties(cerberus, "name", "cerberus", "type", "monster");
+
+        Vertex tartarus = graph.addVertex(null);
+        ElementHelper.setProperties(tartarus, "name", "tartarus", "type", "location");
+
+        // edges
+
+        jupiter.addEdge("father", saturn);
+        jupiter.addEdge("lives", sky).setProperty("reason", "loves fresh breezes");
+        jupiter.addEdge("brother", neptune);
+        jupiter.addEdge("brother", pluto);
+
+        neptune.addEdge("lives", sea).setProperty("reason", "loves waves");
+        neptune.addEdge("brother", jupiter);
+        neptune.addEdge("brother", pluto);
+
+        hercules.addEdge("father", jupiter);
+        hercules.addEdge("mother", alcmene);
+        ElementHelper.setProperties(hercules.addEdge("battled", nemean), "time", 1, "place", Geoshape.point(38.1f, 23.7f));
+        ElementHelper.setProperties(hercules.addEdge("battled", hydra), "time", 2, "place", Geoshape.point(37.7f, 23.9f));
+        ElementHelper.setProperties(hercules.addEdge("battled", cerberus), "time", 12, "place", Geoshape.point(39f, 22f));
+
+        pluto.addEdge("brother", jupiter);
+        pluto.addEdge("brother", neptune);
+        pluto.addEdge("lives", tartarus).setProperty("reason", "no fear of death");
+        pluto.addEdge("pet", cerberus);
+
+        cerberus.addEdge("lives", tartarus);
+
+        // commit the transaction to disk
+        graph.commit();
+    }
+}
\ No newline at end of file
--- a/src/test/java/com/syncleus/titangraph/example/TitanGodsTest.java
+++ b/src/test/java/com/syncleus/titangraph/example/TitanGodsTest.java
+package com.syncleus.titangraph.example;
+
+import com.thinkaurelius.titan.core.TitanGraph;
+import com.tinkerpop.blueprints.Vertex;
+import junit.framework.Test;
+import junit.framework.TestCase;
+import junit.framework.TestSuite;
+
+/**
+ * Unit test for simple App.
+ */
+public class TitanGodsTest
+        extends TestCase {
+    /**
+     * Create the test case
+     *
+     * @param testName name of the test case
+     */
+    public TitanGodsTest(String testName) {
+        super(testName);
+    }
+
+    /**
+     * @return the suite of tests being tested
+     */
+    public static Test suite() {
+        return new TestSuite(TitanGodsTest.class);
+    }
+
+    /**
+     * Rigourous Test :-)
+     */
+    public void testApp() {
+        TitanGraph godGraph = TitanGods.create("./target/TitanTestDB");
+        Iterable<Vertex> skyVertices = godGraph.getVertices("name", "sky");
+        assertTrue("no sky vertices found", skyVertices.iterator().hasNext());
+
+        for( Vertex skyVertex : skyVertices ) {
+            skyVertex.
+        }
+    }
+}