Add test for ExportBufferManager

neo4j · Oct 9, 2024 · 2c70135 · 2c70135
1 parent 170afaf
commit 2c70135
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/core-utils/src/main/java/org/neo4j/gds/utils/ExceptionUtil.java b/core-utils/src/main/java/org/neo4j/gds/utils/ExceptionUtil.java
@@ -31,6 +31,7 @@
 import java.util.function.Consumer;
 import java.util.function.Function;
 import java.util.function.Supplier;
+import java.util.stream.Stream;
 
 public final class ExceptionUtil {
 
@@ -95,6 +96,13 @@ public static <E extends Exception> void closeAll(
         closeAll(handler, Arrays.asList(closeables));
     }
 
+    public static <E extends Exception> void closeAll(
+        CheckedConsumer<Exception, E> handler,
+        Stream<AutoCloseable> closeables
+    ) throws E {
+        closeAll(handler, closeables.iterator());
+    }
+
     public static <E extends Exception> void closeAll(
         CheckedConsumer<Exception, E> handler,
         Iterable<? extends AutoCloseable> closeables