ClickHouse/dbms/include/DB/DataStreams/ArrayJoiningBlockInputStream.h

#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;
};

}