mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-02 22:50:47 +00:00
32 lines
766 B
C++
32 lines
766 B
C++
#include <DB/Storages/VirtualColumnFactory.h>
|
|
#include <DB/DataTypes/DataTypeString.h>
|
|
#include <DB/DataTypes/DataTypesNumberFixed.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
DataTypePtr VirtualColumnFactory::getType(const String & name)
|
|
{
|
|
auto res = tryGetType(name);
|
|
if (!res)
|
|
throw Exception("There is no column " + name + " in table.", ErrorCodes::NO_SUCH_COLUMN_IN_TABLE);
|
|
return res;
|
|
}
|
|
|
|
bool VirtualColumnFactory::hasColumn(const String & name)
|
|
{
|
|
return !!tryGetType(name);
|
|
}
|
|
|
|
DataTypePtr VirtualColumnFactory::tryGetType(const String & name)
|
|
{
|
|
if (name == "_table") return new DataTypeString;
|
|
if (name == "_part") return new DataTypeString;
|
|
if (name == "_part_index") return new DataTypeUInt64;
|
|
if (name == "_replicated") return new DataTypeUInt8;
|
|
return nullptr;
|
|
}
|
|
|
|
}
|