mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 19:02:04 +00:00
91 lines
3.3 KiB
C++
91 lines
3.3 KiB
C++
#include <Storages/MergeTree/ReplicatedMergeTreeAltersSequence.h>
|
|
#include <cassert>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
int ReplicatedMergeTreeAltersSequence::getHeadAlterVersion(std::lock_guard<std::mutex> & /*state_lock*/) const
|
|
{
|
|
/// If queue empty, than we don't have version
|
|
if (!queue_state.empty())
|
|
return queue_state.begin()->first;
|
|
return -1;
|
|
}
|
|
|
|
void ReplicatedMergeTreeAltersSequence::addMutationForAlter(int alter_version, std::lock_guard<std::mutex> & /*state_lock*/)
|
|
{
|
|
/// Metadata alter can be added before, or
|
|
/// maybe already finished if we startup after metadata alter was finished.
|
|
if (!queue_state.count(alter_version))
|
|
queue_state.emplace(alter_version, AlterState{.metadata_finished=true, .data_finished=false});
|
|
else
|
|
queue_state[alter_version].data_finished = false;
|
|
}
|
|
|
|
void ReplicatedMergeTreeAltersSequence::addMetadataAlter(
|
|
int alter_version, std::lock_guard<std::mutex> & /*state_lock*/)
|
|
{
|
|
/// Data alter (mutation) always added before. See ReplicatedMergeTreeQueue::pullLogsToQueue.
|
|
/// So mutation already added to this sequence or doesn't exist.
|
|
if (!queue_state.count(alter_version))
|
|
queue_state.emplace(alter_version, AlterState{.metadata_finished=false, .data_finished=true});
|
|
else
|
|
queue_state[alter_version].metadata_finished = false;
|
|
}
|
|
|
|
void ReplicatedMergeTreeAltersSequence::finishMetadataAlter(int alter_version, std::unique_lock<std::mutex> & /*state_lock*/)
|
|
{
|
|
/// Sequence must not be empty
|
|
assert(!queue_state.empty());
|
|
/// Alters have to be finished in order
|
|
assert(queue_state.begin()->first == alter_version);
|
|
|
|
/// If metadata stage finished (or was never added) than we can remove this alter
|
|
if (queue_state[alter_version].data_finished)
|
|
queue_state.erase(alter_version);
|
|
else
|
|
queue_state[alter_version].metadata_finished = true;
|
|
}
|
|
|
|
void ReplicatedMergeTreeAltersSequence::finishDataAlter(int alter_version, std::lock_guard<std::mutex> & /*state_lock*/)
|
|
{
|
|
/// Queue can be empty after load of finished mutation without move of mutation pointer
|
|
if (queue_state.empty())
|
|
return;
|
|
|
|
/// Mutations may finish multiple times (for example, after server restart, before update of mutation pointer)
|
|
if (alter_version >= queue_state.begin()->first)
|
|
{
|
|
/// All alter versions bigger than head must present in queue.
|
|
assert(queue_state.count(alter_version));
|
|
|
|
if (queue_state[alter_version].metadata_finished)
|
|
queue_state.erase(alter_version);
|
|
else
|
|
queue_state[alter_version].data_finished = true;
|
|
}
|
|
}
|
|
|
|
bool ReplicatedMergeTreeAltersSequence::canExecuteDataAlter(int alter_version, std::lock_guard<std::mutex> & /*state_lock*/) const
|
|
{
|
|
/// Queue maybe empty when we start after server shutdown
|
|
/// and have some MUTATE_PART records in queue
|
|
if (queue_state.empty())
|
|
return true;
|
|
|
|
/// All versions smaller than head, can be executed
|
|
if (alter_version < queue_state.begin()->first)
|
|
return true;
|
|
|
|
return queue_state.at(alter_version).metadata_finished;
|
|
}
|
|
|
|
bool ReplicatedMergeTreeAltersSequence::canExecuteMetaAlter(int alter_version, std::lock_guard<std::mutex> & /*state_lock*/) const
|
|
{
|
|
assert(!queue_state.empty());
|
|
|
|
/// We can execute only alters of metadata which are in head.
|
|
return queue_state.begin()->first == alter_version;
|
|
}
|
|
}
|