2016-11-20 12:43:20 +00:00
|
|
|
#include <DB/Storages/MergeTree/MergeTreeReader.h>
|
|
|
|
#include <DB/Storages/MergeTree/MergeTreeReadPool.h>
|
|
|
|
#include <DB/Storages/MergeTree/MergeTreeThreadBlockInputStream.h>
|
2016-12-10 04:51:36 +00:00
|
|
|
#include <DB/Columns/ColumnNullable.h>
|
2016-11-20 12:43:20 +00:00
|
|
|
#include <ext/range.hpp>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
|
|
|
|
|
|
|
namespace ErrorCodes
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
extern const int ILLEGAL_TYPE_OF_COLUMN_FOR_FILTER;
|
2016-11-20 12:43:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
MergeTreeThreadBlockInputStream::~MergeTreeThreadBlockInputStream() = default;
|
|
|
|
|
|
|
|
|
|
|
|
MergeTreeThreadBlockInputStream::MergeTreeThreadBlockInputStream(
|
2017-04-01 07:20:54 +00:00
|
|
|
const std::size_t thread,
|
|
|
|
const MergeTreeReadPoolPtr & pool, const std::size_t min_marks_to_read, const std::size_t block_size,
|
|
|
|
MergeTreeData & storage, const bool use_uncompressed_cache, const ExpressionActionsPtr & prewhere_actions,
|
|
|
|
const String & prewhere_column, const Settings & settings, const Names & virt_column_names)
|
|
|
|
: thread{thread}, pool{pool}, block_size_marks{block_size / storage.index_granularity},
|
|
|
|
/// round min_marks_to_read up to nearest multiple of block_size expressed in marks
|
|
|
|
min_marks_to_read{block_size
|
|
|
|
? (min_marks_to_read * storage.index_granularity + block_size - 1)
|
|
|
|
/ block_size * block_size / storage.index_granularity
|
|
|
|
: min_marks_to_read
|
|
|
|
},
|
|
|
|
storage{storage}, use_uncompressed_cache{use_uncompressed_cache}, prewhere_actions{prewhere_actions},
|
|
|
|
prewhere_column{prewhere_column}, min_bytes_to_use_direct_io{settings.min_bytes_to_use_direct_io},
|
|
|
|
max_read_buffer_size{settings.max_read_buffer_size}, virt_column_names{virt_column_names},
|
|
|
|
log{&Logger::get("MergeTreeThreadBlockInputStream")}
|
2016-11-20 12:43:20 +00:00
|
|
|
{}
|
|
|
|
|
|
|
|
|
|
|
|
String MergeTreeThreadBlockInputStream::getID() const
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
std::stringstream res;
|
|
|
|
/// @todo print some meaningful information
|
|
|
|
res << static_cast<const void *>(this);
|
|
|
|
return res.str();
|
2016-11-20 12:43:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Block MergeTreeThreadBlockInputStream::readImpl()
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
Block res;
|
2016-11-20 12:43:20 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
while (!res && !isCancelled())
|
|
|
|
{
|
|
|
|
if (!task && !getNewTask())
|
|
|
|
break;
|
2016-11-20 12:43:20 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
res = readFromPart();
|
2016-11-20 12:43:20 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
if (res)
|
|
|
|
injectVirtualColumns(res);
|
2016-11-20 12:43:20 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
if (task->mark_ranges.empty())
|
|
|
|
task = {};
|
|
|
|
}
|
2016-11-20 12:43:20 +00:00
|
|
|
|
2017-04-01 07:20:54 +00:00
|
|
|
return res;
|
2016-11-20 12:43:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/// Requests read task from MergeTreeReadPool and signals whether it got one
|
|
|
|
bool MergeTreeThreadBlockInputStream::getNewTask()
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
task = pool->getTask(min_marks_to_read, thread);
|
|
|
|
|
|
|
|
if (!task)
|
|
|
|
{
|
|
|
|
/** Close the files (before destroying the object).
|
|
|
|
* When many sources are created, but simultaneously reading only a few of them,
|
|
|
|
* buffers don't waste memory.
|
|
|
|
*/
|
|
|
|
reader = {};
|
|
|
|
pre_reader = {};
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
const auto path = storage.getFullPath() + task->data_part->name + '/';
|
|
|
|
|
|
|
|
/// Allows pool to reduce number of threads in case of too slow reads.
|
|
|
|
auto profile_callback = [this](ReadBufferFromFileBase::ProfileInfo info) { pool->profileFeedback(info); };
|
|
|
|
|
|
|
|
if (!reader)
|
|
|
|
{
|
|
|
|
if (use_uncompressed_cache)
|
|
|
|
owned_uncompressed_cache = storage.context.getUncompressedCache();
|
|
|
|
|
|
|
|
owned_mark_cache = storage.context.getMarkCache();
|
|
|
|
|
|
|
|
reader = std::make_unique<MergeTreeReader>(
|
|
|
|
path, task->data_part, task->columns, owned_uncompressed_cache.get(), owned_mark_cache.get(), true,
|
|
|
|
storage, task->mark_ranges, min_bytes_to_use_direct_io, max_read_buffer_size, MergeTreeReader::ValueSizeMap{}, profile_callback);
|
|
|
|
|
|
|
|
if (prewhere_actions)
|
|
|
|
pre_reader = std::make_unique<MergeTreeReader>(
|
|
|
|
path, task->data_part, task->pre_columns, owned_uncompressed_cache.get(), owned_mark_cache.get(), true,
|
|
|
|
storage, task->mark_ranges, min_bytes_to_use_direct_io,
|
|
|
|
max_read_buffer_size, MergeTreeReader::ValueSizeMap{}, profile_callback);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/// retain avg_value_size_hints
|
|
|
|
reader = std::make_unique<MergeTreeReader>(
|
|
|
|
path, task->data_part, task->columns, owned_uncompressed_cache.get(), owned_mark_cache.get(), true,
|
|
|
|
storage, task->mark_ranges, min_bytes_to_use_direct_io, max_read_buffer_size,
|
|
|
|
reader->getAvgValueSizeHints(), profile_callback);
|
|
|
|
|
|
|
|
if (prewhere_actions)
|
|
|
|
pre_reader = std::make_unique<MergeTreeReader>(
|
|
|
|
path, task->data_part, task->pre_columns, owned_uncompressed_cache.get(), owned_mark_cache.get(), true,
|
|
|
|
storage, task->mark_ranges, min_bytes_to_use_direct_io,
|
|
|
|
max_read_buffer_size, pre_reader->getAvgValueSizeHints(), profile_callback);
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
2016-11-20 12:43:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Block MergeTreeThreadBlockInputStream::readFromPart()
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
Block res;
|
|
|
|
|
|
|
|
if (prewhere_actions)
|
|
|
|
{
|
|
|
|
do
|
|
|
|
{
|
|
|
|
/// Let's read the full block of columns needed to calculate the expression in PREWHERE.
|
|
|
|
size_t space_left = std::max(1LU, block_size_marks);
|
|
|
|
MarkRanges ranges_to_read;
|
|
|
|
|
|
|
|
while (!task->mark_ranges.empty() && space_left && !isCancelled())
|
|
|
|
{
|
|
|
|
auto & range = task->mark_ranges.back();
|
|
|
|
|
|
|
|
size_t marks_to_read = std::min(range.end - range.begin, space_left);
|
|
|
|
pre_reader->readRange(range.begin, range.begin + marks_to_read, res);
|
|
|
|
|
|
|
|
ranges_to_read.emplace_back(range.begin, range.begin + marks_to_read);
|
|
|
|
space_left -= marks_to_read;
|
|
|
|
range.begin += marks_to_read;
|
|
|
|
if (range.begin == range.end)
|
|
|
|
task->mark_ranges.pop_back();
|
|
|
|
}
|
|
|
|
|
|
|
|
/// In case of isCancelled.
|
|
|
|
if (!res)
|
|
|
|
return res;
|
|
|
|
|
|
|
|
progressImpl({ res.rows(), res.bytes() });
|
|
|
|
pre_reader->fillMissingColumns(res, task->ordered_names, task->should_reorder);
|
|
|
|
|
|
|
|
/// Compute the expression in PREWHERE.
|
|
|
|
prewhere_actions->execute(res);
|
|
|
|
|
|
|
|
ColumnPtr column = res.getByName(prewhere_column).column;
|
|
|
|
if (task->remove_prewhere_column)
|
|
|
|
res.erase(prewhere_column);
|
|
|
|
|
|
|
|
const auto pre_bytes = res.bytes();
|
|
|
|
|
|
|
|
ColumnPtr observed_column;
|
|
|
|
if (column->isNullable())
|
|
|
|
{
|
|
|
|
ColumnNullable & nullable_col = static_cast<ColumnNullable &>(*column);
|
|
|
|
observed_column = nullable_col.getNestedColumn();
|
|
|
|
}
|
|
|
|
else
|
|
|
|
observed_column = column;
|
|
|
|
|
|
|
|
/** If the filter is a constant (for example, it says PREWHERE 1),
|
|
|
|
* then either return an empty block, or return the block unchanged.
|
|
|
|
*/
|
|
|
|
if (const auto column_const = typeid_cast<const ColumnConstUInt8 *>(observed_column.get()))
|
|
|
|
{
|
|
|
|
if (!column_const->getData())
|
|
|
|
{
|
|
|
|
res.clear();
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const auto & range : ranges_to_read)
|
|
|
|
reader->readRange(range.begin, range.end, res);
|
|
|
|
|
|
|
|
progressImpl({ 0, res.bytes() - pre_bytes });
|
|
|
|
}
|
|
|
|
else if (const auto column_vec = typeid_cast<const ColumnUInt8 *>(observed_column.get()))
|
|
|
|
{
|
|
|
|
size_t index_granularity = storage.index_granularity;
|
|
|
|
|
|
|
|
const auto & pre_filter = column_vec->getData();
|
|
|
|
IColumn::Filter post_filter(pre_filter.size());
|
|
|
|
|
|
|
|
/// Let's read the rest of the columns in the required segments and compose our own filter for them.
|
|
|
|
size_t pre_filter_pos = 0;
|
|
|
|
size_t post_filter_pos = 0;
|
|
|
|
|
|
|
|
for (const auto & range : ranges_to_read)
|
|
|
|
{
|
|
|
|
auto begin = range.begin;
|
|
|
|
auto pre_filter_begin_pos = pre_filter_pos;
|
|
|
|
|
|
|
|
for (auto mark = range.begin; mark <= range.end; ++mark)
|
|
|
|
{
|
|
|
|
UInt8 nonzero = 0;
|
|
|
|
|
|
|
|
if (mark != range.end)
|
|
|
|
{
|
|
|
|
const size_t limit = std::min(pre_filter.size(), pre_filter_pos + index_granularity);
|
|
|
|
for (size_t row = pre_filter_pos; row < limit; ++row)
|
|
|
|
nonzero |= pre_filter[row];
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!nonzero)
|
|
|
|
{
|
|
|
|
if (mark > begin)
|
|
|
|
{
|
|
|
|
memcpy(
|
|
|
|
&post_filter[post_filter_pos],
|
|
|
|
&pre_filter[pre_filter_begin_pos],
|
|
|
|
pre_filter_pos - pre_filter_begin_pos);
|
|
|
|
post_filter_pos += pre_filter_pos - pre_filter_begin_pos;
|
|
|
|
reader->readRange(begin, mark, res);
|
|
|
|
}
|
|
|
|
begin = mark + 1;
|
|
|
|
pre_filter_begin_pos = std::min(pre_filter_pos + index_granularity, pre_filter.size());
|
|
|
|
}
|
|
|
|
|
|
|
|
if (mark < range.end)
|
|
|
|
pre_filter_pos = std::min(pre_filter_pos + index_granularity, pre_filter.size());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!post_filter_pos)
|
|
|
|
{
|
|
|
|
res.clear();
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
progressImpl({ 0, res.bytes() - pre_bytes });
|
|
|
|
|
|
|
|
post_filter.resize(post_filter_pos);
|
|
|
|
|
|
|
|
/// Filter the columns related to PREWHERE using pre_filter,
|
|
|
|
/// other columns - using post_filter.
|
|
|
|
size_t rows = 0;
|
|
|
|
for (const auto i : ext::range(0, res.columns()))
|
|
|
|
{
|
|
|
|
auto & col = res.safeGetByPosition(i);
|
|
|
|
if (col.name == prewhere_column && res.columns() > 1)
|
|
|
|
continue;
|
|
|
|
col.column =
|
|
|
|
col.column->filter(task->column_name_set.count(col.name) ? post_filter : pre_filter, -1);
|
|
|
|
rows = col.column->size();
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Replace column with condition value from PREWHERE to a constant.
|
|
|
|
if (!task->remove_prewhere_column)
|
|
|
|
res.getByName(prewhere_column).column = std::make_shared<ColumnConstUInt8>(rows, 1);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
throw Exception{
|
|
|
|
"Illegal type " + column->getName() + " of column for filter. Must be ColumnUInt8 or ColumnConstUInt8.",
|
|
|
|
ErrorCodes::ILLEGAL_TYPE_OF_COLUMN_FOR_FILTER
|
|
|
|
};
|
|
|
|
|
|
|
|
if (res)
|
|
|
|
reader->fillMissingColumnsAndReorder(res, task->ordered_names);
|
|
|
|
}
|
|
|
|
while (!task->mark_ranges.empty() && !res && !isCancelled());
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
size_t space_left = std::max(1LU, block_size_marks);
|
|
|
|
|
|
|
|
while (!task->mark_ranges.empty() && space_left && !isCancelled())
|
|
|
|
{
|
|
|
|
auto & range = task->mark_ranges.back();
|
|
|
|
|
|
|
|
const size_t marks_to_read = std::min(range.end - range.begin, space_left);
|
|
|
|
reader->readRange(range.begin, range.begin + marks_to_read, res);
|
|
|
|
|
|
|
|
space_left -= marks_to_read;
|
|
|
|
range.begin += marks_to_read;
|
|
|
|
if (range.begin == range.end)
|
|
|
|
task->mark_ranges.pop_back();
|
|
|
|
}
|
|
|
|
|
|
|
|
/// In the case of isCancelled.
|
|
|
|
if (!res)
|
|
|
|
return res;
|
|
|
|
|
|
|
|
progressImpl({ res.rows(), res.bytes() });
|
|
|
|
reader->fillMissingColumns(res, task->ordered_names, task->should_reorder);
|
|
|
|
}
|
|
|
|
|
|
|
|
return res;
|
2016-11-20 12:43:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void MergeTreeThreadBlockInputStream::injectVirtualColumns(Block & block)
|
|
|
|
{
|
2017-04-01 07:20:54 +00:00
|
|
|
const auto rows = block.rows();
|
|
|
|
|
|
|
|
/// add virtual columns
|
|
|
|
/// Except _sample_factor, which is added from the outside.
|
|
|
|
if (!virt_column_names.empty())
|
|
|
|
{
|
|
|
|
for (const auto & virt_column_name : virt_column_names)
|
|
|
|
{
|
|
|
|
if (virt_column_name == "_part")
|
|
|
|
{
|
|
|
|
block.insert(ColumnWithTypeAndName{
|
|
|
|
ColumnConst<String>{rows, task->data_part->name}.convertToFullColumn(),
|
|
|
|
std::make_shared<DataTypeString>(),
|
|
|
|
virt_column_name
|
|
|
|
});
|
|
|
|
}
|
|
|
|
else if (virt_column_name == "_part_index")
|
|
|
|
{
|
|
|
|
block.insert(ColumnWithTypeAndName{
|
|
|
|
ColumnConst<UInt64>{rows, task->part_index_in_query}.convertToFullColumn(),
|
|
|
|
std::make_shared<DataTypeUInt64>(),
|
|
|
|
virt_column_name
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-11-20 12:43:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|