ClickHouse/libs/libzkutil/include/zkutil/SingleBarrier.h

44 lines
956 B
C++
Raw Normal View History

2016-03-01 17:47:53 +00:00
#pragma once
#include <zkutil/Common.h>
2016-03-01 17:47:53 +00:00
#include <string>
#include <functional>
namespace zkutil
{
/** Single distributed barrier for ZooKeeper.
*/
class SingleBarrier final
{
public:
using CancellationHook = std::function<void()>;
public:
SingleBarrier(GetZooKeeper get_zookeeper_, const std::string & path_, size_t counter_);
2016-03-01 17:47:53 +00:00
SingleBarrier(const SingleBarrier &) = delete;
SingleBarrier & operator=(const SingleBarrier &) = delete;
SingleBarrier(SingleBarrier &&) = default;
SingleBarrier & operator=(SingleBarrier &&) = default;
/// Register a function that checks whether barrier operation should be cancelled.
void setCancellationHook(CancellationHook cancellation_hook_);
void enter(uint64_t timeout = 0);
private:
void abortIfRequested();
private:
GetZooKeeper get_zookeeper;
EventPtr event = std::make_shared<Poco::Event>();
2016-03-01 17:47:53 +00:00
CancellationHook cancellation_hook;
std::string path;
2016-03-25 11:48:45 +00:00
std::string token;
2016-03-01 17:47:53 +00:00
size_t counter;
};
}