improve proto conversion batching (#1798)

ShiftLeftSecurity · Jan 10, 2025 · 71180f2 · 71180f2
1 parent f04c034
commit 71180f2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/...opertygraph/src/main/scala/io/shiftleft/codepropertygraph/cpgloading/ProtoCpgLoader.scala b/...opertygraph/src/main/scala/io/shiftleft/codepropertygraph/cpgloading/ProtoCpgLoader.scala
@@ -43,9 +43,7 @@ object ProtoCpgLoader {
     val cpg                               = openOrCreateCpg(storagePath)
 
     // first pass: add the raw nodes without any properties or edges
-    protoCpgs().foreach { cpgProto =>
-      addNodesRaw(nodesIter(cpgProto), cpg.graph, protoToGraphNodeMappings)
-    }
+    addNodesRaw(protoCpgs().flatMap(cpgProto => nodesIter(cpgProto)), cpg.graph, protoToGraphNodeMappings)
 
     // second pass: set node properties and add edges
     val diffGraph = Cpg.newDiffGraphBuilder