ClickHouse/dbms/src/Dictionaries/ExternalResultDescription.cpp

#include <ext/range.h>
#include <Dictionaries/ExternalResultDescription.h>
#include <DataTypes/DataTypesNumber.h>
#include <DataTypes/DataTypeString.h>
#include <DataTypes/DataTypeDate.h>
#include <DataTypes/DataTypeDateTime.h>
#include <DataTypes/DataTypeUUID.h>
#include <DataTypes/DataTypeNullable.h>
#include <Common/typeid_cast.h>


namespace DB
{

namespace ErrorCodes
{
    extern const int UNKNOWN_TYPE;
}

void ExternalResultDescription::init(const Block & sample_block_)
{
    sample_block = sample_block_;

    types.reserve(sample_block.columns());

    for (auto & elem : sample_block)
    {
        /// If default value for column was not provided, use default from data type.
        if (elem.column->empty())
            elem.column = elem.type->createColumnConstWithDefaultValue(1)->convertToFullColumnIfConst();

        DataTypePtr type_not_nullable = removeNullable(elem.type);
        const IDataType * type = type_not_nullable.get();

        if (typeid_cast<const DataTypeUInt8 *>(type))
            types.push_back(ValueType::UInt8);
        else if (typeid_cast<const DataTypeUInt16 *>(type))
            types.push_back(ValueType::UInt16);
        else if (typeid_cast<const DataTypeUInt32 *>(type))
            types.push_back(ValueType::UInt32);
        else if (typeid_cast<const DataTypeUInt64 *>(type))
            types.push_back(ValueType::UInt64);
        else if (typeid_cast<const DataTypeInt8 *>(type))
            types.push_back(ValueType::Int8);
        else if (typeid_cast<const DataTypeInt16 *>(type))
            types.push_back(ValueType::Int16);
        else if (typeid_cast<const DataTypeInt32 *>(type))
            types.push_back(ValueType::Int32);
        else if (typeid_cast<const DataTypeInt64 *>(type))
            types.push_back(ValueType::Int64);
        else if (typeid_cast<const DataTypeFloat32 *>(type))
            types.push_back(ValueType::Float32);
        else if (typeid_cast<const DataTypeFloat64 *>(type))
            types.push_back(ValueType::Float64);
        else if (typeid_cast<const DataTypeString *>(type))
            types.push_back(ValueType::String);
        else if (typeid_cast<const DataTypeDate *>(type))
            types.push_back(ValueType::Date);
        else if (typeid_cast<const DataTypeDateTime *>(type))
            types.push_back(ValueType::DateTime);
        else if (typeid_cast<const DataTypeUUID *>(type))
            types.push_back(ValueType::UUID);
        else
            throw Exception{"Unsupported type " + type->getName(), ErrorCodes::UNKNOWN_TYPE};
    }
}

}