Fix

Author: matthewmturner

src/tui/execution.rs

@@ -411,7 +411,11 @@ impl TuiExecution {
             if let Some((ticket, batch_result)) = s.next().await {
                 match batch_result {
                     Ok(batch) => {
-                        info!("Fetched next FlightSQL batch from {}: {} rows", ticket, batch.num_rows());
+                        info!(
+                            "Fetched next FlightSQL batch from {}: {} rows",
+                            ticket,
+                            batch.num_rows()
+                        );
                         let duration = start.elapsed();
                         let results = ExecutionResultsBatch {
                             query: sql,

src/tui/handlers/flightsql.rs

@@ -83,7 +83,11 @@ pub fn normal_mode_handler(app: &mut App, key: KeyEvent) {
                 let next_page = current_page + 1;
 
                 // Check if we need more batches for the next page
-                if app.state.flightsql_tab.needs_more_batches_for_page(next_page) {
+                if app
+                    .state
+                    .flightsql_tab
+                    .needs_more_batches_for_page(next_page)
+                {
                     info!("Fetching more batches for page {}", next_page);
 
                     if let Some(last_query) = app.state.history_tab.history().last() {

src/tui/handlers/mod.rs

@@ -258,8 +258,15 @@ pub fn app_event_handler(app: &mut App, event: AppEvent) -> Result<()> {
                 // If not, automatically fetch another batch
                 if let Some(current_page) = app.state.flightsql_tab.current_page() {
                     let next_page = current_page + 1;
-                    if app.state.flightsql_tab.needs_more_batches_for_page(next_page) {
-                        info!("Still need more batches for page {}, fetching next batch", next_page);
+                    if app
+                        .state
+                        .flightsql_tab
+                        .needs_more_batches_for_page(next_page)
+                    {
+                        info!(
+                            "Still need more batches for page {}, fetching next batch",
+                            next_page
+                        );
                         let execution = Arc::clone(&app.execution);
                         let sql = query.clone();
                         let _event_tx = app.event_tx();
@@ -269,7 +276,10 @@ pub fn app_event_handler(app: &mut App, event: AppEvent) -> Result<()> {
                     } else {
                         // We now have enough data, advance to the page
                         info!("Sufficient data loaded, advancing to page {}", next_page);
-                        if let Err(e) = app.event_tx().send(AppEvent::FlightSQLExecutionResultsNextPage) {
+                        if let Err(e) = app
+                            .event_tx()
+                            .send(AppEvent::FlightSQLExecutionResultsNextPage)
+                        {
                             error!("Error advancing to next page: {e}");
                         }
                     }

src/tui/state/tabs/flightsql.rs

@@ -19,10 +19,7 @@ use core::cell::RefCell;
 use std::sync::Arc;
 
 use color_eyre::Result;
-use datafusion::arrow::{
-    array::RecordBatch,
-    datatypes::Schema,
-};
+use datafusion::arrow::{array::RecordBatch, datatypes::Schema};
 use log::{error, info};
 use ratatui::crossterm::event::KeyEvent;
 use ratatui::style::palette::tailwind;
@@ -213,15 +210,13 @@ impl FlightSQLTabState<'_> {
 
     pub fn current_page_results(&self) -> Option<RecordBatch> {
         match (self.current_page, self.result_batches.as_ref()) {
-            (Some(page), Some(batches)) => {
-                match extract_page(batches, page, PAGE_SIZE) {
-                    Ok(batch) => Some(batch),
-                    Err(err) => {
-                        error!("Error getting page {}: {}", page, err);
-                        None
-                    }
+            (Some(page), Some(batches)) => match extract_page(batches, page, PAGE_SIZE) {
+                Ok(batch) => Some(batch),
+                Err(err) => {
+                    error!("Error getting page {}: {}", page, err);
+                    None
                 }
-            }
+            },
             _ => Some(RecordBatch::new_empty(Arc::new(Schema::empty()))),
         }
     }
@@ -304,4 +299,3 @@ impl FlightSQLTabState<'_> {
         }
     }
 }
-