bobo-contrib: catch block to close SegmentWriter resources if an erro…

…r occurs in constructor
LinkedInAttic · Feb 10, 2012 · 41daae6 · 41daae6
1 parent 8e88ad5
commit 41daae6
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/bobo-contrib/src/main/java/com/browseengine/bobo/geosearch/index/impl/GeoSegmentWriter.java b/bobo-contrib/src/main/java/com/browseengine/bobo/geosearch/index/impl/GeoSegmentWriter.java
@@ -41,8 +41,11 @@ public GeoSegmentWriter(Set<G> tree, Directory directory, String fileName,
         this.geoRecordSerializer = geoRecordSerializer;
 
         indexOutput = directory.createOutput(fileName);
-
-        buildBTreeFromSet(tree);
+        try {
+            buildBTreeFromSet(tree);
+        } finally {
+            close();
+        }
     }
 
     public GeoSegmentWriter(int treeSize, Iterator<G> inputIterator, Directory directory, String fileName,
@@ -54,7 +57,11 @@ public GeoSegmentWriter(int treeSize, Iterator<G> inputIterator, Directory direc
 
         indexOutput = directory.createOutput(fileName);
 
-        buildBTreeFromIterator(inputIterator);
+        try {
+            buildBTreeFromIterator(inputIterator);
+        } finally {
+            close();
+        }
     }
 
     private void buildBTreeFromIterator(Iterator<G> geoIter) throws IOException {