fix diffgraph usage

Privado-Inc · Jul 30, 2024 · 04aa8fb · 04aa8fb
1 parent 287fd5e
commit 04aa8fb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/codepropertygraph/src/main/scala/io/shiftleft/passes/CpgPass.scala b/codepropertygraph/src/main/scala/io/shiftleft/passes/CpgPass.scala
@@ -122,7 +122,7 @@ abstract class ForkJoinParallelCpgPass[T <: AnyRef](
     var nDiff      = -1
     var nDiffT     = -1
     try {
-      val diffGraph = Cpg.newDiffGraphBuilder
+      val diffGraph = new DiffGraphBuilder
       nParts = runWithBuilder(diffGraph)
       nanosBuilt = System.nanoTime()
       nDiff = diffGraph.size()
@@ -257,7 +257,7 @@ abstract class NewStyleCpgPassBaseWithTimeout[T <: AnyRef](timeout: Long) extend
               .collect(
                 new Supplier[DiffGraphBuilder] {
                   override def get(): DiffGraphBuilder =
-                    Cpg.newDiffGraphBuilder
+                    new DiffGraphBuilder
                 },
                 new BiConsumer[DiffGraphBuilder, AnyRef] {
                   override def accept(builder: DiffGraphBuilder, part: AnyRef): Unit =
@@ -273,11 +273,11 @@ abstract class NewStyleCpgPassBaseWithTimeout[T <: AnyRef](timeout: Long) extend
             implicit val ec: scala.concurrent.ExecutionContext = scala.concurrent.ExecutionContext.global
             val stopAt                                         = System.currentTimeMillis() + timeout * 1000
             var exitByTimeout                                  = false
-            val diffGraphAccumulator                           = Cpg.newDiffGraphBuilder
+            val diffGraphAccumulator                           = new DiffGraphBuilder
 
             val futures = parts.map { part =>
               val future = Future {
-                val diffGraphBuilder = Cpg.newDiffGraphBuilder
+                val diffGraphBuilder = new DiffGraphBuilder
                 runOnPart(diffGraphBuilder, part.asInstanceOf[T])
                 diffGraphBuilder
               }