Demonstrates loading a graph from Neo4j into CAPS and running a GraphX algorithm

Neo4jGraphXExample.scala

// 1) Create CAPS session
implicit val session: CAPSSession = CAPSSession.local()

// 2) Load a graph from a running Neo4j instance.
implicit val neo4jConfig = Neo4jConfig(uri = URI.create("bolt://localhost"), password = Some("password"))

// 3) Create a data source
val neo4jSource = new Neo4jPropertyGraphDataSource(neo4jConfig)

// 4) Load the whole Neo4j DB
val graph = neo4jSource.graph(GraphName("graph"))

// 5) Get nodes for GraphX graph construction
val nodes = graph.cypher(
    """|MATCH (n:Person)
       |RETURN id(n), n.name""".stripMargin)
val rels = graph.cypher(
    """|MATCH (:Person)-[r]->(:Person)
       |RETURN startNode(r), endNode(r)
    """.stripMargin)

// 4) Create GraphX compatible RDDs from nodes and relationships
val graphXNodeRDD = nodes.getRecords.asDataFrame.rdd.map(row => row.getLong(0) -> row.getString(1))
val graphXRelRDD = rels.getRecords.asDataFrame.rdd.map(row => Edge(row.getLong(0), row.getLong(1), ()))

// 5) Compute Page Rank via GraphX
val graph = Graph(graphXNodeRDD, graphXRelRDD)
val ranks = graph.pageRank(0.0001).vertices

// 6) Print Page Ranks
ranks.show()