mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-19 14:11:58 +00:00
39 lines
979 B
C++
39 lines
979 B
C++
#include <DB/Storages/VirtualColumnFactory.h>
|
|
#include <DB/DataTypes/DataTypeString.h>
|
|
#include <DB/DataTypes/DataTypesNumberFixed.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int NO_SUCH_COLUMN_IN_TABLE;
|
|
}
|
|
|
|
|
|
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 std::make_shared<DataTypeString>();
|
|
if (name == "_part") return std::make_shared<DataTypeString>();
|
|
if (name == "_part_index") return std::make_shared<DataTypeUInt64>();
|
|
if (name == "_sample_factor") return std::make_shared<DataTypeFloat64>();
|
|
if (name == "_replicated") return std::make_shared<DataTypeUInt8>();
|
|
return nullptr;
|
|
}
|
|
|
|
}
|