mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-16 20:53:27 +00:00
93 lines
2.9 KiB
C++
93 lines
2.9 KiB
C++
#pragma once
|
||
#include <Poco/Mutex.h>
|
||
#include <Poco/RegularExpression.h>
|
||
#include <Yandex/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;
|
||
UInt64 left;
|
||
UInt64 right;
|
||
UInt32 level;
|
||
std::string name;
|
||
DayNum_t left_month;
|
||
DayNum_t right_month;
|
||
|
||
bool operator<(const Part & rhs) const
|
||
{
|
||
if (left_month != rhs.left_month)
|
||
return left_month < rhs.left_month;
|
||
if (right_month != rhs.right_month)
|
||
return right_month < rhs.right_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 left_month == rhs.left_month /// Куски за разные месяцы не объединяются
|
||
&& right_month == rhs.right_month
|
||
&& level > rhs.level
|
||
&& left_date <= rhs.left_date
|
||
&& right_date >= rhs.right_date
|
||
&& left <= rhs.left
|
||
&& right >= rhs.right;
|
||
}
|
||
};
|
||
|
||
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, UInt64 left_id, UInt64 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:
|
||
typedef std::set<Part> Parts;
|
||
|
||
mutable Poco::Mutex mutex;
|
||
Parts parts;
|
||
|
||
void addImpl(const String & name);
|
||
};
|
||
|
||
}
|