Fix S3DataSegmentPuller retry bug

This commit is contained in:
Gian Merlino 2013-09-25 07:44:17 -07:00
parent 9b57b4e1f3
commit 45e22d98f1
1 changed files with 10 additions and 1 deletions

View File

@ -110,7 +110,6 @@ public class S3DataSegmentPuller implements DataSegmentPuller
return null; return null;
} }
catch (IOException e) { catch (IOException e) {
FileUtils.deleteDirectory(outDir);
throw new IOException(String.format("Problem decompressing object[%s]", s3Obj), e); throw new IOException(String.format("Problem decompressing object[%s]", s3Obj), e);
} }
finally { finally {
@ -125,6 +124,16 @@ public class S3DataSegmentPuller implements DataSegmentPuller
); );
} }
catch (Exception e) { catch (Exception e) {
try {
FileUtils.deleteDirectory(outDir);
} catch (IOException ioe) {
log.warn(
ioe,
"Failed to remove output directory for segment[%s] after exception: %s",
segment.getIdentifier(),
outDir
);
}
throw new SegmentLoadingException(e, e.getMessage()); throw new SegmentLoadingException(e, e.getMessage());
} }
} }