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

#pragma once

#include <DB/DataStreams/IProfilingBlockInputStream.h>
#include <DB/Interpreters/Set.h>
#include <DB/Interpreters/Join.h>


namespace DB
{

/** Отдает без изменений данные из потока блоков, но перед чтением первого блока инициализирует все переданные множества.
  */
class CreatingSetsBlockInputStream : public IProfilingBlockInputStream
{
public:
	CreatingSetsBlockInputStream(BlockInputStreamPtr input, const Sets & sets_, const Joins & joins_)
		: sets(sets_), joins(joins_)
	{
		for (auto & set : sets)
			children.push_back(set->getSource());

		for (auto & join : joins)
			children.push_back(join->getSource());

		children.push_back(input);
	}

	String getName() const { return "CreatingSetsBlockInputStream"; }

	String getID() const
	{
		std::stringstream res;
		res << "CreatingSets(";

		Strings children_ids(children.size());
		for (size_t i = 0; i < children.size(); ++i)
			children_ids[i] = children[i]->getID();

		/// Будем считать, что порядок создания множеств не имеет значения.
		std::sort(children_ids.begin(), children_ids.end() - 1);

		for (size_t i = 0; i < children_ids.size(); ++i)
			res << (i == 0 ? "" : ", ") << children_ids[i];

		res << ")";
		return res.str();
	}

protected:
	Block readImpl();

private:
	Sets sets;
	Joins joins;
	bool created = false;

	Logger * log = &Logger::get("CreatingSetsBlockInputStream");

	void createSet(SetPtr & set);
	void createJoin(JoinPtr & join);
	void logProfileInfo(Stopwatch & watch, IBlockInputStream & in, size_t entries);
};

}