improve proto conversion batching

ShiftLeftSecurity · Jan 9, 2025 · ebb742d · ebb742d
1 parent 9ef1ccb
commit ebb742d
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