2014-10-17 01:05:51 +00:00
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
#include <Poco/Event.h>
|
2015-09-29 19:19:54 +00:00
|
|
|
|
#include <common/logger_useful.h>
|
2014-10-17 01:05:51 +00:00
|
|
|
|
#include <DB/Core/Types.h>
|
|
|
|
|
#include <thread>
|
2015-11-05 19:44:19 +00:00
|
|
|
|
#include <atomic>
|
2014-10-17 01:05:51 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
class StorageReplicatedMergeTree;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** Инициализирует сессию в ZK.
|
|
|
|
|
* Выставляет эфемерные ноды. Выставляет нужные для обнаружения реплики значения нод.
|
|
|
|
|
* Запускает участие в выборе лидера. Запускает все фоновые потоки.
|
|
|
|
|
* Затем следит за тем, не истекла ли сессия. И если истекла - переинициализирует её.
|
|
|
|
|
*/
|
|
|
|
|
class ReplicatedMergeTreeRestartingThread
|
|
|
|
|
{
|
|
|
|
|
public:
|
|
|
|
|
ReplicatedMergeTreeRestartingThread(StorageReplicatedMergeTree & storage_);
|
|
|
|
|
|
|
|
|
|
~ReplicatedMergeTreeRestartingThread()
|
|
|
|
|
{
|
|
|
|
|
if (thread.joinable())
|
|
|
|
|
thread.join();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void wakeup()
|
|
|
|
|
{
|
|
|
|
|
wakeup_event.set();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Poco::Event & getWakeupEvent()
|
|
|
|
|
{
|
|
|
|
|
return wakeup_event;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void stop()
|
|
|
|
|
{
|
|
|
|
|
need_stop = true;
|
|
|
|
|
wakeup();
|
|
|
|
|
}
|
|
|
|
|
|
2015-11-05 19:44:19 +00:00
|
|
|
|
void getReplicaDelays(time_t & out_absolute_delay, time_t & out_relative_delay) const
|
|
|
|
|
{
|
|
|
|
|
out_absolute_delay = absolute_delay.load(std::memory_order_relaxed);
|
|
|
|
|
out_relative_delay = relative_delay.load(std::memory_order_relaxed);
|
|
|
|
|
}
|
|
|
|
|
|
2014-10-17 01:05:51 +00:00
|
|
|
|
private:
|
|
|
|
|
StorageReplicatedMergeTree & storage;
|
|
|
|
|
Logger * log;
|
|
|
|
|
Poco::Event wakeup_event;
|
|
|
|
|
volatile bool need_stop = false;
|
|
|
|
|
|
|
|
|
|
/// Случайные данные, которые мы записали в /replicas/me/is_active.
|
|
|
|
|
String active_node_identifier;
|
|
|
|
|
|
2014-10-17 21:44:56 +00:00
|
|
|
|
std::thread thread;
|
|
|
|
|
|
2015-11-05 19:44:19 +00:00
|
|
|
|
/// Отставание реплики.
|
|
|
|
|
std::atomic<time_t> absolute_delay {};
|
|
|
|
|
std::atomic<time_t> relative_delay {};
|
|
|
|
|
|
|
|
|
|
|
2014-10-17 01:05:51 +00:00
|
|
|
|
void run();
|
|
|
|
|
|
|
|
|
|
/// Запустить или остановить фоновые потоки. Используется для частичной переинициализации при пересоздании сессии в ZooKeeper.
|
|
|
|
|
bool tryStartup(); /// Возвращает false, если недоступен ZooKeeper.
|
|
|
|
|
|
|
|
|
|
/// Отметить в ZooKeeper, что эта реплика сейчас активна.
|
|
|
|
|
void activateReplica();
|
|
|
|
|
|
2015-09-20 11:02:59 +00:00
|
|
|
|
/// Удалить куски, для которых кворум пофейлился (за то время, когда реплика была неактивной).
|
|
|
|
|
void removeFailedQuorumParts();
|
|
|
|
|
|
|
|
|
|
/// Если есть недостигнутый кворум, и у нас есть кусок, то добавить эту реплику в кворум.
|
|
|
|
|
void updateQuorumIfWeHavePart();
|
|
|
|
|
|
2014-10-17 01:05:51 +00:00
|
|
|
|
void partialShutdown();
|
|
|
|
|
|
|
|
|
|
/// Запретить запись в таблицу и завершить все фоновые потоки.
|
|
|
|
|
void goReadOnlyPermanently();
|
2015-11-05 17:38:24 +00:00
|
|
|
|
|
|
|
|
|
/// Получить информацию об отставании реплик.
|
|
|
|
|
void checkReplicationDelays(time_t & out_absolute_delay, time_t & out_relative_delay);
|
2014-10-17 01:05:51 +00:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|