mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-13 19:14:30 +00:00
118 lines
4.4 KiB
C++
118 lines
4.4 KiB
C++
#pragma once
|
||
|
||
#include <DB/Storages/MergeTree/MergeTreeData.h>
|
||
#include <DB/Storages/MergeTree/DiskSpaceMonitor.h>
|
||
#include <atomic>
|
||
#include <functional>
|
||
|
||
namespace DB
|
||
{
|
||
|
||
class MergeListEntry;
|
||
struct ReshardingJob;
|
||
|
||
|
||
/** Умеет выбирать куски для слияния и сливать их.
|
||
*/
|
||
class MergeTreeDataMerger
|
||
{
|
||
public:
|
||
using CancellationHook = std::function<void()>;
|
||
using AllowedMergingPredicate = std::function<bool (const MergeTreeData::DataPartPtr &, const MergeTreeData::DataPartPtr &)>;
|
||
|
||
public:
|
||
static const size_t NO_LIMIT = std::numeric_limits<size_t>::max();
|
||
|
||
MergeTreeDataMerger(MergeTreeData & data_);
|
||
|
||
void setCancellationHook(CancellationHook cancellation_hook_);
|
||
|
||
/** Выбирает, какие куски слить. Использует кучу эвристик.
|
||
* Если merge_anything_for_old_months, для кусков за прошедшие месяцы снимается ограничение на соотношение размеров.
|
||
* Выбирает куски так, чтобы available_disk_space, скорее всего, хватило с запасом для их слияния.
|
||
*
|
||
* can_merge - функция, определяющая, можно ли объединить пару соседних кусков.
|
||
* Эта функция должна координировать слияния со вставками и другими слияниями, обеспечивая, что:
|
||
* - Куски, между которыми еще может появиться новый кусок, нельзя сливать. См. METR-7001.
|
||
* - Кусок, который уже сливается с кем-то в одном месте, нельзя начать сливать в кем-то другим в другом месте.
|
||
*/
|
||
bool selectPartsToMerge(
|
||
MergeTreeData::DataPartsVector & what,
|
||
String & merged_name,
|
||
size_t available_disk_space,
|
||
bool merge_anything_for_old_months,
|
||
bool aggressive,
|
||
bool only_small,
|
||
const AllowedMergingPredicate & can_merge);
|
||
|
||
/** Сливает куски.
|
||
* Если reservation != nullptr, то и дело уменьшает размер зарезервированного места
|
||
* приблизительно пропорционально количеству уже выписанных данных.
|
||
*/
|
||
MergeTreeData::DataPartPtr mergeParts(
|
||
MergeTreeData::DataPartsVector & parts, const String & merged_name, MergeListEntry & merge_entry,
|
||
size_t aio_threshold, MergeTreeData::Transaction * out_transaction = nullptr,
|
||
DiskSpaceMonitor::Reservation * disk_reservation = nullptr);
|
||
|
||
/** Перешардирует заданную партицию.
|
||
*/
|
||
MergeTreeData::PerShardDataParts reshardPartition(
|
||
const ReshardingJob & job,
|
||
DiskSpaceMonitor::Reservation * disk_reservation = nullptr);
|
||
|
||
/// Примерное количество места на диске, нужное для мерджа. С запасом.
|
||
static size_t estimateDiskSpaceForMerge(const MergeTreeData::DataPartsVector & parts);
|
||
|
||
/** Отменяет все мерджи. Все выполняющиеся сейчас вызовы mergeParts скоро бросят исключение.
|
||
* Все новые вызовы будут бросать исключения, пока не будет вызван uncancel().
|
||
*/
|
||
void cancel() { cancelled = true; }
|
||
void uncancel() { cancelled = false; }
|
||
bool isCancelled() const { return cancelled; }
|
||
|
||
void abortIfRequested();
|
||
|
||
private:
|
||
/** Выбрать все куски принадлежащие одной партиции.
|
||
*/
|
||
MergeTreeData::DataPartsVector selectAllPartsFromPartition(DayNum_t partition);
|
||
|
||
private:
|
||
using FrozenPartitions = std::unordered_set<DayNum_t>;
|
||
|
||
private:
|
||
MergeTreeData & data;
|
||
|
||
Logger * log;
|
||
|
||
/// Когда в последний раз писали в лог, что место на диске кончилось (чтобы не писать об этом слишком часто).
|
||
time_t disk_space_warning_time = 0;
|
||
|
||
CancellationHook cancellation_hook;
|
||
|
||
std::atomic<bool> cancelled {false};
|
||
};
|
||
|
||
|
||
/** Временно приостанавливает мерджи.
|
||
*/
|
||
class MergeTreeMergeBlocker
|
||
{
|
||
public:
|
||
MergeTreeMergeBlocker(MergeTreeDataMerger & merger_)
|
||
: merger(merger_)
|
||
{
|
||
merger.cancel();
|
||
}
|
||
|
||
~MergeTreeMergeBlocker()
|
||
{
|
||
merger.uncancel();
|
||
}
|
||
|
||
private:
|
||
MergeTreeDataMerger & merger;
|
||
};
|
||
|
||
}
|