2014-05-26 10:41:40 +00:00
|
|
|
|
#pragma once
|
2016-05-28 10:15:36 +00:00
|
|
|
|
#include <mutex>
|
2014-05-26 10:41:40 +00:00
|
|
|
|
#include <Poco/RegularExpression.h>
|
2015-09-29 19:19:54 +00:00
|
|
|
|
#include <common/DateLUT.h>
|
2014-05-26 10:41:40 +00:00
|
|
|
|
#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:
|
2014-10-10 00:46:37 +00:00
|
|
|
|
ActiveDataPartSet() {}
|
|
|
|
|
ActiveDataPartSet(const Strings & names);
|
|
|
|
|
|
2014-05-26 10:41:40 +00:00
|
|
|
|
struct Part
|
|
|
|
|
{
|
|
|
|
|
DayNum_t left_date;
|
|
|
|
|
DayNum_t right_date;
|
2015-08-17 21:09:36 +00:00
|
|
|
|
Int64 left;
|
|
|
|
|
Int64 right;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
UInt32 level;
|
|
|
|
|
std::string name;
|
2015-08-17 21:09:36 +00:00
|
|
|
|
DayNum_t month;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
|
|
|
|
|
bool operator<(const Part & rhs) const
|
|
|
|
|
{
|
2015-08-17 21:09:36 +00:00
|
|
|
|
if (month != rhs.month)
|
|
|
|
|
return month < rhs.month;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
|
|
|
|
|
if (left != rhs.left)
|
|
|
|
|
return left < rhs.left;
|
|
|
|
|
if (right != rhs.right)
|
|
|
|
|
return right < rhs.right;
|
|
|
|
|
|
2016-05-16 18:43:38 +00:00
|
|
|
|
if (level != rhs.level)
|
|
|
|
|
return level < rhs.level;
|
|
|
|
|
|
2014-05-26 10:41:40 +00:00
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Содержит другой кусок (получен после объединения другого куска с каким-то ещё)
|
|
|
|
|
bool contains(const Part & rhs) const
|
|
|
|
|
{
|
2015-08-17 21:09:36 +00:00
|
|
|
|
return month == rhs.month /// Куски за разные месяцы не объединяются
|
2014-05-26 10:41:40 +00:00
|
|
|
|
&& left_date <= rhs.left_date
|
|
|
|
|
&& right_date >= rhs.right_date
|
|
|
|
|
&& left <= rhs.left
|
2016-05-16 18:43:38 +00:00
|
|
|
|
&& right >= rhs.right
|
|
|
|
|
&& level >= rhs.level;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
void add(const String & name);
|
2014-10-13 17:28:59 +00:00
|
|
|
|
|
|
|
|
|
/// Если не найдено - возвращает пустую строку.
|
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;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
|
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);
|
2014-05-26 10:41:40 +00:00
|
|
|
|
|
|
|
|
|
/// Возвращает true если имя директории совпадает с форматом имени директории кусочков
|
2014-08-08 08:28:13 +00:00
|
|
|
|
static bool isPartDirectory(const String & dir_name, Poco::RegularExpression::MatchVec * out_matches = nullptr);
|
2014-05-26 10:41:40 +00:00
|
|
|
|
|
|
|
|
|
/// Кладет в 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);
|
|
|
|
|
|
2014-05-26 10:41:40 +00:00
|
|
|
|
private:
|
2016-05-28 10:35:44 +00:00
|
|
|
|
using Parts = std::set<Part>;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
|
2016-05-28 10:15:36 +00:00
|
|
|
|
mutable std::mutex mutex;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
Parts parts;
|
2014-10-10 00:46:37 +00:00
|
|
|
|
|
2016-05-30 01:16:22 +00:00
|
|
|
|
/// Не блокируют mutex.
|
2014-10-10 00:46:37 +00:00
|
|
|
|
void addImpl(const String & name);
|
2016-05-30 01:16:22 +00:00
|
|
|
|
String getContainingPartImpl(const String & name) const;
|
2014-05-26 10:41:40 +00:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
}
|