HDDS-10502. Improve generic type of IOUtils.close (apache#6361)

sarvekshayr · Mar 12, 2024 · 6c95fc9 · 6c95fc9
1 parent d68ea97
commit 6c95fc9
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 14 deletions.
diff --git a/hadoop-hdds/common/src/main/java/org/apache/hadoop/hdds/utils/IOUtils.java b/hadoop-hdds/common/src/main/java/org/apache/hadoop/hdds/utils/IOUtils.java
@@ -67,7 +67,7 @@ public static void close(Logger logger, AutoCloseable... closeables) {
    * Close each argument, catching exceptions and logging them as error.
    */
   public static void close(Logger logger,
-      Collection<AutoCloseable> closeables) {
+      Collection<? extends AutoCloseable> closeables) {
     if (closeables == null) {
       return;
     }
@@ -94,7 +94,7 @@ public static void closeQuietly(AutoCloseable... closeables) {
   /**
    * Close each argument, swallowing exceptions.
    */
-  public static void closeQuietly(Collection<AutoCloseable> closeables) {
+  public static void closeQuietly(Collection<? extends AutoCloseable> closeables) {
     close(null, closeables);
   }
 }
diff --git a/hadoop-ozone/ozone-manager/src/main/java/org/apache/hadoop/ozone/om/ListIterator.java b/hadoop-ozone/ozone-manager/src/main/java/org/apache/hadoop/ozone/om/ListIterator.java
@@ -342,7 +342,7 @@ public HeapEntry next() {
     }
 
     public void close() throws IOException {
-      iterators.forEach(IOUtils::closeQuietly);
+      IOUtils.closeQuietly(iterators);
     }
   }
 }
diff --git a/...ne-manager/src/test/java/org/apache/hadoop/ozone/om/snapshot/TestSnapshotDiffManager.java b/...ne-manager/src/test/java/org/apache/hadoop/ozone/om/snapshot/TestSnapshotDiffManager.java
@@ -391,13 +391,8 @@ public OmSnapshot load(@Nonnull UUID key) {
   @AfterEach
   public void tearDown() {
     IOUtils.closeQuietly(snapshotDiffManager);
-    if (columnFamilyHandles != null) {
-      columnFamilyHandles.forEach(IOUtils::closeQuietly);
-    }
-
-    IOUtils.closeQuietly(db);
-    IOUtils.closeQuietly(dbOptions);
-    IOUtils.closeQuietly(columnFamilyOptions);
+    IOUtils.closeQuietly(columnFamilyHandles);
+    IOUtils.closeQuietly(db, dbOptions, columnFamilyOptions);
   }
 
   private OmSnapshot getMockedOmSnapshot(UUID snapshotId) {

diff --git a/hadoop-ozone/tools/src/main/java/org/apache/hadoop/ozone/debug/ReadReplicas.java b/hadoop-ozone/tools/src/main/java/org/apache/hadoop/ozone/debug/ReadReplicas.java
@@ -216,8 +216,7 @@ private void downloadReplicasAndCreateManifest(
       blockJson.add(JSON_PROPERTY_BLOCK_REPLICAS, replicasJson);
       blocks.add(blockJson);
 
-      blockReplicasWithoutChecksum.values()
-          .forEach(each -> IOUtils.close(LOG, each));
+      IOUtils.close(LOG, blockReplicasWithoutChecksum.values());
     }
   }
 

diff --git a/hadoop-ozone/tools/src/main/java/org/apache/hadoop/ozone/freon/DatanodeSimulator.java b/hadoop-ozone/tools/src/main/java/org/apache/hadoop/ozone/freon/DatanodeSimulator.java
@@ -17,7 +17,6 @@
 package org.apache.hadoop.ozone.freon;
 
 
-import org.apache.commons.io.IOUtils;
 import org.apache.commons.lang3.RandomUtils;
 import org.apache.hadoop.conf.Configuration;
 import org.apache.hadoop.hdds.DatanodeVersion;
@@ -43,6 +42,7 @@
 import org.apache.hadoop.hdds.upgrade.HDDSLayoutVersionManager;
 import org.apache.hadoop.hdds.utils.HAUtils;
 import org.apache.hadoop.hdds.utils.HddsServerUtil;
+import org.apache.hadoop.hdds.utils.IOUtils;
 import org.apache.hadoop.hdds.utils.LegacyHadoopConfigurationSource;
 import org.apache.hadoop.io.retry.RetryPolicies;
 import org.apache.hadoop.io.retry.RetryPolicy;
@@ -186,7 +186,7 @@ public Void call() throws Exception {
           } catch (InterruptedException e) {
             throw new RuntimeException(e);
           }
-          scmClients.values().forEach(IOUtils::closeQuietly);
+          IOUtils.closeQuietly(scmClients.values());
           IOUtils.closeQuietly(reconClient);
           LOGGER.info("Successfully closed all the used resources");
           saveDatanodesToFile();