add aggregate to simple aggregation

probably-nothing-labs · Aug 15, 2024 · 81d141d · 81d141d
1 parent b10ac78
commit 81d141d
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/crates/core/src/datastream.rs b/crates/core/src/datastream.rs
@@ -132,14 +132,11 @@ impl DataStream {
         loop {
             match stream.next().await.transpose() {
                 Ok(Some(batch)) => {
-                    for i in 0..batch.num_rows() {
-                        let row = batch.slice(i, 1);
-                        println!(
-                            "{}",
-                            datafusion::common::arrow::util::pretty::pretty_format_batches(&[row])
-                                .unwrap()
-                        );
-                    }
+                    println!(
+                        "{}",
+                        datafusion::common::arrow::util::pretty::pretty_format_batches(&[batch])
+                            .unwrap()
+                    );
                 }
                 Ok(None) => {
                     log::warn!("No RecordBatch in stream");