mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-29 21:20:49 +00:00
f6c72f8e56
Conflicts: dbms/src/Common/ActionBlocker.h dbms/src/Storages/MergeTree/ActiveDataPartSet.cpp dbms/src/Storages/MergeTree/ActiveDataPartSet.h dbms/src/Storages/MergeTree/MergeTreeData.h dbms/src/Storages/MergeTree/MergeTreeDataPart.cpp dbms/src/Storages/MergeTree/ReplicatedMergeTreeAlterThread.cpp dbms/src/Storages/MergeTree/ReplicatedMergeTreeLogEntry.cpp dbms/src/Storages/MergeTree/ReplicatedMergeTreeLogEntry.h dbms/src/Storages/MergeTree/ReplicatedMergeTreeQueue.cpp dbms/src/Storages/MergeTree/ReplicatedMergeTreeQueue.h dbms/src/Storages/MergeTree/ReplicatedMergeTreeRestartingThread.cpp dbms/src/Storages/StorageMergeTree.cpp dbms/src/Storages/StorageReplicatedMergeTree.cpp dbms/src/Storages/StorageReplicatedMergeTree.h
37 lines
927 B
C++
37 lines
927 B
C++
#pragma once
|
|
|
|
#include <atomic>
|
|
#include <memory>
|
|
#include <Common/ActionLock.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
/// An atomic variable that is used to block and interrupt certain actions
|
|
/// If it is not zero then actions related with it should be considered as interrupted
|
|
class ActionBlocker
|
|
{
|
|
public:
|
|
ActionBlocker() : counter(std::make_shared<Counter>(0)) {}
|
|
|
|
bool isCancelled() const { return *counter > 0; }
|
|
|
|
/// Temporarily blocks corresponding actions (while the returned object is alive)
|
|
friend class ActionLock;
|
|
ActionLock cancel() { return ActionLock(*this); }
|
|
|
|
/// Cancel the actions forever.
|
|
void cancelForever() { ++(*counter); }
|
|
|
|
/// Returns reference to counter to allow to watch on it directly.
|
|
const std::atomic<int> & getCounter() const { return *counter; }
|
|
|
|
private:
|
|
using Counter = std::atomic<int>;
|
|
using CounterPtr = std::shared_ptr<Counter>;
|
|
|
|
CounterPtr counter;
|
|
};
|
|
|
|
}
|