mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
91 lines
3.1 KiB
C++
91 lines
3.1 KiB
C++
#pragma once
|
||
#include <mutex>
|
||
#include <Poco/RegularExpression.h>
|
||
#include <common/DateLUT.h>
|
||
#include <DB/Core/Types.h>
|
||
#include <set>
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Поддерживает множество названий активных кусков данных.
|
||
* Повторяет часть функциональности MergeTreeData.
|
||
* TODO: обобщить с MergeTreeData. Можно этот класс оставить примерно как есть и использовать его из MergeTreeData.
|
||
* Тогда в MergeTreeData можно сделать map<String, DataPartPtr> data_parts и all_data_parts.
|
||
*/
|
||
class ActiveDataPartSet
|
||
{
|
||
public:
|
||
ActiveDataPartSet() {}
|
||
ActiveDataPartSet(const Strings & names);
|
||
|
||
struct Part
|
||
{
|
||
DayNum_t left_date;
|
||
DayNum_t right_date;
|
||
Int64 left;
|
||
Int64 right;
|
||
UInt32 level;
|
||
std::string name;
|
||
DayNum_t month;
|
||
|
||
bool operator<(const Part & rhs) const
|
||
{
|
||
if (month != rhs.month)
|
||
return month < rhs.month;
|
||
|
||
if (left != rhs.left)
|
||
return left < rhs.left;
|
||
if (right != rhs.right)
|
||
return right < rhs.right;
|
||
|
||
if (level != rhs.level)
|
||
return level < rhs.level;
|
||
|
||
return false;
|
||
}
|
||
|
||
/// Содержит другой кусок (получен после объединения другого куска с каким-то ещё)
|
||
bool contains(const Part & rhs) const
|
||
{
|
||
return month == rhs.month /// Куски за разные месяцы не объединяются
|
||
&& left_date <= rhs.left_date
|
||
&& right_date >= rhs.right_date
|
||
&& left <= rhs.left
|
||
&& right >= rhs.right
|
||
&& level >= rhs.level;
|
||
}
|
||
};
|
||
|
||
void add(const String & name);
|
||
|
||
/// Если не найдено - возвращает пустую строку.
|
||
String getContainingPart(const String & name) const;
|
||
|
||
Strings getParts() const; /// В порядке возрастания месяца и номера блока.
|
||
|
||
size_t size() const;
|
||
|
||
static String getPartName(DayNum_t left_date, DayNum_t right_date, Int64 left_id, Int64 right_id, UInt64 level);
|
||
|
||
/// Возвращает true если имя директории совпадает с форматом имени директории кусочков
|
||
static bool isPartDirectory(const String & dir_name, Poco::RegularExpression::MatchVec * out_matches = nullptr);
|
||
|
||
/// Кладет в DataPart данные из имени кусочка.
|
||
static void parsePartName(const String & file_name, Part & part, const Poco::RegularExpression::MatchVec * matches = nullptr);
|
||
|
||
static bool contains(const String & outer_part_name, const String & inner_part_name);
|
||
|
||
private:
|
||
using Parts = std::set<Part>;
|
||
|
||
mutable std::mutex mutex;
|
||
Parts parts;
|
||
|
||
/// Не блокируют mutex.
|
||
void addImpl(const String & name);
|
||
String getContainingPartImpl(const String & name) const;
|
||
};
|
||
|
||
}
|