chore: better error print (#119)

databendlabs · Jun 1, 2023 · f04c89b · f04c89b
1 parent 77eb039
commit f04c89b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cli/src/display.rs b/cli/src/display.rs
@@ -85,6 +85,7 @@ impl<'a> FormatDisplay<'a> {
             println!("\n{}\n", format_sql);
         }
         let mut rows = Vec::new();
+        let mut error = None;
         while let Some(line) = self.data.next().await {
             match line {
                 Ok(RowWithProgress::Row(row)) => {
@@ -96,20 +97,25 @@ impl<'a> FormatDisplay<'a> {
                     self.stats = Some(pg);
                 }
                 Err(err) => {
-                    eprintln!("error: {}", err);
+                    error = Some(err);
                     break;
                 }
             }
         }
         if let Some(pb) = self.progress.take() {
             pb.finish_and_clear();
         }
+
         if !rows.is_empty() {
             println!(
                 "{}",
                 create_table(self.schema.clone(), &rows, self.settings.max_display_rows)?
             );
         }
+
+        if let Some(err) = error {
+            eprintln!("error happens after fetched {} rows: {}", rows.len(), err);
+        }
         Ok(())
     }