thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-30 19:42:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
zhongyuankai 2024-03-26 09:17:58 +08:00
parent 64961edb3f
commit 97d4fb01f4
1 changed files with 19 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/Parsers/ASTDropQuery.cpp
View File

@ -78,10 +78,19 @@ void ASTDropQuery::formatQueryImpl(const FormatSettings & settings, FormatState
if (it != list.children.begin())
settings.ostr << ", ";
auto identifier = dynamic_pointer_cast<ASTIdentifier>(*it);
settings.ostr << (identifier->name_parts.size() == 2
? backQuoteIfNeed(identifier->name_parts[0]) + "." + backQuoteIfNeed(identifier->name_parts[1])
: backQuoteIfNeed(identifier->name_parts[0]));
auto identifier = dynamic_pointer_cast<ASTTableIdentifier>(*it);
if (!identifier)
throw Exception(ErrorCodes::SYNTAX_ERROR, "Unexpected ASTIdentifier type for list of table names.");
if (auto db = identifier->getDatabase())
{
db->formatImpl(settings, state, frame);
settings.ostr << '.';
}
auto tb = identifier->getTable();
chassert(tb);
tb->formatImpl(settings, state, frame);
}
}
else
@ -122,16 +131,12 @@ ASTs ASTDropQuery::getRewrittenASTWithoutMultipleTables()
query.database_and_tables = nullptr;
query.children.clear();
auto database_and_table = dynamic_pointer_cast<ASTIdentifier>(child);
if (database_and_table->name_parts.size() == 2)
{
query.database = std::make_shared<ASTIdentifier>(database_and_table->name_parts[0]);
query.table = std::make_shared<ASTIdentifier>(database_and_table->name_parts[1]);
}
else
{
query.table = std::make_shared<ASTIdentifier>(database_and_table->name_parts[0]);
}
auto database_and_table = dynamic_pointer_cast<ASTTableIdentifier>(child);
if (!database_and_table)
throw Exception(ErrorCodes::SYNTAX_ERROR, "Unexpected ASTIdentifier type for list of table names.");
query.database = database_and_table->getDatabase();
query.table = database_and_table->getTable();
if (query.database)
query.children.push_back(query.database);