From 27771d314b30da66652ba288abb3636e969b5d8b Mon Sep 17 00:00:00 2001 From: baishen Date: Fri, 12 Jul 2024 19:11:50 +0800 Subject: [PATCH] fix: fix ast pretty_statement panic when version different --- cli/src/ast/mod.rs | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/cli/src/ast/mod.rs b/cli/src/ast/mod.rs index d734f711..8d1890f9 100644 --- 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() +} \ No newline at end of file