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

#pragma once

#include <Poco/SharedPtr.h>

#include <DB/DataStreams/IBlockOutputStream.h>
#include <DB/Columns/ColumnConst.h>

#include <DB/Storages/ColumnDefault.h>
#include <DB/Interpreters/Context.h>
#include <DB/Interpreters/evaluateMissingDefaults.h>


namespace DB
{


/** Добавляет в блок недостающие столбцы со значениями по-умолчанию.
  * Эти столбцы - материалированные (не константы).
  */
class AddingDefaultBlockOutputStream : public IBlockOutputStream
{
public:
	AddingDefaultBlockOutputStream(
		BlockOutputStreamPtr output_,
		NamesAndTypesListPtr required_columns_,
		const ColumnDefaults & column_defaults_,
		const Context & context_,
		bool only_explicit_column_defaults_)
		: output(output_), required_columns(required_columns_),
		  column_defaults(column_defaults_), context(context_),
		  only_explicit_column_defaults(only_explicit_column_defaults_)
	{
	}

	void write(const Block & block) override
	{
		Block res = block;

		/// Вычисляет явно указанные (в column_defaults) значения по-умолчанию.
		evaluateMissingDefaults(res, *required_columns, column_defaults, context);

		/// Добавляет не указанные значения по-умолчанию.
		if (!only_explicit_column_defaults)
			res.addDefaults(*required_columns);

		output->write(res);
	}

	void flush() override { output->flush(); }

	void writePrefix() override { output->writePrefix(); }
	void writeSuffix() override { output->writeSuffix(); }

private:
	BlockOutputStreamPtr output;
	NamesAndTypesListPtr required_columns;
	const ColumnDefaults & column_defaults;
	Context context;
	bool only_explicit_column_defaults;
};


}