2019-10-01 16:50:08 +00:00
|
|
|
#include <Storages/MergeTree/MergeTreeBaseSelectProcessor.h>
|
2018-02-13 19:34:15 +00:00
|
|
|
#include <Storages/MergeTree/MergeTreeRangeReader.h>
|
2019-10-10 16:30:30 +00:00
|
|
|
#include <Storages/MergeTree/IMergeTreeDataPart.h>
|
|
|
|
#include <Storages/MergeTree/IMergeTreeReader.h>
|
2017-04-06 17:21:45 +00:00
|
|
|
#include <Storages/MergeTree/MergeTreeBlockReadUtils.h>
|
2017-12-15 20:48:46 +00:00
|
|
|
#include <Columns/FilterDescription.h>
|
2017-07-13 20:58:19 +00:00
|
|
|
#include <Common/typeid_cast.h>
|
2018-04-06 13:58:06 +00:00
|
|
|
#include <DataTypes/DataTypeNothing.h>
|
2020-06-30 16:41:43 +00:00
|
|
|
#include <DataTypes/DataTypeNullable.h>
|
2017-04-06 17:21:45 +00:00
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
2017-08-01 13:04:48 +00:00
|
|
|
extern const int LOGICAL_ERROR;
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
MergeTreeBaseSelectProcessor::MergeTreeBaseSelectProcessor(
|
2019-09-13 15:41:09 +00:00
|
|
|
Block header,
|
2019-08-03 11:02:40 +00:00
|
|
|
const MergeTreeData & storage_,
|
2020-06-16 14:25:08 +00:00
|
|
|
const StorageMetadataPtr & metadata_snapshot_,
|
2019-08-03 11:02:40 +00:00
|
|
|
const PrewhereInfoPtr & prewhere_info_,
|
|
|
|
UInt64 max_block_size_rows_,
|
|
|
|
UInt64 preferred_block_size_bytes_,
|
|
|
|
UInt64 preferred_max_column_in_block_size_bytes_,
|
2019-12-18 15:54:45 +00:00
|
|
|
const MergeTreeReaderSettings & reader_settings_,
|
2019-08-03 11:02:40 +00:00
|
|
|
bool use_uncompressed_cache_,
|
|
|
|
const Names & virt_column_names_)
|
2020-06-16 14:25:08 +00:00
|
|
|
: SourceWithProgress(getHeader(std::move(header), prewhere_info_, virt_column_names_))
|
|
|
|
, storage(storage_)
|
|
|
|
, metadata_snapshot(metadata_snapshot_)
|
|
|
|
, prewhere_info(prewhere_info_)
|
|
|
|
, max_block_size_rows(max_block_size_rows_)
|
|
|
|
, preferred_block_size_bytes(preferred_block_size_bytes_)
|
|
|
|
, preferred_max_column_in_block_size_bytes(preferred_max_column_in_block_size_bytes_)
|
|
|
|
, reader_settings(reader_settings_)
|
|
|
|
, use_uncompressed_cache(use_uncompressed_cache_)
|
|
|
|
, virt_column_names(virt_column_names_)
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2019-10-02 11:57:17 +00:00
|
|
|
header_without_virtual_columns = getPort().getHeader();
|
|
|
|
|
|
|
|
for (auto it = virt_column_names.rbegin(); it != virt_column_names.rend(); ++it)
|
|
|
|
if (header_without_virtual_columns.has(*it))
|
|
|
|
header_without_virtual_columns.erase(*it);
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
Chunk MergeTreeBaseSelectProcessor::generate()
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2019-09-13 15:41:09 +00:00
|
|
|
while (!isCancelled())
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2019-07-18 14:41:11 +00:00
|
|
|
if ((!task || task->isFinished()) && !getNewTask())
|
2019-09-13 15:41:09 +00:00
|
|
|
return {};
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2019-09-13 15:41:09 +00:00
|
|
|
auto res = readFromPart();
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2019-10-21 16:26:29 +00:00
|
|
|
if (res.hasRows())
|
2019-09-13 15:41:09 +00:00
|
|
|
{
|
|
|
|
injectVirtualColumns(res, task.get(), virt_column_names);
|
|
|
|
return res;
|
|
|
|
}
|
2017-04-10 14:06:44 +00:00
|
|
|
}
|
|
|
|
|
2019-09-13 15:41:09 +00:00
|
|
|
return {};
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
void MergeTreeBaseSelectProcessor::initializeRangeReaders(MergeTreeReadTask & current_task)
|
2019-07-18 14:41:11 +00:00
|
|
|
{
|
|
|
|
if (prewhere_info)
|
|
|
|
{
|
|
|
|
if (reader->getColumns().empty())
|
|
|
|
{
|
2019-11-15 03:38:35 +00:00
|
|
|
current_task.range_reader = MergeTreeRangeReader(pre_reader.get(), nullptr, prewhere_info, true);
|
2019-07-18 14:41:11 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
MergeTreeRangeReader * pre_reader_ptr = nullptr;
|
|
|
|
if (pre_reader != nullptr)
|
|
|
|
{
|
2019-11-15 03:38:35 +00:00
|
|
|
current_task.pre_range_reader = MergeTreeRangeReader(pre_reader.get(), nullptr, prewhere_info, false);
|
2019-07-18 18:34:15 +00:00
|
|
|
pre_reader_ptr = ¤t_task.pre_range_reader;
|
2019-07-18 14:41:11 +00:00
|
|
|
}
|
|
|
|
|
2019-11-15 03:38:35 +00:00
|
|
|
current_task.range_reader = MergeTreeRangeReader(reader.get(), pre_reader_ptr, nullptr, true);
|
2019-07-18 14:41:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2019-11-15 03:38:35 +00:00
|
|
|
current_task.range_reader = MergeTreeRangeReader(reader.get(), nullptr, nullptr, true);
|
2019-07-18 14:41:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
Chunk MergeTreeBaseSelectProcessor::readFromPartImpl()
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2017-04-07 11:43:24 +00:00
|
|
|
if (task->size_predictor)
|
|
|
|
task->size_predictor->startBlock();
|
|
|
|
|
2019-06-14 19:11:41 +00:00
|
|
|
const UInt64 current_max_block_size_rows = max_block_size_rows;
|
|
|
|
const UInt64 current_preferred_block_size_bytes = preferred_block_size_bytes;
|
|
|
|
const UInt64 current_preferred_max_column_in_block_size_bytes = preferred_max_column_in_block_size_bytes;
|
|
|
|
const MergeTreeIndexGranularity & index_granularity = task->data_part->index_granularity;
|
2017-06-30 16:28:27 +00:00
|
|
|
const double min_filtration_ratio = 0.00001;
|
2017-06-21 17:19:35 +00:00
|
|
|
|
2020-03-23 02:12:31 +00:00
|
|
|
auto estimate_num_rows = [current_preferred_block_size_bytes, current_max_block_size_rows,
|
2019-06-14 19:11:41 +00:00
|
|
|
&index_granularity, current_preferred_max_column_in_block_size_bytes, min_filtration_ratio](
|
2019-01-04 12:10:00 +00:00
|
|
|
MergeTreeReadTask & current_task, MergeTreeRangeReader & current_reader)
|
2017-06-21 17:19:35 +00:00
|
|
|
{
|
2019-01-04 12:10:00 +00:00
|
|
|
if (!current_task.size_predictor)
|
2019-05-08 14:25:17 +00:00
|
|
|
return static_cast<size_t>(current_max_block_size_rows);
|
2017-06-30 16:28:27 +00:00
|
|
|
|
2017-07-24 15:06:32 +00:00
|
|
|
/// Calculates number of rows will be read using preferred_block_size_bytes.
|
2019-03-25 13:55:24 +00:00
|
|
|
/// Can't be less than avg_index_granularity.
|
2019-05-08 14:25:17 +00:00
|
|
|
size_t rows_to_read = current_task.size_predictor->estimateNumRows(current_preferred_block_size_bytes);
|
2017-07-24 13:59:03 +00:00
|
|
|
if (!rows_to_read)
|
|
|
|
return rows_to_read;
|
2019-05-08 12:26:35 +00:00
|
|
|
auto total_row_in_current_granule = current_reader.numRowsInCurrentGranule();
|
|
|
|
rows_to_read = std::max(total_row_in_current_granule, rows_to_read);
|
2017-07-21 17:45:51 +00:00
|
|
|
|
2019-01-04 12:10:00 +00:00
|
|
|
if (current_preferred_max_column_in_block_size_bytes)
|
2017-07-21 17:45:51 +00:00
|
|
|
{
|
2017-07-24 13:59:03 +00:00
|
|
|
/// Calculates number of rows will be read using preferred_max_column_in_block_size_bytes.
|
2019-05-08 12:26:35 +00:00
|
|
|
auto rows_to_read_for_max_size_column
|
2019-01-04 12:10:00 +00:00
|
|
|
= current_task.size_predictor->estimateNumRowsForMaxSizeColumn(current_preferred_max_column_in_block_size_bytes);
|
|
|
|
double filtration_ratio = std::max(min_filtration_ratio, 1.0 - current_task.size_predictor->filtered_rows_ratio);
|
2018-02-20 11:45:58 +00:00
|
|
|
auto rows_to_read_for_max_size_column_with_filtration
|
2019-05-08 12:26:35 +00:00
|
|
|
= static_cast<size_t>(rows_to_read_for_max_size_column / filtration_ratio);
|
2017-07-21 18:02:02 +00:00
|
|
|
|
2019-03-28 15:03:49 +00:00
|
|
|
/// If preferred_max_column_in_block_size_bytes is used, number of rows to read can be less than current_index_granularity.
|
2017-07-21 17:45:51 +00:00
|
|
|
rows_to_read = std::min(rows_to_read, rows_to_read_for_max_size_column_with_filtration);
|
|
|
|
}
|
2017-07-11 14:16:00 +00:00
|
|
|
|
2019-05-08 12:26:35 +00:00
|
|
|
auto unread_rows_in_current_granule = current_reader.numPendingRowsInCurrentGranule();
|
2017-07-11 14:16:00 +00:00
|
|
|
if (unread_rows_in_current_granule >= rows_to_read)
|
|
|
|
return rows_to_read;
|
|
|
|
|
2019-03-28 15:03:49 +00:00
|
|
|
return index_granularity.countMarksForRows(current_reader.currentMark(), rows_to_read, current_reader.numReadRowsInCurrentGranule());
|
2017-06-21 17:19:35 +00:00
|
|
|
};
|
2017-06-20 10:12:20 +00:00
|
|
|
|
2020-03-23 02:12:31 +00:00
|
|
|
UInt64 recommended_rows = estimate_num_rows(*task, task->range_reader);
|
2019-01-04 12:10:00 +00:00
|
|
|
UInt64 rows_to_read = std::max(UInt64(1), std::min(current_max_block_size_rows, recommended_rows));
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2018-02-20 11:45:58 +00:00
|
|
|
auto read_result = task->range_reader.read(rows_to_read, task->mark_ranges);
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2018-02-20 13:59:19 +00:00
|
|
|
/// All rows were filtered. Repeat.
|
2019-09-23 19:22:02 +00:00
|
|
|
if (read_result.num_rows == 0)
|
|
|
|
read_result.columns.clear();
|
2018-02-20 13:59:19 +00:00
|
|
|
|
2020-04-22 06:22:14 +00:00
|
|
|
const auto & sample_block = task->range_reader.getSampleBlock();
|
2019-09-26 17:29:41 +00:00
|
|
|
if (read_result.num_rows != 0 && sample_block.columns() != read_result.columns.size())
|
|
|
|
throw Exception("Inconsistent number of columns got from MergeTreeRangeReader. "
|
|
|
|
"Have " + toString(sample_block.columns()) + " in sample block "
|
|
|
|
"and " + toString(read_result.columns.size()) + " columns in list", ErrorCodes::LOGICAL_ERROR);
|
|
|
|
|
|
|
|
/// TODO: check columns have the same types as in header.
|
2018-02-20 13:59:19 +00:00
|
|
|
|
2019-09-23 19:22:02 +00:00
|
|
|
UInt64 num_filtered_rows = read_result.numReadRows() - read_result.num_rows;
|
2018-02-20 14:26:22 +00:00
|
|
|
|
2019-10-04 15:40:05 +00:00
|
|
|
progress({ read_result.numReadRows(), read_result.numBytesRead() });
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2018-02-20 11:45:58 +00:00
|
|
|
if (task->size_predictor)
|
|
|
|
{
|
2018-03-05 14:41:43 +00:00
|
|
|
task->size_predictor->updateFilteredRowsRation(read_result.numReadRows(), num_filtered_rows);
|
2017-08-01 13:04:48 +00:00
|
|
|
|
2019-09-23 19:22:02 +00:00
|
|
|
if (!read_result.columns.empty())
|
2019-09-26 17:29:41 +00:00
|
|
|
task->size_predictor->update(sample_block, read_result.columns, read_result.num_rows);
|
2018-02-20 11:45:58 +00:00
|
|
|
}
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
if (read_result.num_rows == 0)
|
|
|
|
return {};
|
|
|
|
|
|
|
|
Columns ordered_columns;
|
2019-10-02 11:57:17 +00:00
|
|
|
ordered_columns.reserve(header_without_virtual_columns.columns());
|
2019-10-01 16:50:08 +00:00
|
|
|
|
|
|
|
/// Reorder columns. TODO: maybe skip for default case.
|
2019-10-02 11:57:17 +00:00
|
|
|
for (size_t ps = 0; ps < header_without_virtual_columns.columns(); ++ps)
|
2018-02-22 11:54:26 +00:00
|
|
|
{
|
2019-10-02 11:57:17 +00:00
|
|
|
auto pos_in_sample_block = sample_block.getPositionByName(header_without_virtual_columns.getByPosition(ps).name);
|
2019-10-01 16:50:08 +00:00
|
|
|
ordered_columns.emplace_back(std::move(read_result.columns[pos_in_sample_block]));
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
2017-04-10 17:10:33 +00:00
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
return Chunk(std::move(ordered_columns), read_result.num_rows);
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
Chunk MergeTreeBaseSelectProcessor::readFromPart()
|
2019-07-18 14:41:11 +00:00
|
|
|
{
|
|
|
|
if (!task->range_reader.isInitialized())
|
|
|
|
initializeRangeReaders(*task);
|
|
|
|
|
|
|
|
return readFromPartImpl();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-10-31 11:32:24 +00:00
|
|
|
namespace
|
|
|
|
{
|
2019-10-31 13:18:21 +00:00
|
|
|
/// Simple interfaces to insert virtual columns.
|
2019-10-31 11:32:24 +00:00
|
|
|
struct VirtualColumnsInserter
|
|
|
|
{
|
2019-10-31 13:18:21 +00:00
|
|
|
virtual ~VirtualColumnsInserter() = default;
|
|
|
|
|
2019-10-31 11:32:24 +00:00
|
|
|
virtual void insertStringColumn(const ColumnPtr & column, const String & name) = 0;
|
|
|
|
virtual void insertUInt64Column(const ColumnPtr & column, const String & name) = 0;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
static void injectVirtualColumnsImpl(size_t rows, VirtualColumnsInserter & inserter,
|
|
|
|
MergeTreeReadTask * task, const Names & virtual_columns)
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
|
|
|
/// add virtual columns
|
|
|
|
/// Except _sample_factor, which is added from the outside.
|
2019-09-13 15:41:09 +00:00
|
|
|
if (!virtual_columns.empty())
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2019-09-13 15:41:09 +00:00
|
|
|
if (unlikely(rows && !task))
|
|
|
|
throw Exception("Cannot insert virtual columns to non-empty chunk without specified task.",
|
|
|
|
ErrorCodes::LOGICAL_ERROR);
|
2018-11-28 15:05:28 +00:00
|
|
|
|
2019-09-23 19:22:02 +00:00
|
|
|
for (const auto & virtual_column_name : virtual_columns)
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2019-09-23 19:22:02 +00:00
|
|
|
if (virtual_column_name == "_part")
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2018-02-19 03:56:08 +00:00
|
|
|
ColumnPtr column;
|
|
|
|
if (rows)
|
|
|
|
column = DataTypeString().createColumnConst(rows, task->data_part->name)->convertToFullColumnIfConst();
|
|
|
|
else
|
|
|
|
column = DataTypeString().createColumn();
|
|
|
|
|
2019-10-31 11:32:24 +00:00
|
|
|
inserter.insertStringColumn(column, virtual_column_name);
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
2019-09-23 19:22:02 +00:00
|
|
|
else if (virtual_column_name == "_part_index")
|
2017-04-06 17:21:45 +00:00
|
|
|
{
|
2018-02-19 03:56:08 +00:00
|
|
|
ColumnPtr column;
|
|
|
|
if (rows)
|
2018-10-22 08:54:54 +00:00
|
|
|
column = DataTypeUInt64().createColumnConst(rows, task->part_index_in_query)->convertToFullColumnIfConst();
|
2018-02-19 03:56:08 +00:00
|
|
|
else
|
|
|
|
column = DataTypeUInt64().createColumn();
|
|
|
|
|
2019-10-31 11:32:24 +00:00
|
|
|
inserter.insertUInt64Column(column, virtual_column_name);
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
2019-09-23 19:22:02 +00:00
|
|
|
else if (virtual_column_name == "_partition_id")
|
2018-09-10 09:53:13 +00:00
|
|
|
{
|
|
|
|
ColumnPtr column;
|
|
|
|
if (rows)
|
|
|
|
column = DataTypeString().createColumnConst(rows, task->data_part->info.partition_id)->convertToFullColumnIfConst();
|
|
|
|
else
|
|
|
|
column = DataTypeString().createColumn();
|
|
|
|
|
2019-10-31 11:32:24 +00:00
|
|
|
inserter.insertStringColumn(column, virtual_column_name);
|
2018-09-10 09:53:13 +00:00
|
|
|
}
|
2017-04-06 17:21:45 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-09-13 15:41:09 +00:00
|
|
|
namespace
|
|
|
|
{
|
2019-10-31 11:32:24 +00:00
|
|
|
struct VirtualColumnsInserterIntoBlock : public VirtualColumnsInserter
|
2019-09-13 15:41:09 +00:00
|
|
|
{
|
2019-10-31 11:32:24 +00:00
|
|
|
explicit VirtualColumnsInserterIntoBlock(Block & block_) : block(block_) {}
|
|
|
|
|
|
|
|
void insertStringColumn(const ColumnPtr & column, const String & name) final
|
2019-09-13 15:41:09 +00:00
|
|
|
{
|
2019-10-31 11:32:24 +00:00
|
|
|
block.insert({column, std::make_shared<DataTypeString>(), name});
|
|
|
|
}
|
|
|
|
|
|
|
|
void insertUInt64Column(const ColumnPtr & column, const String & name) final
|
|
|
|
{
|
|
|
|
block.insert({column, std::make_shared<DataTypeUInt64>(), name});
|
2019-09-13 15:41:09 +00:00
|
|
|
}
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2019-09-13 15:41:09 +00:00
|
|
|
Block & block;
|
|
|
|
};
|
2017-04-06 17:21:45 +00:00
|
|
|
|
2019-10-31 11:32:24 +00:00
|
|
|
struct VirtualColumnsInserterIntoColumns : public VirtualColumnsInserter
|
2019-09-13 15:41:09 +00:00
|
|
|
{
|
2019-10-31 11:32:24 +00:00
|
|
|
explicit VirtualColumnsInserterIntoColumns(Columns & columns_) : columns(columns_) {}
|
|
|
|
|
|
|
|
void insertStringColumn(const ColumnPtr & column, const String &) final
|
|
|
|
{
|
|
|
|
columns.push_back(column);
|
|
|
|
}
|
|
|
|
|
|
|
|
void insertUInt64Column(const ColumnPtr & column, const String &) final
|
2019-09-13 15:41:09 +00:00
|
|
|
{
|
|
|
|
columns.push_back(column);
|
|
|
|
}
|
|
|
|
|
|
|
|
Columns & columns;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
void MergeTreeBaseSelectProcessor::injectVirtualColumns(Block & block, MergeTreeReadTask * task, const Names & virtual_columns)
|
2019-09-13 15:41:09 +00:00
|
|
|
{
|
2019-10-31 11:32:24 +00:00
|
|
|
VirtualColumnsInserterIntoBlock inserter { block };
|
|
|
|
injectVirtualColumnsImpl(block.rows(), inserter, task, virtual_columns);
|
2019-09-13 15:41:09 +00:00
|
|
|
}
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
void MergeTreeBaseSelectProcessor::injectVirtualColumns(Chunk & chunk, MergeTreeReadTask * task, const Names & virtual_columns)
|
2019-09-13 15:41:09 +00:00
|
|
|
{
|
|
|
|
UInt64 num_rows = chunk.getNumRows();
|
|
|
|
auto columns = chunk.detachColumns();
|
|
|
|
|
2019-10-31 11:32:24 +00:00
|
|
|
VirtualColumnsInserterIntoColumns inserter { columns };
|
|
|
|
injectVirtualColumnsImpl(num_rows, inserter, task, virtual_columns);
|
2019-09-13 15:41:09 +00:00
|
|
|
|
|
|
|
chunk.setColumns(columns, num_rows);
|
|
|
|
}
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
void MergeTreeBaseSelectProcessor::executePrewhereActions(Block & block, const PrewhereInfoPtr & prewhere_info)
|
2018-04-06 13:58:06 +00:00
|
|
|
{
|
2018-04-11 14:31:54 +00:00
|
|
|
if (prewhere_info)
|
2018-04-06 13:58:06 +00:00
|
|
|
{
|
2018-09-03 17:24:46 +00:00
|
|
|
if (prewhere_info->alias_actions)
|
|
|
|
prewhere_info->alias_actions->execute(block);
|
|
|
|
|
2018-04-11 14:31:54 +00:00
|
|
|
prewhere_info->prewhere_actions->execute(block);
|
2020-06-30 14:20:27 +00:00
|
|
|
auto & prewhere_column = block.getByName(prewhere_info->prewhere_column_name);
|
|
|
|
|
2020-07-02 07:44:47 +00:00
|
|
|
if (!prewhere_column.type->canBeUsedInBooleanContext())
|
2020-06-30 14:20:27 +00:00
|
|
|
throw Exception("Invalid type for filter in PREWHERE: " + prewhere_column.type->getName(),
|
|
|
|
ErrorCodes::LOGICAL_ERROR);
|
|
|
|
|
2018-04-11 14:31:54 +00:00
|
|
|
if (prewhere_info->remove_prewhere_column)
|
|
|
|
block.erase(prewhere_info->prewhere_column_name);
|
2019-11-15 03:38:35 +00:00
|
|
|
else
|
|
|
|
{
|
|
|
|
auto & ctn = block.getByName(prewhere_info->prewhere_column_name);
|
|
|
|
ctn.column = ctn.type->createColumnConst(block.rows(), 1u)->convertToFullColumnIfConst();
|
|
|
|
}
|
2018-04-06 13:58:06 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
Block MergeTreeBaseSelectProcessor::getHeader(
|
2019-09-13 15:41:09 +00:00
|
|
|
Block block, const PrewhereInfoPtr & prewhere_info, const Names & virtual_columns)
|
|
|
|
{
|
|
|
|
executePrewhereActions(block, prewhere_info);
|
|
|
|
injectVirtualColumns(block, nullptr, virtual_columns);
|
|
|
|
return block;
|
|
|
|
}
|
|
|
|
|
2018-04-06 13:58:06 +00:00
|
|
|
|
2019-10-01 16:50:08 +00:00
|
|
|
MergeTreeBaseSelectProcessor::~MergeTreeBaseSelectProcessor() = default;
|
2017-04-06 17:21:45 +00:00
|
|
|
|
|
|
|
}
|