blob: 972e109afb317c5c59224277d79b0a6e21929e10 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
#include <Interpreters/getColumnFromBlock.h>
#include <Interpreters/castColumn.h>
namespace DB
{
namespace ErrorCodes
{
extern const int NOT_FOUND_COLUMN_IN_BLOCK;
}
ColumnPtr tryGetColumnFromBlock(const Block & block, const NameAndTypePair & requested_column)
{
const auto * elem = block.findByName(requested_column.getNameInStorage());
if (!elem)
return nullptr;
auto elem_type = elem->type;
auto elem_column = elem->column->decompress();
if (requested_column.isSubcolumn())
{
auto subcolumn_name = requested_column.getSubcolumnName();
elem_column = elem_type->tryGetSubcolumn(subcolumn_name, elem_column);
elem_type = elem_type->tryGetSubcolumnType(subcolumn_name);
if (!elem_type || !elem_column)
return nullptr;
}
return castColumn({elem_column, elem_type, ""}, requested_column.type);
}
ColumnPtr getColumnFromBlock(const Block & block, const NameAndTypePair & requested_column)
{
auto result_column = tryGetColumnFromBlock(block, requested_column);
if (!result_column)
throw Exception(ErrorCodes::NOT_FOUND_COLUMN_IN_BLOCK,
"Not found column or subcolumn {} in block. There are only columns: {}",
requested_column.name, block.dumpNames());
return result_column;
}
}
|
