mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 10:22:10 +00:00
46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
#include "ASTColumnsMatcher.h"
|
|
|
|
#include <IO/WriteBuffer.h>
|
|
#include <IO/WriteHelpers.h>
|
|
#include <IO/WriteBufferFromString.h>
|
|
|
|
#include <re2/re2.h>
|
|
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
ASTPtr ASTColumnsMatcher::clone() const
|
|
{
|
|
auto clone = std::make_shared<ASTColumnsMatcher>(*this);
|
|
clone->cloneChildren();
|
|
return clone;
|
|
}
|
|
|
|
void ASTColumnsMatcher::appendColumnName(WriteBuffer & ostr) const { writeString(original_pattern, ostr); }
|
|
|
|
void ASTColumnsMatcher::formatImpl(const FormatSettings & settings, FormatState &, FormatStateStacked) const
|
|
{
|
|
WriteBufferFromOwnString pattern_quoted;
|
|
writeQuotedString(original_pattern, pattern_quoted);
|
|
|
|
settings.ostr << (settings.hilite ? hilite_keyword : "") << "COLUMNS" << (settings.hilite ? hilite_none : "") << "(" << pattern_quoted.str() << ")";
|
|
}
|
|
|
|
void ASTColumnsMatcher::setPattern(String pattern)
|
|
{
|
|
original_pattern = std::move(pattern);
|
|
column_matcher = std::make_shared<RE2>(original_pattern, RE2::Quiet);
|
|
if (!column_matcher->ok())
|
|
throw DB::Exception("COLUMNS pattern " + original_pattern + " cannot be compiled: " + column_matcher->error(), DB::ErrorCodes::CANNOT_COMPILE_REGEXP);
|
|
}
|
|
|
|
bool ASTColumnsMatcher::isColumnMatching(const String & column_name) const
|
|
{
|
|
return RE2::PartialMatch(column_name, *column_matcher);
|
|
}
|
|
|
|
|
|
}
|