mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-01 14:10:50 +00:00
60 lines
2.2 KiB
C++
60 lines
2.2 KiB
C++
|
#include "GatherUtils.h"
|
||
|
|
||
|
namespace DB
|
||
|
{
|
||
|
/// Creates IArraySource from ColumnArray
|
||
|
|
||
|
template <typename... Types>
|
||
|
struct ArraySourceCreator;
|
||
|
|
||
|
template <typename Type, typename... Types>
|
||
|
struct ArraySourceCreator<Type, Types...>
|
||
|
{
|
||
|
static std::unique_ptr<IArraySource> create(const ColumnArray & col, const ColumnUInt8 * null_map, bool is_const, size_t total_rows)
|
||
|
{
|
||
|
if (typeid_cast<const ColumnVector<Type> *>(&col.getData()))
|
||
|
{
|
||
|
if (null_map)
|
||
|
{
|
||
|
if (is_const)
|
||
|
return std::make_unique<ConstSource<NullableArraySource<NumericArraySource<Type>>>>(col, *null_map, total_rows);
|
||
|
return std::make_unique<NullableArraySource<NumericArraySource<Type>>>(col, *null_map);
|
||
|
}
|
||
|
if (is_const)
|
||
|
return std::make_unique<ConstSource<NumericArraySource<Type>>>(col, total_rows);
|
||
|
return std::make_unique<NumericArraySource<Type>>(col);
|
||
|
}
|
||
|
|
||
|
return ArraySourceCreator<Types...>::create(col, null_map, is_const, total_rows);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
template <>
|
||
|
struct ArraySourceCreator<>
|
||
|
{
|
||
|
static std::unique_ptr<IArraySource> create(const ColumnArray & col, const ColumnUInt8 * null_map, bool is_const, size_t total_rows)
|
||
|
{
|
||
|
if (null_map)
|
||
|
{
|
||
|
if (is_const)
|
||
|
return std::make_unique<ConstSource<NullableArraySource<GenericArraySource>>>(col, *null_map, total_rows);
|
||
|
return std::make_unique<NullableArraySource<GenericArraySource>>(col, *null_map);
|
||
|
}
|
||
|
if (is_const)
|
||
|
return std::make_unique<ConstSource<GenericArraySource>>(col, total_rows);
|
||
|
return std::make_unique<GenericArraySource>(col);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
std::unique_ptr<IArraySource> createArraySource(const ColumnArray & col, bool is_const, size_t total_rows)
|
||
|
{
|
||
|
using Creator = typename ApplyTypeListForClass<ArraySourceCreator, TypeListNumbers>::Type;
|
||
|
if (auto column_nullable = typeid_cast<const ColumnNullable *>(&col.getData()))
|
||
|
{
|
||
|
ColumnArray column(column_nullable->getNestedColumn(), col.getOffsetsColumn());
|
||
|
return Creator::create(column, &column_nullable->getNullMapConcreteColumn(), is_const, total_rows);
|
||
|
}
|
||
|
return Creator::create(col, nullptr, is_const, total_rows);
|
||
|
}
|
||
|
}
|