summaryrefslogtreecommitdiff
path: root/src/main/java/com/amazon/carbonado/cursor
diff options
context:
space:
mode:
authorBrian S. O'Neill <bronee@gmail.com>2007-01-29 23:24:57 +0000
committerBrian S. O'Neill <bronee@gmail.com>2007-01-29 23:24:57 +0000
commit17f39437744585c656cf37d4515a0b115cc73cc8 (patch)
tree90f1212ac057cbdb278beea5c273e95cafe85139 /src/main/java/com/amazon/carbonado/cursor
parent27483a11b591e9aa02a7efdbcc8bb8cabe040b1e (diff)
Fixed multiple cursor implementations to be extra paranoid and close the cursor when an exception is thrown.
Diffstat (limited to 'src/main/java/com/amazon/carbonado/cursor')
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/AbstractCursor.java85
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/DifferenceCursor.java72
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/FilteredCursor.java45
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/GroupedCursor.java45
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/IntersectionCursor.java84
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/MultiTransformedCursor.java83
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/SortedCursor.java61
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/SymmetricDifferenceCursor.java87
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/ThrottledCursor.java49
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/TransformedCursor.java45
-rw-r--r--src/main/java/com/amazon/carbonado/cursor/UnionCursor.java62
11 files changed, 500 insertions, 218 deletions
diff --git a/src/main/java/com/amazon/carbonado/cursor/AbstractCursor.java b/src/main/java/com/amazon/carbonado/cursor/AbstractCursor.java
index fb34617..5fb0fa8 100644
--- a/src/main/java/com/amazon/carbonado/cursor/AbstractCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/AbstractCursor.java
@@ -37,31 +37,67 @@ public abstract class AbstractCursor<S> implements Cursor<S> {
}
public int copyInto(Collection<? super S> c) throws FetchException {
- int originalSize = c.size();
- while (hasNext()) {
- c.add(next());
+ try {
+ int originalSize = c.size();
+ while (hasNext()) {
+ c.add(next());
+ }
+ return c.size() - originalSize;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
- return c.size() - originalSize;
}
public int copyInto(Collection<? super S> c, int limit) throws FetchException {
- int originalSize = c.size();
- while (--limit >= 0 && hasNext()) {
- c.add(next());
+ try {
+ int originalSize = c.size();
+ while (--limit >= 0 && hasNext()) {
+ c.add(next());
+ }
+ return c.size() - originalSize;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
- return c.size() - originalSize;
}
public List<S> toList() throws FetchException {
- List<S> list = new ArrayList<S>();
- copyInto(list);
- return list;
+ try {
+ List<S> list = new ArrayList<S>();
+ copyInto(list);
+ return list;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
public List<S> toList(int limit) throws FetchException {
- List<S> list = new ArrayList<S>();
- copyInto(list, limit);
- return list;
+ try {
+ List<S> list = new ArrayList<S>();
+ copyInto(list, limit);
+ return list;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
public int skipNext(int amount) throws FetchException {
@@ -72,12 +108,21 @@ public abstract class AbstractCursor<S> implements Cursor<S> {
return 0;
}
- int count = 0;
- while (--amount >= 0 && hasNext()) {
- next();
- count++;
- }
+ try {
+ int count = 0;
+ while (--amount >= 0 && hasNext()) {
+ next();
+ count++;
+ }
- return count;
+ return count;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
}
diff --git a/src/main/java/com/amazon/carbonado/cursor/DifferenceCursor.java b/src/main/java/com/amazon/carbonado/cursor/DifferenceCursor.java
index 8b330ea..a04aec0 100644
--- a/src/main/java/com/amazon/carbonado/cursor/DifferenceCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/DifferenceCursor.java
@@ -73,47 +73,65 @@ public class DifferenceCursor<S> extends AbstractCursor<S> {
S nextLeft;
- while (true) {
- if (mLeftCursor.hasNext()) {
- nextLeft = mLeftCursor.next();
- } else {
- close();
- return false;
- }
- if (mNextRight == null) {
- if (mRightCursor.hasNext()) {
- mNextRight = mRightCursor.next();
+ try {
+ while (true) {
+ if (mLeftCursor.hasNext()) {
+ nextLeft = mLeftCursor.next();
} else {
- mNext = nextLeft;
- return true;
+ close();
+ return false;
}
- }
-
- while (true) {
- int result = mOrder.compare(nextLeft, mNextRight);
- if (result < 0) {
- mNext = nextLeft;
- return true;
- } else if (result > 0) {
+ if (mNextRight == null) {
if (mRightCursor.hasNext()) {
mNextRight = mRightCursor.next();
} else {
- mNextRight = null;
mNext = nextLeft;
return true;
}
- } else {
- break;
+ }
+
+ while (true) {
+ int result = mOrder.compare(nextLeft, mNextRight);
+ if (result < 0) {
+ mNext = nextLeft;
+ return true;
+ } else if (result > 0) {
+ if (mRightCursor.hasNext()) {
+ mNextRight = mRightCursor.next();
+ } else {
+ mNextRight = null;
+ mNext = nextLeft;
+ return true;
+ }
+ } else {
+ break;
+ }
}
}
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
}
public S next() throws FetchException {
- if (hasNext()) {
- S next = mNext;
- mNext = null;
- return next;
+ try {
+ if (hasNext()) {
+ S next = mNext;
+ mNext = null;
+ return next;
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
throw new NoSuchElementException();
}
diff --git a/src/main/java/com/amazon/carbonado/cursor/FilteredCursor.java b/src/main/java/com/amazon/carbonado/cursor/FilteredCursor.java
index f426f36..c25f8db 100644
--- a/src/main/java/com/amazon/carbonado/cursor/FilteredCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/FilteredCursor.java
@@ -101,15 +101,31 @@ public abstract class FilteredCursor<S> extends AbstractCursor<S> {
interruptCheck(++count);
}
} catch (NoSuchElementException e) {
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
return false;
}
public S next() throws FetchException {
- if (hasNext()) {
- S next = mNext;
- mNext = null;
- return next;
+ try {
+ if (hasNext()) {
+ S next = mNext;
+ mNext = null;
+ return next;
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
throw new NoSuchElementException();
}
@@ -122,13 +138,22 @@ public abstract class FilteredCursor<S> extends AbstractCursor<S> {
return 0;
}
- int count = 0;
- while (--amount >= 0 && hasNext()) {
- interruptCheck(++count);
- mNext = null;
- }
+ try {
+ int count = 0;
+ while (--amount >= 0 && hasNext()) {
+ interruptCheck(++count);
+ mNext = null;
+ }
- return count;
+ return count;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
private void interruptCheck(int count) throws FetchException {
diff --git a/src/main/java/com/amazon/carbonado/cursor/GroupedCursor.java b/src/main/java/com/amazon/carbonado/cursor/GroupedCursor.java
index 202f73a..daf9419 100644
--- a/src/main/java/com/amazon/carbonado/cursor/GroupedCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/GroupedCursor.java
@@ -157,16 +157,32 @@ public abstract class GroupedCursor<S, G> extends AbstractCursor<G> {
}
}
} catch (NoSuchElementException e) {
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
return false;
}
public G next() throws FetchException {
- if (hasNext()) {
- G next = mNextAggregate;
- mNextAggregate = null;
- return next;
+ try {
+ if (hasNext()) {
+ G next = mNextAggregate;
+ mNextAggregate = null;
+ return next;
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
throw new NoSuchElementException();
}
@@ -179,13 +195,22 @@ public abstract class GroupedCursor<S, G> extends AbstractCursor<G> {
return 0;
}
- int count = 0;
- while (--amount >= 0 && hasNext()) {
- interruptCheck(++count);
- mNextAggregate = null;
- }
+ try {
+ int count = 0;
+ while (--amount >= 0 && hasNext()) {
+ interruptCheck(++count);
+ mNextAggregate = null;
+ }
- return count;
+ return count;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
private void interruptCheck(int count) throws FetchException {
diff --git a/src/main/java/com/amazon/carbonado/cursor/IntersectionCursor.java b/src/main/java/com/amazon/carbonado/cursor/IntersectionCursor.java
index 58aba0d..6eb3179 100644
--- a/src/main/java/com/amazon/carbonado/cursor/IntersectionCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/IntersectionCursor.java
@@ -71,47 +71,65 @@ public class IntersectionCursor<S> extends AbstractCursor<S> {
S nextLeft, nextRight;
- if (mLeftCursor.hasNext()) {
- nextLeft = mLeftCursor.next();
- } else {
- close();
- return false;
- }
- if (mRightCursor.hasNext()) {
- nextRight = mRightCursor.next();
- } else {
- close();
- return false;
- }
+ try {
+ if (mLeftCursor.hasNext()) {
+ nextLeft = mLeftCursor.next();
+ } else {
+ close();
+ return false;
+ }
+ if (mRightCursor.hasNext()) {
+ nextRight = mRightCursor.next();
+ } else {
+ close();
+ return false;
+ }
- while (true) {
- int result = mOrder.compare(nextLeft, nextRight);
- if (result < 0) {
- if (mLeftCursor.hasNext()) {
- nextLeft = mLeftCursor.next();
+ while (true) {
+ int result = mOrder.compare(nextLeft, nextRight);
+ if (result < 0) {
+ if (mLeftCursor.hasNext()) {
+ nextLeft = mLeftCursor.next();
+ } else {
+ close();
+ return false;
+ }
+ } else if (result > 0) {
+ if (mRightCursor.hasNext()) {
+ nextRight = mRightCursor.next();
+ } else {
+ close();
+ return false;
+ }
} else {
- close();
- return false;
+ mNext = nextLeft;
+ return true;
}
- } else if (result > 0) {
- if (mRightCursor.hasNext()) {
- nextRight = mRightCursor.next();
- } else {
- close();
- return false;
- }
- } else {
- mNext = nextLeft;
- return true;
}
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
}
public S next() throws FetchException {
- if (hasNext()) {
- S next = mNext;
- mNext = null;
- return next;
+ try {
+ if (hasNext()) {
+ S next = mNext;
+ mNext = null;
+ return next;
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
throw new NoSuchElementException();
}
diff --git a/src/main/java/com/amazon/carbonado/cursor/MultiTransformedCursor.java b/src/main/java/com/amazon/carbonado/cursor/MultiTransformedCursor.java
index 7606eab..4b4a25a 100644
--- a/src/main/java/com/amazon/carbonado/cursor/MultiTransformedCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/MultiTransformedCursor.java
@@ -64,34 +64,52 @@ public abstract class MultiTransformedCursor<S, T> extends AbstractCursor<T> {
}
public boolean hasNext() throws FetchException {
- if (mNextCursor != null) {
- if (mNextCursor.hasNext()) {
- return true;
- }
- mNextCursor.close();
- mNextCursor = null;
- }
try {
- int count = 0;
- while (mCursor.hasNext()) {
- Cursor<T> nextCursor = transform(mCursor.next());
- if (nextCursor != null) {
- if (nextCursor.hasNext()) {
- mNextCursor = nextCursor;
- return true;
+ if (mNextCursor != null) {
+ if (mNextCursor.hasNext()) {
+ return true;
+ }
+ mNextCursor.close();
+ mNextCursor = null;
+ }
+ try {
+ int count = 0;
+ while (mCursor.hasNext()) {
+ Cursor<T> nextCursor = transform(mCursor.next());
+ if (nextCursor != null) {
+ if (nextCursor.hasNext()) {
+ mNextCursor = nextCursor;
+ return true;
+ }
+ nextCursor.close();
}
- nextCursor.close();
+ interruptCheck(++count);
}
- interruptCheck(++count);
+ } catch (NoSuchElementException e) {
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
}
- } catch (NoSuchElementException e) {
+ throw e;
}
return false;
}
public T next() throws FetchException {
- if (hasNext()) {
- return mNextCursor.next();
+ try {
+ if (hasNext()) {
+ return mNextCursor.next();
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
throw new NoSuchElementException();
}
@@ -104,17 +122,26 @@ public abstract class MultiTransformedCursor<S, T> extends AbstractCursor<T> {
return 0;
}
- int count = 0;
- while (hasNext()) {
- int chunk = mNextCursor.skipNext(amount);
- count += chunk;
- if ((amount -= chunk) <= 0) {
- break;
+ try {
+ int count = 0;
+ while (hasNext()) {
+ int chunk = mNextCursor.skipNext(amount);
+ count += chunk;
+ if ((amount -= chunk) <= 0) {
+ break;
+ }
+ interruptCheck(count);
}
- interruptCheck(count);
- }
- return count;
+ return count;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
private void interruptCheck(int count) throws FetchException {
diff --git a/src/main/java/com/amazon/carbonado/cursor/SortedCursor.java b/src/main/java/com/amazon/carbonado/cursor/SortedCursor.java
index 9cee89b..15a72d9 100644
--- a/src/main/java/com/amazon/carbonado/cursor/SortedCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/SortedCursor.java
@@ -205,28 +205,46 @@ public class SortedCursor<S> extends AbstractCursor<S> {
}
public boolean hasNext() throws FetchException {
- prepareNextChunk();
try {
- if (mChunkIterator.hasNext()) {
- return true;
+ prepareNextChunk();
+ try {
+ if (mChunkIterator.hasNext()) {
+ return true;
+ }
+ } catch (UndeclaredThrowableException e) {
+ throw toFetchException(e);
}
- } catch (UndeclaredThrowableException e) {
- throw toFetchException(e);
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
close();
return false;
}
public S next() throws FetchException {
- prepareNextChunk();
try {
- return mChunkIterator.next();
- } catch (UndeclaredThrowableException e) {
- throw toFetchException(e);
- } catch (NoSuchElementException e) {
+ prepareNextChunk();
+ try {
+ return mChunkIterator.next();
+ } catch (UndeclaredThrowableException e) {
+ throw toFetchException(e);
+ } catch (NoSuchElementException e) {
+ try {
+ close();
+ } catch (FetchException e2) {
+ // Don't care.
+ }
+ throw e;
+ }
+ } catch (FetchException e) {
try {
close();
- } catch (FetchException e2) {
+ } catch (Exception e2) {
// Don't care.
}
throw e;
@@ -241,13 +259,22 @@ public class SortedCursor<S> extends AbstractCursor<S> {
return 0;
}
- int count = 0;
- while (--amount >= 0 && hasNext()) {
- next();
- count++;
- }
+ try {
+ int count = 0;
+ while (--amount >= 0 && hasNext()) {
+ next();
+ count++;
+ }
- return count;
+ return count;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
private void prepareNextChunk() throws FetchException {
diff --git a/src/main/java/com/amazon/carbonado/cursor/SymmetricDifferenceCursor.java b/src/main/java/com/amazon/carbonado/cursor/SymmetricDifferenceCursor.java
index 1f94e30..1cd381a 100644
--- a/src/main/java/com/amazon/carbonado/cursor/SymmetricDifferenceCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/SymmetricDifferenceCursor.java
@@ -68,7 +68,16 @@ public class SymmetricDifferenceCursor<S> extends AbstractCursor<S> {
}
public boolean hasNext() throws FetchException {
- return compareNext() != 0;
+ try {
+ return compareNext() != 0;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
/**
@@ -81,43 +90,61 @@ public class SymmetricDifferenceCursor<S> extends AbstractCursor<S> {
return mCompareResult;
}
- while (true) {
- if (mNextLeft == null && mLeftCursor.hasNext()) {
- mNextLeft = mLeftCursor.next();
- }
- if (mNextRight == null && mRightCursor.hasNext()) {
- mNextRight = mRightCursor.next();
- }
+ try {
+ while (true) {
+ if (mNextLeft == null && mLeftCursor.hasNext()) {
+ mNextLeft = mLeftCursor.next();
+ }
+ if (mNextRight == null && mRightCursor.hasNext()) {
+ mNextRight = mRightCursor.next();
+ }
- if (mNextLeft == null) {
- return mNextRight != null ? 1 : 0;
+ if (mNextLeft == null) {
+ return mNextRight != null ? 1 : 0;
+ }
+ if (mNextRight == null) {
+ return -1;
+ }
+
+ if ((mCompareResult = mOrder.compare(mNextLeft, mNextRight)) == 0) {
+ mNextLeft = null;
+ mNextRight = null;
+ } else {
+ return mCompareResult;
+ }
}
- if (mNextRight == null) {
- return -1;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
}
+ throw e;
+ }
+ }
- if ((mCompareResult = mOrder.compare(mNextLeft, mNextRight)) == 0) {
+ public S next() throws FetchException {
+ try {
+ S next;
+ int result = compareNext();
+ if (result < 0) {
+ next = mNextLeft;
mNextLeft = null;
+ } else if (result > 0) {
+ next = mNextRight;
mNextRight = null;
} else {
- return mCompareResult;
+ throw new NoSuchElementException();
}
+ mCompareResult = 0;
+ return next;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
}
-
- public S next() throws FetchException {
- S next;
- int result = compareNext();
- if (result < 0) {
- next = mNextLeft;
- mNextLeft = null;
- } else if (result > 0) {
- next = mNextRight;
- mNextRight = null;
- } else {
- throw new NoSuchElementException();
- }
- mCompareResult = 0;
- return next;
- }
}
diff --git a/src/main/java/com/amazon/carbonado/cursor/ThrottledCursor.java b/src/main/java/com/amazon/carbonado/cursor/ThrottledCursor.java
index 2f4c5a0..d108d95 100644
--- a/src/main/java/com/amazon/carbonado/cursor/ThrottledCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/ThrottledCursor.java
@@ -61,12 +61,30 @@ public class ThrottledCursor<S> extends AbstractCursor<S> {
}
public boolean hasNext() throws FetchException {
- return mCursor.hasNext();
+ try {
+ return mCursor.hasNext();
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
public S next() throws FetchException {
- throttle();
- return mCursor.next();
+ try {
+ throttle();
+ return mCursor.next();
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
public int skipNext(int amount) throws FetchException {
@@ -77,16 +95,25 @@ public class ThrottledCursor<S> extends AbstractCursor<S> {
return 0;
}
- int count = 0;
- while (--amount >= 0) {
- throttle();
- if (skipNext(1) <= 0) {
- break;
+ try {
+ int count = 0;
+ while (--amount >= 0) {
+ throttle();
+ if (skipNext(1) <= 0) {
+ break;
+ }
+ count++;
}
- count++;
- }
- return count;
+ return count;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
private void throttle() throws FetchInterruptedException {
diff --git a/src/main/java/com/amazon/carbonado/cursor/TransformedCursor.java b/src/main/java/com/amazon/carbonado/cursor/TransformedCursor.java
index 70161cc..46dedd1 100644
--- a/src/main/java/com/amazon/carbonado/cursor/TransformedCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/TransformedCursor.java
@@ -74,15 +74,31 @@ public abstract class TransformedCursor<S, T> extends AbstractCursor<T> {
interruptCheck(++count);
}
} catch (NoSuchElementException e) {
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
return false;
}
public T next() throws FetchException {
- if (hasNext()) {
- T next = mNext;
- mNext = null;
- return next;
+ try {
+ if (hasNext()) {
+ T next = mNext;
+ mNext = null;
+ return next;
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
throw new NoSuchElementException();
}
@@ -95,13 +111,22 @@ public abstract class TransformedCursor<S, T> extends AbstractCursor<T> {
return 0;
}
- int count = 0;
- while (--amount >= 0 && hasNext()) {
- interruptCheck(++count);
- mNext = null;
- }
+ try {
+ int count = 0;
+ while (--amount >= 0 && hasNext()) {
+ interruptCheck(++count);
+ mNext = null;
+ }
- return count;
+ return count;
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
+ }
}
private void interruptCheck(int count) throws FetchException {
diff --git a/src/main/java/com/amazon/carbonado/cursor/UnionCursor.java b/src/main/java/com/amazon/carbonado/cursor/UnionCursor.java
index 7e4fde3..c2bdd19 100644
--- a/src/main/java/com/amazon/carbonado/cursor/UnionCursor.java
+++ b/src/main/java/com/amazon/carbonado/cursor/UnionCursor.java
@@ -67,39 +67,57 @@ public class UnionCursor<S> extends AbstractCursor<S> {
}
public boolean hasNext() throws FetchException {
- if (mNextLeft == null && mLeftCursor.hasNext()) {
- mNextLeft = mLeftCursor.next();
- }
- if (mNextRight == null && mRightCursor.hasNext()) {
- mNextRight = mRightCursor.next();
+ try {
+ if (mNextLeft == null && mLeftCursor.hasNext()) {
+ mNextLeft = mLeftCursor.next();
+ }
+ if (mNextRight == null && mRightCursor.hasNext()) {
+ mNextRight = mRightCursor.next();
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
+ }
+ throw e;
}
return mNextLeft != null || mNextRight != null;
}
public S next() throws FetchException {
- if (hasNext()) {
- S next;
- if (mNextLeft == null) {
- next = mNextRight;
- mNextRight = null;
- } else if (mNextRight == null) {
- next = mNextLeft;
- mNextLeft = null;
- } else {
- int result = mOrder.compare(mNextLeft, mNextRight);
- if (result < 0) {
- next = mNextLeft;
- mNextLeft = null;
- } else if (result > 0) {
+ try {
+ if (hasNext()) {
+ S next;
+ if (mNextLeft == null) {
next = mNextRight;
mNextRight = null;
- } else {
+ } else if (mNextRight == null) {
next = mNextLeft;
mNextLeft = null;
- mNextRight = null;
+ } else {
+ int result = mOrder.compare(mNextLeft, mNextRight);
+ if (result < 0) {
+ next = mNextLeft;
+ mNextLeft = null;
+ } else if (result > 0) {
+ next = mNextRight;
+ mNextRight = null;
+ } else {
+ next = mNextLeft;
+ mNextLeft = null;
+ mNextRight = null;
+ }
}
+ return next;
+ }
+ } catch (FetchException e) {
+ try {
+ close();
+ } catch (Exception e2) {
+ // Don't care.
}
- return next;
+ throw e;
}
throw new NoSuchElementException();
}