fix(cli): ignore error for PROMPT_SQL query (#191)

databendlabs · Aug 23, 2023 · 362518c · 362518c
1 parent c1dc353
commit 362518c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/cli/src/session.rs b/cli/src/session.rs
@@ -66,10 +66,12 @@ impl Session {
             println!("Connected to {}", version);
             println!();
 
-            let mut rows = conn.query_iter(PROMPT_SQL).await.unwrap();
-            while let Some(row) = rows.next().await {
-                let name: (String,) = row.unwrap().try_into().unwrap();
-                keywords.push(name.0);
+            let rows = conn.query_iter(PROMPT_SQL).await;
+            if let Ok(mut rows) = rows {
+                while let Some(row) = rows.next().await {
+                    let name: (String,) = row.unwrap().try_into().unwrap();
+                    keywords.push(name.0);
+                }
             }
         }