ClickHouse/dbms/Storages/MergeTree/ActiveDataPartSet.h

96 lines
2.8 KiB
C++
Raw Normal View History

#pragma once
#include <Storages/MergeTree/MergeTreePartInfo.h>
#include <Core/Types.h>
#include <map>
namespace DB
{
2017-04-16 15:00:33 +00:00
/** Supports multiple names of active parts of data.
* Repeats part of the MergeTreeData functionality.
* TODO: generalize with MergeTreeData
*/
class ActiveDataPartSet
{
public:
ActiveDataPartSet(MergeTreeDataFormatVersion format_version_) : format_version(format_version_) {}
ActiveDataPartSet(MergeTreeDataFormatVersion format_version_, const Strings & names);
ActiveDataPartSet(const ActiveDataPartSet & other)
: format_version(other.format_version)
, part_info_to_name(other.part_info_to_name)
{}
ActiveDataPartSet(ActiveDataPartSet && other) noexcept { swap(other); }
void swap(ActiveDataPartSet & other) noexcept
{
std::swap(format_version, other.format_version);
std::swap(part_info_to_name, other.part_info_to_name);
}
ActiveDataPartSet & operator=(const ActiveDataPartSet & other)
{
if (&other != this)
{
ActiveDataPartSet tmp(other);
swap(tmp);
}
return *this;
}
2018-06-22 10:43:35 +00:00
/// Returns true if the part was actually added. If out_replaced_parts != nullptr, it will contain
/// parts that were replaced from the set by the newly added part.
bool add(const String & name, Strings * out_replaced_parts = nullptr);
bool remove(const MergeTreePartInfo & part_info)
{
return part_info_to_name.erase(part_info) > 0;
}
bool remove(const String & part_name)
{
return remove(MergeTreePartInfo::fromPartName(part_name, format_version));
}
2020-02-05 16:30:02 +00:00
/// Remove part and all covered parts from active set
bool removePartAndCoveredParts(const String & part_name)
{
Strings parts_covered_by = getPartsCoveredBy(MergeTreePartInfo::fromPartName(part_name, format_version));
bool result = true;
result &= remove(part_name);
for (const auto & part : parts_covered_by)
result &= remove(part);
return result;
}
/// If not found, return an empty string.
String getContainingPart(const MergeTreePartInfo & part_info) const;
String getContainingPart(const String & name) const;
2014-05-26 18:14:52 +00:00
Strings getPartsCoveredBy(const MergeTreePartInfo & part_info) const;
/// Returns parts in ascending order of the partition_id and block number.
Strings getParts() const;
2014-08-08 08:28:13 +00:00
size_t size() const;
2020-02-05 16:30:02 +00:00
void clear()
{
part_info_to_name.clear();
}
MergeTreeDataFormatVersion getFormatVersion() const { return format_version; }
private:
MergeTreeDataFormatVersion format_version;
std::map<MergeTreePartInfo, String> part_info_to_name;
std::map<MergeTreePartInfo, String>::const_iterator getContainingPartImpl(const MergeTreePartInfo & part_info) const;
};
}