mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-30 05:30:51 +00:00
53 lines
1.3 KiB
C++
53 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include <DataStreams/IProfilingBlockInputStream.h>
|
|
#include <Columns/ColumnConst.h>
|
|
#include <Core/ColumnWithTypeAndName.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/** Adds a materialized const column to the block with a specified value.
|
|
*/
|
|
template <typename ColumnType>
|
|
class AddingConstColumnBlockInputStream : public IProfilingBlockInputStream
|
|
{
|
|
public:
|
|
AddingConstColumnBlockInputStream(
|
|
BlockInputStreamPtr input_,
|
|
DataTypePtr data_type_,
|
|
ColumnType value_,
|
|
String column_name_)
|
|
: data_type(data_type_), value(value_), column_name(column_name_)
|
|
{
|
|
children.push_back(input_);
|
|
}
|
|
|
|
String getName() const override { return "AddingConstColumn"; }
|
|
|
|
String getID() const override
|
|
{
|
|
std::stringstream res;
|
|
res << "AddingConstColumn(" << children.back()->getID() << ")";
|
|
return res.str();
|
|
}
|
|
|
|
protected:
|
|
Block readImpl() override
|
|
{
|
|
Block res = children.back()->read();
|
|
if (!res)
|
|
return res;
|
|
ColumnPtr column_ptr = ColumnConst<ColumnType>(res.rows(), value, data_type).convertToFullColumn();
|
|
res.insert({column_ptr, data_type, column_name});
|
|
return res;
|
|
}
|
|
|
|
private:
|
|
DataTypePtr data_type;
|
|
ColumnType value;
|
|
String column_name;
|
|
};
|
|
|
|
}
|