mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-27 01:51:59 +00:00
49 lines
1.3 KiB
C++
49 lines
1.3 KiB
C++
#pragma once
|
||
|
||
#include <DB/Parsers/IAST.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
|
||
/** DROP запрос
|
||
*/
|
||
class ASTDropQuery : public IAST
|
||
{
|
||
public:
|
||
bool detach{false}; /// Запрос DETACH, а не DROP.
|
||
bool if_exists{false};
|
||
String database;
|
||
String table;
|
||
|
||
ASTDropQuery() = default;
|
||
ASTDropQuery(const StringRange range_) : IAST(range_) {}
|
||
|
||
/** Получить текст, который идентифицирует этот элемент. */
|
||
String getID() const override { return (detach ? "DetachQuery_" : "DropQuery_") + database + "_" + table; };
|
||
|
||
ASTPtr clone() const override { return std::make_shared<ASTDropQuery>(*this); }
|
||
|
||
protected:
|
||
void formatImpl(const FormatSettings & settings, FormatState & state, FormatStateStacked frame) const override
|
||
{
|
||
if (table.empty() && !database.empty())
|
||
{
|
||
settings.ostr << (settings.hilite ? hilite_keyword : "")
|
||
<< (detach ? "DETACH DATABASE " : "DROP DATABASE ")
|
||
<< (if_exists ? "IF EXISTS " : "")
|
||
<< (settings.hilite ? hilite_none : "")
|
||
<< backQuoteIfNeed(database);
|
||
return;
|
||
}
|
||
|
||
settings.ostr << (settings.hilite ? hilite_keyword : "")
|
||
<< (detach ? "DETACH TABLE " : "DROP TABLE ")
|
||
<< (if_exists ? "IF EXISTS " : "") << (settings.hilite ? hilite_none : "")
|
||
<< (!database.empty() ? backQuoteIfNeed(database) + "." : "") << backQuoteIfNeed(table);
|
||
}
|
||
};
|
||
|
||
}
|