Adding index

Fraunhofer-AISEC · Dec 10, 2024 · 0ebe7a4 · 0ebe7a4
1 parent 26a887f
commit 0ebe7a4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/cpg-neo4j/src/main/kotlin/de/fraunhofer/aisec/cpg/v2/Persistence.kt b/cpg-neo4j/src/main/kotlin/de/fraunhofer/aisec/cpg/v2/Persistence.kt
@@ -67,6 +67,10 @@ fun TranslationResult.persist() {
     val nodes = this@persist.nodes
     val edges = this@persist.allEdges<Edge<*>>()
 
+    neo4jSession.executeWrite { tx ->
+        tx.run("CREATE INDEX IF NOT EXISTS FOR (n:Node) ON (n.id)").consume()
+    }
+
     log.info("Persisting {} nodes", nodes.size)
     nodes.persist()
 
@@ -117,8 +121,8 @@ private fun Collection<Edge<*>>.persist() {
                 tx.run(
                         """
             UNWIND ${'$'}props AS map
-            MATCH (s {id: map.startId})
-            MATCH (e {id: map.endId})
+            MATCH (s:Node {id: map.startId})
+            MATCH (e:Node {id: map.endId})
             CREATE (s)-[r:${it.key} {}]->(e)
             """
                             .trimIndent(),

diff --git a/cpg-neo4j/src/test/kotlin/de/fraunhofer/aisec/cpg/v2/TestPersistence.kt b/cpg-neo4j/src/test/kotlin/de/fraunhofer/aisec/cpg/v2/TestPersistence.kt
@@ -26,6 +26,7 @@
 package de.fraunhofer.aisec.cpg.v2
 
 import de.fraunhofer.aisec.cpg.frontends.python.PythonLanguage
+import de.fraunhofer.aisec.cpg.helpers.Benchmark
 import de.fraunhofer.aisec.cpg.test.analyze
 import de.fraunhofer.aisec.cpg_vis_neo4j.createTranslationResult
 import java.nio.file.Path
@@ -60,7 +61,9 @@ class TestPersistence {
                 it.failOnError(false)
             }
 
+        val bench = Benchmark(this.javaClass, "Persist")
         neo4jSession.executeWrite { tx -> tx.run("MATCH (n) DETACH DELETE n").consume() }
         result.persist()
+        bench.stop()
     }
 }