Use loop to drain exchange reader

BryanCutler · Oct 11, 2023 · 7740c76 · 7740c76
1 parent 34360b8
commit 7740c76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/java/flight/flight-core/src/main/java/org/apache/arrow/flight/FlightClient.java b/java/flight/flight-core/src/main/java/org/apache/arrow/flight/FlightClient.java
@@ -436,7 +436,7 @@ public ClientStreamListener getWriter() {
      */
     public void getResult() {
       // After exchange is complete, make sure stream is drained to propagate errors through reader
-      reader.next();
+      while (reader.next()) { };
     }
 
     /** Shut down the streams in this call. */