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

68 lines
2.1 KiB
C++
Raw Normal View History

2013-10-25 14:56:47 +00:00
#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>
2013-10-25 14:56:47 +00:00
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_)
2013-10-25 14:56:47 +00:00
{
}
void write(const Block & block) override
2013-10-31 18:38:34 +00:00
{
2013-10-25 14:56:47 +00:00
Block res = block;
/// Вычисляет явно указанные (в column_defaults) значения по-умолчанию.
2015-09-11 14:26:26 +00:00
/** @todo if somehow block does not contain values for implicitly-defaulted columns that are prerequisites
* for explicitly-defaulted ones, exception will be thrown during evaluating such columns
* (implicitly-defaulted columns are evaluated on the line after following one. */
evaluateMissingDefaults(res, *required_columns, column_defaults, context);
/// Добавляет не указанные значения по-умолчанию.
if (!only_explicit_column_defaults)
2015-09-11 14:26:26 +00:00
/// @todo this line may be moved before `evaluateMissingDefaults` with passing {required_columns - explicitly-defaulted columns}
res.addDefaults(*required_columns);
2013-10-31 18:38:34 +00:00
output->write(res);
2013-10-25 14:56:47 +00:00
}
void flush() override { output->flush(); }
void writePrefix() override { output->writePrefix(); }
void writeSuffix() override { output->writeSuffix(); }
2013-10-25 14:56:47 +00:00
private:
BlockOutputStreamPtr output;
NamesAndTypesListPtr required_columns;
2015-11-15 09:17:51 +00:00
const ColumnDefaults column_defaults;
Context context;
bool only_explicit_column_defaults;
2013-10-25 14:56:47 +00:00
};
2013-10-25 14:56:47 +00:00
}