ClickHouse/dbms/include/DB/Storages/MergeTree/MergeTreeDataMerger.h

140 lines
5.2 KiB
C++
Raw Normal View History

2014-03-13 12:48:07 +00:00
#pragma once
#include <DB/Storages/MergeTree/MergeTreeData.h>
2014-08-04 11:41:59 +00:00
#include <DB/Storages/MergeTree/DiskSpaceMonitor.h>
#include <atomic>
2016-03-01 17:47:53 +00:00
#include <functional>
2014-03-13 12:48:07 +00:00
namespace DB
{
2015-04-16 06:12:35 +00:00
class MergeListEntry;
2016-03-07 04:35:06 +00:00
struct ReshardingJob;
2015-04-16 06:12:35 +00:00
2014-03-13 12:48:07 +00:00
/** Умеет выбирать куски для слияния и сливать их.
*/
class MergeTreeDataMerger
{
2016-03-01 17:47:53 +00:00
public:
using CancellationHook = std::function<void()>;
using AllowedMergingPredicate = std::function<bool (const MergeTreeData::DataPartPtr &, const MergeTreeData::DataPartPtr &)>;
2014-03-13 12:48:07 +00:00
public:
2016-03-25 11:48:45 +00:00
MergeTreeDataMerger(MergeTreeData & data_);
2014-03-13 12:48:07 +00:00
2016-03-01 17:47:53 +00:00
void setCancellationHook(CancellationHook cancellation_hook_);
2014-03-13 17:44:00 +00:00
/** Get maximum total size of parts to do merge, at current moment of time.
* It depends on number of free threads in background_pool and amount of free space in disk.
*/
size_t getMaxPartsSizeForMerge();
2014-03-13 17:44:00 +00:00
/** Выбирает, какие куски слить. Использует кучу эвристик.
*
* can_merge - функция, определяющая, можно ли объединить пару соседних кусков.
* Эта функция должна координировать слияния со вставками и другими слияниями, обеспечивая, что:
* - Куски, между которыми еще может появиться новый кусок, нельзя сливать. См. METR-7001.
* - Кусок, который уже сливается с кем-то в одном месте, нельзя начать сливать в кем-то другим в другом месте.
*/
2014-03-13 12:48:07 +00:00
bool selectPartsToMerge(
MergeTreeData::DataPartsVector & what,
2014-04-04 10:37:33 +00:00
String & merged_name,
2014-03-13 17:44:00 +00:00
bool aggressive,
size_t max_total_size_to_merge,
2014-03-13 17:44:00 +00:00
const AllowedMergingPredicate & can_merge);
2014-03-13 12:48:07 +00:00
/** Выбрать для слияния все куски в заданной партиции, если возможно.
* final - выбирать для слияния даже единственный кусок - то есть, позволять мерджить один кусок "сам с собой".
*/
bool selectAllPartsToMergeWithinPartition(
MergeTreeData::DataPartsVector & what,
String & merged_name,
size_t available_disk_space,
const AllowedMergingPredicate & can_merge,
DayNum_t partition,
bool final);
2014-08-04 11:41:59 +00:00
/** Сливает куски.
* Если reservation != nullptr, то и дело уменьшает размер зарезервированного места
* приблизительно пропорционально количеству уже выписанных данных.
*
* Создаёт и возвращает временный кусок.
* Чтобы закончить мердж, вызовите функцию renameTemporaryMergedPart.
2016-04-24 09:44:47 +00:00
*
* time_of_merge - время, когда мердж был назначен.
* Важно при использовании ReplicatedGraphiteMergeTree для обеспечения одинакового мерджа на репликах.
2014-08-04 11:41:59 +00:00
*/
MergeTreeData::MutableDataPartPtr mergePartsToTemporaryPart(
2016-03-25 11:48:45 +00:00
MergeTreeData::DataPartsVector & parts, const String & merged_name, MergeListEntry & merge_entry,
2016-04-24 09:44:47 +00:00
size_t aio_threshold, time_t time_of_merge, DiskSpaceMonitor::Reservation * disk_reservation = nullptr);
MergeTreeData::DataPartPtr renameMergedTemporaryPart(
MergeTreeData::DataPartsVector & parts,
MergeTreeData::MutableDataPartPtr & new_data_part,
const String & merged_name,
MergeTreeData::Transaction * out_transaction = nullptr);
2014-03-13 12:48:07 +00:00
2016-01-28 16:06:57 +00:00
/** Перешардирует заданную партицию.
*/
MergeTreeData::PerShardDataParts reshardPartition(
const ReshardingJob & job,
DiskSpaceMonitor::Reservation * disk_reservation = nullptr);
2014-03-13 12:48:07 +00:00
/// Примерное количество места на диске, нужное для мерджа. С запасом.
2015-06-11 00:35:36 +00:00
static size_t estimateDiskSpaceForMerge(const MergeTreeData::DataPartsVector & parts);
2014-03-13 12:48:07 +00:00
2016-01-28 16:06:57 +00:00
private:
/** Выбрать все куски принадлежащие одной партиции.
*/
MergeTreeData::DataPartsVector selectAllPartsFromPartition(DayNum_t partition);
2016-03-25 11:48:45 +00:00
/** Temporarily cancel merges.
*/
class BlockerImpl
{
public:
BlockerImpl(MergeTreeDataMerger * merger_) : merger(merger_)
{
++merger->cancelled;
}
~BlockerImpl()
{
--merger->cancelled;
}
private:
MergeTreeDataMerger * merger;
};
public:
/** Cancel all merges. All currently running 'mergeParts' methods will throw exception soon.
* All new calls to 'mergeParts' will throw exception till all 'Blocker' objects will be destroyed.
*/
using Blocker = std::unique_ptr<BlockerImpl>;
Blocker cancel() { return std::make_unique<BlockerImpl>(this); }
/** Cancel all merges forever.
*/
void cancelForever() { ++cancelled; }
bool isCancelled() const { return cancelled > 0; }
2016-01-28 16:06:57 +00:00
2014-03-13 12:48:07 +00:00
private:
MergeTreeData & data;
Logger * log;
2014-05-21 10:20:41 +00:00
/// Когда в последний раз писали в лог, что место на диске кончилось (чтобы не писать об этом слишком часто).
time_t disk_space_warning_time = 0;
2016-03-01 17:47:53 +00:00
CancellationHook cancellation_hook;
std::atomic<int> cancelled {0};
void abortReshardPartitionIfRequested();
2014-03-13 12:48:07 +00:00
};
2014-03-13 12:48:07 +00:00
}