tyler.durden/icinga2
1
0
Fork 0
mirror of https://github.com/Icinga/icinga2.git synced 2025-07-25 06:34:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

cluster: Make log replays non-blocking.

Browse Source
This commit is contained in:
Gunnar Beutner 2013-09-16 09:30:31 +02:00
parent 21c35aabf4
commit e1b8e1180c
3 changed files with 95 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
components/cluster/clustercomponent.cpp
View File

@ -248,9 +248,14 @@ void ClusterComponent::RelayMessage(const Endpoint::Ptr& except, const Dictionar
if (endpoint->GetName() == GetIdentity())
continue;
{
ObjectLock olock(endpoint);
if (!endpoint->IsSyncing())
endpoint->SendMessage(message);
}
}
}
String ClusterComponent::GetClusterDir(void) const
{
@ -321,7 +326,10 @@ void ClusterComponent::ReplayLog(const Endpoint::Ptr& endpoint, const Stream::Pt
{
int count = 0;
ASSERT(OwnsLock());
ASSERT(!OwnsLock());
for (;;) {
ObjectLock olock(this);
CloseLogFile();
RotateLogFile();
@ -330,6 +338,11 @@ void ClusterComponent::ReplayLog(const Endpoint::Ptr& endpoint, const Stream::Pt
Utility::Glob(GetClusterDir() + "log/*", boost::bind(&ClusterComponent::LogGlobHandler, boost::ref(files), _1));
std::sort(files.begin(), files.end());
if (files.size() > 1) {
OpenLogFile();
olock.Unlock();
}
BOOST_FOREACH(int ts, files) {
String path = GetClusterDir() + "log/" + Convert::ToString(ts);
@ -369,7 +382,13 @@ void ClusterComponent::ReplayLog(const Endpoint::Ptr& endpoint, const Stream::Pt
Log(LogInformation, "cluster", "Replayed " + Convert::ToString(count) + " messages.");
OpenLogFile();
if (files.size() == 1) {
ObjectLock olock2(endpoint);
endpoint->SetSyncing(false);
break;
}
}
}
void ClusterComponent::ConfigGlobHandler(const Dictionary::Ptr& config, const String& file, bool basename)
@ -411,7 +430,17 @@ void ClusterComponent::NewClientHandler(const Socket::Ptr& client, TlsRole role)
return;
}
{
ObjectLock olock(endpoint);
Stream::Ptr oldClient = endpoint->GetClient();
if (oldClient)
oldClient->Close();
endpoint->SetSyncing(true);
endpoint->SetSeen(Utility::GetTime());
endpoint->SetClient(tlsStream);
}
Dictionary::Ptr config = boost::make_shared<Dictionary>();
Array::Ptr configFiles = endpoint->GetConfigFiles();
@ -435,18 +464,8 @@ void ClusterComponent::NewClientHandler(const Socket::Ptr& client, TlsRole role)
String json = Value(message).Serialize();
NetString::WriteStringToStream(tlsStream, json);
{
ObjectLock olock(this);
Stream::Ptr oldClient = endpoint->GetClient();
if (oldClient)
oldClient->Close();
endpoint->SetClient(tlsStream);
ReplayLog(endpoint, tlsStream);
}
}
void ClusterComponent::ClusterTimerHandler(void)
{

24
components/cluster/endpoint.cpp
View File

@ -34,6 +34,10 @@ REGISTER_TYPE(Endpoint);
boost::signals2::signal<void (const Endpoint::Ptr&)> Endpoint::OnConnected;
boost::signals2::signal<void (const Endpoint::Ptr&, const Dictionary::Ptr&)> Endpoint::OnMessageReceived;
Endpoint::Endpoint(void)
: m_Syncing(false)
{ }
/**
* Checks whether this endpoint is connected.
*
@ -46,18 +50,12 @@ bool Endpoint::IsConnected(void) const
Stream::Ptr Endpoint::GetClient(void) const
{
ObjectLock olock(this);
return m_Client;
}
void Endpoint::SetClient(const Stream::Ptr& client)
{
{
ObjectLock olock(this);
m_Client = client;
}
if (client) {
boost::thread thread(boost::bind(&Endpoint::MessageThreadProc, this, client));
@ -111,8 +109,6 @@ void Endpoint::MessageThreadProc(const Stream::Ptr& stream)
*/
String Endpoint::GetHost(void) const
{
ObjectLock olock(this);
return m_Host;
}
@ -123,8 +119,6 @@ String Endpoint::GetHost(void) const
*/
String Endpoint::GetPort(void) const
{
ObjectLock olock(this);
return m_Port;
}
@ -168,6 +162,16 @@ void Endpoint::SetRemoteLogPosition(double ts)
m_RemoteLogPosition = ts;
}
bool Endpoint::IsSyncing(void) const
{
return m_Syncing;
}
void Endpoint::SetSyncing(bool syncing)
{
m_Syncing = syncing;
}
Dictionary::Ptr Endpoint::GetFeatures(void) const
{
return m_Features;

6
components/cluster/endpoint.h
View File

@ -41,6 +41,8 @@ public:
DECLARE_PTR_TYPEDEFS(Endpoint);
DECLARE_TYPENAME(Endpoint);
Endpoint(void);
static boost::signals2::signal<void (const Endpoint::Ptr&)> OnConnected;
static boost::signals2::signal<void (const Endpoint::Ptr&, const Dictionary::Ptr&)> OnMessageReceived;
@ -70,6 +72,9 @@ public:
bool HasFeature(const String& type) const;
bool IsSyncing(void) const;
void SetSyncing(bool syncing);
protected:
virtual void InternalSerialize(const Dictionary::Ptr& bag, int attributeTypes) const;
virtual void InternalDeserialize(const Dictionary::Ptr& bag, int attributeTypes);
@ -86,6 +91,7 @@ private:
double m_LocalLogPosition;
double m_RemoteLogPosition;
Dictionary::Ptr m_Features;
bool m_Syncing;
void MessageThreadProc(const Stream::Ptr& stream);
};