From c7815cc9b8dcab812fdb765d4e1518b95200c555 Mon Sep 17 00:00:00 2001
From: harveyyue <yw_yuewei@126.com>
Date: Mon, 5 Aug 2024 12:31:58 +0800
Subject: [PATCH] fix: execute "select count(*) from tbl" always getting zero

---
 examples/datafusion_integration.rs | 5 +----
 src/array_decoder/mod.rs           | 6 ++++--
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/examples/datafusion_integration.rs b/examples/datafusion_integration.rs
index 38731301..8f0b7475 100644
--- a/examples/datafusion_integration.rs
+++ b/examples/datafusion_integration.rs
@@ -20,10 +20,7 @@ async fn main() -> Result<()> {
         .show()
         .await?;
 
-    ctx.sql("select count(*) from table1")
-        .await?
-        .show()
-        .await?;
+    ctx.sql("select count(*) from table1").await?.show().await?;
 
     ctx.read_orc(
         "tests/basic/data/alltypes.snappy.orc",
diff --git a/src/array_decoder/mod.rs b/src/array_decoder/mod.rs
index 6b62a666..a01987b9 100644
--- a/src/array_decoder/mod.rs
+++ b/src/array_decoder/mod.rs
@@ -576,8 +576,10 @@ impl NaiveStripeDecoder {
                     RecordBatch::try_new_with_options(
                         Arc::clone(&self.schema_ref),
                         fields,
-                        &RecordBatchOptions::new().with_row_count(Some(self.batch_size.min(remaining)))
-                    ).context(error::ConvertRecordBatchSnafu)?,
+                        &RecordBatchOptions::new()
+                            .with_row_count(Some(self.batch_size.min(remaining))),
+                    )
+                    .context(error::ConvertRecordBatchSnafu)?,
                 ))
             }
         } else {