fix(cli): ast pretty_statement panic when version different (#451)

databendlabs · Jul 14, 2024 · ded4cf8 · ded4cf8
1 parent 8a6b664
commit ded4cf8
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/cli/src/ast/mod.rs b/cli/src/ast/mod.rs
@@ -24,7 +24,10 @@ pub fn format_query(query: &str) -> String {
     if kind == QueryKind::Put || kind == QueryKind::Get {
         return query.to_owned();
     }
-    let tokens = databend_common_ast::parser::tokenize_sql(query).unwrap();
-    let (stmt, _) = parse_sql(&tokens, Dialect::Experimental).unwrap();
-    pretty_statement(stmt, 80).unwrap()
+    if let Ok(tokens) = databend_common_ast::parser::tokenize_sql(query) {
+        if let Ok((stmt, _)) = parse_sql(&tokens, Dialect::Experimental) {
+            return pretty_statement(stmt, 80).unwrap();
+        }
+    }
+    query.to_string()
 }