Retry when seeking in PrestoS3FileSystem

gerashegalov · Apr 30, 2015 · 36b320b · 36b320b
1 parent ba1024a
commit 36b320b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/presto-docs/src/main/sphinx/release/release-0.102.rst b/presto-docs/src/main/sphinx/release/release-0.102.rst
@@ -11,3 +11,4 @@ Hive Changes
 ------------
 
 * Collect more metrics from ``PrestoS3FileSystem``.
+* Retry when seeking in ``PrestoS3FileSystem``.
diff --git a/presto-hive/src/main/java/com/facebook/presto/hive/PrestoS3FileSystem.java b/presto-hive/src/main/java/com/facebook/presto/hive/PrestoS3FileSystem.java
@@ -598,9 +598,14 @@ private void seekStream()
  long skip = nextReadPosition - streamPosition;
  if (skip <= max(in.available(), MAX_SKIP_SIZE.toBytes())) {
  // already buffered or seek is small enough
- if (in.skip(skip) == skip) {
- streamPosition = nextReadPosition;
- return;
+ try {
+ if (in.skip(skip) == skip) {
+ streamPosition = nextReadPosition;
+ return;
+ }
+ }
+ catch (IOException ignored) {
+ // will retry by re-opening the stream
  }
  }
  }