chore: attempt to drain gRPC ReadObject stream iterator

Related #2696
googleapis · Sep 10, 2024 · d0ffc84 · d0ffc84
1 parent 96e4049
commit d0ffc84
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/...ud-storage/src/main/java/com/google/cloud/storage/GapicUnbufferedReadableByteChannel.java b/...ud-storage/src/main/java/com/google/cloud/storage/GapicUnbufferedReadableByteChannel.java
@@ -182,6 +182,7 @@ public boolean hasNext() {
         if (!result.isDone()) {
           result.setException(StorageException.coalesce(e));
         }
+        reset();
         throw e;
       }
     }
@@ -194,6 +195,7 @@ public ReadObjectResponse next() {
         if (!result.isDone()) {
           result.setException(StorageException.coalesce(e));
         }
+        reset();
         throw e;
       }
     }
@@ -212,7 +214,7 @@ public void close() {
             } catch (IOException e) {
               if (ioException == null) {
                 ioException = e;
-              } else {
+              } else if (ioException != e) {
                 ioException.addSuppressed(e);
               }
             }
@@ -238,5 +240,11 @@ private Iterator<ReadObjectResponse> ensureResponseIteratorOpen() {
         }
       }
     }
+
+    private void reset() {
+      serverStream = null;
+      responseIterator = null;
+      streamInitialized = false;
+    }
   }
 }
diff --git a/google-cloud-storage/src/main/java/com/google/cloud/storage/GrpcStorageOptions.java b/google-cloud-storage/src/main/java/com/google/cloud/storage/GrpcStorageOptions.java
@@ -1098,7 +1098,7 @@ static void closeAllStreams(Iterable<InputStream> inputStreams) throws IOExcepti
         } catch (IOException e) {
           if (ioException == null) {
             ioException = e;
-          } else {
+          } else if (ioException != e) {
             ioException.addSuppressed(e);
           }
         }