From 3e7f7e3e8da05026c514aeb8c61a2f9a80a5aa30 Mon Sep 17 00:00:00 2001 From: Ruihang Xia Date: Thu, 13 Apr 2023 15:19:00 +0800 Subject: [PATCH] fix: compile error in develop branch (#1376) Signed-off-by: Ruihang Xia --- src/datanode/src/sql/alter.rs | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/src/datanode/src/sql/alter.rs b/src/datanode/src/sql/alter.rs index a23dac6b32..caae6535d7 100644 --- a/src/datanode/src/sql/alter.rs +++ b/src/datanode/src/sql/alter.rs @@ -146,7 +146,8 @@ mod tests { use std::assert_matches::assert_matches; use datatypes::prelude::ConcreteDataType; - use query::parser::QueryLanguageParser; + use query::parser::{QueryLanguageParser, QueryStatement}; + use query::query_engine::SqlStatementExecutor; use session::context::QueryContext; use sql::dialect::GenericDialect; use sql::parser::ParserContext; @@ -231,20 +232,26 @@ mod tests { TIME INDEX (ts), PRIMARY KEY(host) ) engine=mito with(regions=1);"#; - let stmt = QueryLanguageParser::parse_sql(sql).unwrap(); + let stmt = match QueryLanguageParser::parse_sql(sql).unwrap() { + QueryStatement::Sql(sql) => sql, + _ => unreachable!(), + }; let output = instance .inner() - .execute_stmt(stmt, QueryContext::arc()) + .execute_sql(stmt, QueryContext::arc()) .await .unwrap(); assert!(matches!(output, Output::AffectedRows(0))); // Alter table. let sql = r#"alter table test_alter add column memory double"#; - let stmt = QueryLanguageParser::parse_sql(sql).unwrap(); + let stmt = match QueryLanguageParser::parse_sql(sql).unwrap() { + QueryStatement::Sql(sql) => sql, + _ => unreachable!(), + }; let output = instance .inner() - .execute_stmt(stmt, QueryContext::arc()) + .execute_sql(stmt, QueryContext::arc()) .await .unwrap(); assert!(matches!(output, Output::AffectedRows(0)));