mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-16 20:53:27 +00:00
100 lines
2.6 KiB
C++
100 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <Poco/SharedPtr.h>
|
|
|
|
#include <DB/Columns/ColumnConst.h>
|
|
#include <DB/Columns/ColumnArray.h>
|
|
|
|
#include <DB/DataTypes/DataTypeArray.h>
|
|
|
|
#include <DB/DataStreams/IProfilingBlockInputStream.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
|
|
/** Реализует операцию ARRAY JOIN.
|
|
* (Для удобства, эта операция записывается, как функция arrayJoin, применённая к массиву.)
|
|
* Эта операция размножает все строки столько раз, сколько элементов в массиве.
|
|
* Результат функции arrayJoin - столбец единичных значений соответствующих элементов.
|
|
*
|
|
* Например,
|
|
*
|
|
* name arr
|
|
* ------ ------
|
|
* 'вася' [1, 2]
|
|
* 'петя' []
|
|
*
|
|
* преобразуется в
|
|
*
|
|
* name arrayJoin(arr)
|
|
* ------ --------------
|
|
* 'вася' 1
|
|
* 'вася' 2
|
|
*/
|
|
class ArrayJoiningBlockInputStream : public IProfilingBlockInputStream
|
|
{
|
|
public:
|
|
ArrayJoiningBlockInputStream(BlockInputStreamPtr input_, ssize_t array_column_)
|
|
: input(input_), array_column(array_column_)
|
|
{
|
|
children.push_back(input);
|
|
}
|
|
|
|
ArrayJoiningBlockInputStream(BlockInputStreamPtr input_, const String & array_column_name_)
|
|
: input(input_), array_column(-1), array_column_name(array_column_name_)
|
|
{
|
|
children.push_back(input);
|
|
}
|
|
|
|
String getName() const { return "ArrayJoiningBlockInputStream"; }
|
|
|
|
BlockInputStreamPtr clone() { return new ArrayJoiningBlockInputStream(input, array_column); }
|
|
|
|
protected:
|
|
Block readImpl()
|
|
{
|
|
Block block = input->read();
|
|
|
|
if (!block)
|
|
return block;
|
|
|
|
if (-1 == array_column)
|
|
array_column = block.getPositionByName(array_column_name);
|
|
|
|
ColumnPtr array = block.getByPosition(array_column).column;
|
|
|
|
if (array->isConst())
|
|
array = dynamic_cast<const IColumnConst &>(*array).convertToFullColumn();
|
|
|
|
size_t columns = block.columns();
|
|
for (size_t i = 0; i < columns; ++i)
|
|
{
|
|
ColumnWithNameAndType & current = block.getByPosition(i);
|
|
|
|
if (static_cast<ssize_t>(i) == array_column)
|
|
{
|
|
ColumnWithNameAndType result;
|
|
result.column = dynamic_cast<const ColumnArray &>(*current.column).getDataPtr();
|
|
result.type = dynamic_cast<const DataTypeArray &>(*current.type).getNestedType();
|
|
result.name = "arrayJoin(" + current.name + ")";
|
|
|
|
block.erase(i);
|
|
block.insert(i, result);
|
|
}
|
|
else
|
|
current.column->replicate(dynamic_cast<const ColumnArray &>(*array).getOffsets());
|
|
}
|
|
|
|
return block;
|
|
}
|
|
|
|
private:
|
|
BlockInputStreamPtr input;
|
|
ssize_t array_column;
|
|
String array_column_name;
|
|
};
|
|
|
|
}
|