trigger clean dir when unzip fails with IOException (#6178)

apache · Aug 17, 2018 · 78fc5b2 · 78fc5b2
1 parent b0cf8d0
commit 78fc5b2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/...rib/azure-extensions/src/test/java/io/druid/storage/azure/AzureDataSegmentPullerTest.java b/...rib/azure-extensions/src/test/java/io/druid/storage/azure/AzureDataSegmentPullerTest.java
@@ -90,12 +90,10 @@ public void testDeleteOutputDirectoryWhenErrorIsRaisedPullingSegmentFiles()
     final File outDir = Files.createTempDirectory("druid").toFile();
     try {
       expect(azureStorage.getBlobInputStream(containerName, blobPath)).andThrow(
-          new StorageException(
+          new URISyntaxException(
               "error",
               "error",
-              404,
-              null,
-              null
+              404
           )
       );
 

diff --git a/java-util/src/main/java/io/druid/java/util/common/CompressionUtils.java b/java-util/src/main/java/io/druid/java/util/common/CompressionUtils.java
@@ -169,6 +169,9 @@ public static FileUtils.FileCopyResult unzip(
             DEFAULT_RETRY_COUNT
         );
       }
+      catch (IOException e) {
+        throw e;
+      }
       catch (Exception e) {
         throw Throwables.propagate(e);
       }