From 4a8b150ca05f6d4d572f8d05cddd065a49477b3b Mon Sep 17 00:00:00 2001
From: metesynnada <100111937+metesynnada@users.noreply.github.com>
Date: Fri, 5 Jan 2024 17:22:45 +0300
Subject: [PATCH] Error handling.

---
 datafusion-cli/src/print_options.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/datafusion-cli/src/print_options.rs b/datafusion-cli/src/print_options.rs
index b8594352b585..b382eb34f62c 100644
--- a/datafusion-cli/src/print_options.rs
+++ b/datafusion-cli/src/print_options.rs
@@ -141,7 +141,8 @@ impl PrintOptions {
         let mut row_count = 0_usize;
         let mut with_header = true;
 
-        while let Some(Ok(batch)) = stream.next().await {
+        while let Some(maybe_batch) = stream.next().await {
+            let batch = maybe_batch?;
             row_count += batch.num_rows();
             self.format.print_batches(
                 &mut writer,