Enable advance leak detection.

apache · Sep 29, 2024 · 0f4b61e · 0f4b61e
1 parent 9159532
commit 0f4b61e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/ratis-common/src/main/java/org/apache/ratis/util/LeakDetector.java b/ratis-common/src/main/java/org/apache/ratis/util/LeakDetector.java
@@ -85,13 +85,19 @@ synchronized void assertNoLeaks() {
       }
 
       int n = 0;
-      for(LeakTracker tracker : set) {
+      for (LeakTracker tracker : set) {
         if (tracker.reportLeak() != null) {
           n++;
         }
       }
-      final int leaks = n;
-      Preconditions.assertTrue(n == 0, () -> "#leaks = " + leaks + ", set.size = " + set.size());
+      assertNoLeaks(n);
+    }
+
+    synchronized void assertNoLeaks(int leaks) {
+      Preconditions.assertTrue(leaks == 0, () -> {
+        final int size = set.size();
+        return "#leaks = " + leaks + (leaks == size? "==" : "!=") + " set.size = " + size;
+      });
     }
   }
 

diff --git a/ratis-grpc/src/test/java/org/apache/ratis/grpc/MiniRaftClusterWithGrpc.java b/ratis-grpc/src/test/java/org/apache/ratis/grpc/MiniRaftClusterWithGrpc.java
@@ -51,7 +51,7 @@ public MiniRaftClusterWithGrpc newCluster(String[] ids, String[] listenerIds, Ra
   };
 
   static {
-    ReferenceCountedObject.enableLeakDetection();
+    ReferenceCountedObject.enableLeakDetection(true);
   }
 
   public interface FactoryGet extends Factory.Get<MiniRaftClusterWithGrpc> {