Optimize the performance of CHA algorithm

pascal-lab · Apr 3, 2024 · 083f0cc · 083f0cc
1 parent 5225e84
commit 083f0cc
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/src/main/java/pascal/taie/analysis/graph/callgraph/CHABuilder.java b/src/main/java/pascal/taie/analysis/graph/callgraph/CHABuilder.java
@@ -70,17 +70,18 @@ private CallGraph<Invoke, JMethod> buildCallGraph(JMethod entry) {
         workList.add(entry);
         while (!workList.isEmpty()) {
             JMethod method = workList.poll();
-            callGraph.addReachableMethod(method);
-            callGraph.callSitesIn(method).forEach(invoke -> {
-                Set<JMethod> callees = resolveCalleesOf(invoke);
-                callees.forEach(callee -> {
-                    if (!callGraph.contains(callee)) {
-                        workList.add(callee);
-                    }
-                    callGraph.addEdge(new Edge<>(
-                            CallGraphs.getCallKind(invoke), invoke, callee));
+            if (callGraph.addReachableMethod(method)) {
+                callGraph.callSitesIn(method).forEach(invoke -> {
+                    Set<JMethod> callees = resolveCalleesOf(invoke);
+                    callees.forEach(callee -> {
+                        if (!callGraph.contains(callee)) {
+                            workList.add(callee);
+                        }
+                        callGraph.addEdge(new Edge<>(
+                                CallGraphs.getCallKind(invoke), invoke, callee));
+                    });
                 });
-            });
+            }
         }
         return callGraph;
     }