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

91 lines
2.8 KiB
C++
Raw Normal View History

#pragma once
#include <mutex>
#include <Poco/RegularExpression.h>
2015-09-29 19:19:54 +00:00
#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;
2015-08-17 21:09:36 +00:00
Int64 left;
Int64 right;
UInt32 level;
std::string name;
2015-08-17 21:09:36 +00:00
DayNum_t month;
bool operator<(const Part & rhs) const
{
2015-08-17 21:09:36 +00:00
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
{
2015-08-17 21:09:36 +00:00
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);
/// Если не найдено - возвращает пустую строку.
2014-05-26 18:14:52 +00:00
String getContainingPart(const String & name) const;
2014-08-08 08:28:13 +00:00
Strings getParts() const; /// В порядке возрастания месяца и номера блока.
size_t size() const;
2015-08-17 21:09:36 +00:00
static String getPartName(DayNum_t left_date, DayNum_t right_date, Int64 left_id, Int64 right_id, UInt64 level);
/// Возвращает true если имя директории совпадает с форматом имени директории кусочков
2014-08-08 08:28:13 +00:00
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);
2014-07-22 13:49:52 +00:00
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;
};
}