fix: execute "select count(*) from tbl" always getting zero

datafusion-contrib · Aug 5, 2024 · c7815cc · c7815cc
1 parent 9d36f27
commit c7815cc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git 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
@@ -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 {