icinga2/lib/icinga/icingaapplication.cpp

217 lines
6.0 KiB
C++
Raw Normal View History

/******************************************************************************
* Icinga 2 *
2013-09-25 07:43:57 +02:00
* Copyright (C) 2012-2013 Icinga Development Team (http://www.icinga.org/) *
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License *
* as published by the Free Software Foundation; either version 2 *
* of the License, or (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the Free Software Foundation *
2012-05-11 13:33:57 +02:00
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
******************************************************************************/
2013-03-17 20:19:29 +01:00
#include "icinga/icingaapplication.h"
2013-03-16 21:18:53 +01:00
#include "base/dynamictype.h"
#include "base/logger_fwd.h"
#include "base/objectlock.h"
#include "base/convert.h"
2013-08-28 08:18:58 +02:00
#include "base/debug.h"
#include "base/utility.h"
2013-03-18 11:02:18 +01:00
#include "base/timer.h"
2013-09-10 16:03:36 +02:00
#include "base/scriptvariable.h"
#include "base/initialize.h"
2013-03-16 21:18:53 +01:00
#include <boost/smart_ptr/make_shared.hpp>
2012-03-31 16:29:53 +02:00
using namespace icinga;
2013-03-18 11:02:18 +01:00
static Timer::Ptr l_RetentionTimer;
2013-03-01 12:07:52 +01:00
REGISTER_TYPE(IcingaApplication);
INITIALIZE_ONCE(IcingaApplication, &IcingaApplication::StaticInitialize);
void IcingaApplication::StaticInitialize(void)
2013-10-08 11:57:35 +02:00
{
ScriptVariable::Set("IcingaEnableNotifications", true);
ScriptVariable::Set("IcingaEnableEventHandlers", true);
ScriptVariable::Set("IcingaEnableFlapping", true);
ScriptVariable::Set("IcingaEnableChecks", true);
ScriptVariable::Set("IcingaEnablePerfdata", true);
2013-10-08 11:57:35 +02:00
}
/**
* The entry point for the Icinga application.
*
* @returns An exit status.
*/
int IcingaApplication::Main(void)
2012-03-31 16:29:53 +02:00
{
2013-03-16 21:18:53 +01:00
Log(LogDebug, "icinga", "In IcingaApplication::Main()");
2012-08-05 03:10:53 +02:00
/* periodically dump the program state */
2013-03-18 11:02:18 +01:00
l_RetentionTimer = boost::make_shared<Timer>();
l_RetentionTimer->SetInterval(300);
l_RetentionTimer->OnTimerExpired.connect(boost::bind(&IcingaApplication::DumpProgramState, this));
l_RetentionTimer->Start();
2012-08-05 03:10:53 +02:00
2012-03-31 16:29:53 +02:00
RunEventLoop();
2013-03-16 21:18:53 +01:00
Log(LogInformation, "icinga", "Icinga has shut down.");
2012-07-24 15:38:19 +02:00
2012-04-02 13:09:33 +02:00
return EXIT_SUCCESS;
}
2013-02-24 01:10:34 +01:00
void IcingaApplication::OnShutdown(void)
{
ASSERT(!OwnsLock());
2013-03-02 09:07:47 +01:00
{
ObjectLock olock(this);
2013-03-18 11:02:18 +01:00
l_RetentionTimer->Stop();
2013-03-02 09:07:47 +01:00
}
2013-02-24 01:10:34 +01:00
DumpProgramState();
}
void IcingaApplication::DumpProgramState(void)
{
DynamicObject::DumpObjects(GetStatePath());
2012-07-24 13:13:02 +02:00
}
2012-06-27 18:43:34 +02:00
IcingaApplication::Ptr IcingaApplication::GetInstance(void)
{
return static_pointer_cast<IcingaApplication>(Application::GetInstance());
}
2012-07-13 11:29:23 +02:00
Dictionary::Ptr IcingaApplication::GetMacros(void) const
{
2013-09-10 16:03:36 +02:00
return ScriptVariable::Get("IcingaMacros");
2012-07-13 11:29:23 +02:00
}
2013-04-16 10:55:23 +02:00
bool IcingaApplication::ResolveMacro(const String& macro, const Dictionary::Ptr&, String *result) const
2013-02-24 01:10:34 +01:00
{
double now = Utility::GetTime();
if (macro == "TIMET") {
*result = Convert::ToString((long)now);
return true;
} else if (macro == "LONGDATETIME") {
*result = Utility::FormatDateTime("%Y-%m-%d %H:%M:%S %z", now);
return true;
} else if (macro == "SHORTDATETIME") {
*result = Utility::FormatDateTime("%Y-%m-%d %H:%M:%S", now);
return true;
} else if (macro == "DATE") {
*result = Utility::FormatDateTime("%Y-%m-%d", now);
return true;
} else if (macro == "TIME") {
*result = Utility::FormatDateTime("%H:%M:%S %z", now);
return true;
}
Dictionary::Ptr macros = GetMacros();
if (macros && macros->Contains(macro)) {
*result = macros->Get(macro);
return true;
}
2013-02-24 01:10:34 +01:00
return false;
2013-02-24 01:10:34 +01:00
}
2013-10-08 11:57:35 +02:00
bool IcingaApplication::GetEnableNotifications(void) const
{
2013-10-26 09:41:45 +02:00
if (!GetOverrideEnableNotifications().IsEmpty())
return GetOverrideEnableNotifications();
2013-10-08 11:57:35 +02:00
else
return ScriptVariable::Get("IcingaEnableNotifications");
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::SetEnableNotifications(bool enabled)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableNotifications(enabled);
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::ClearEnableNotifications(void)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableNotifications(Empty);
2013-10-08 11:57:35 +02:00
}
bool IcingaApplication::GetEnableEventHandlers(void) const
{
2013-10-26 09:41:45 +02:00
if (!GetOverrideEnableEventHandlers().IsEmpty())
return GetOverrideEnableEventHandlers();
2013-10-08 11:57:35 +02:00
else
return ScriptVariable::Get("IcingaEnableEventHandlers");
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::SetEnableEventHandlers(bool enabled)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableEventHandlers(enabled);
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::ClearEnableEventHandlers(void)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableEventHandlers(Empty);
2013-10-08 11:57:35 +02:00
}
bool IcingaApplication::GetEnableFlapping(void) const
{
2013-10-26 09:41:45 +02:00
if (!GetOverrideEnableFlapping().IsEmpty())
return GetOverrideEnableFlapping();
2013-10-08 11:57:35 +02:00
else
return ScriptVariable::Get("IcingaEnableFlapping");
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::SetEnableFlapping(bool enabled)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableFlapping(enabled);
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::ClearEnableFlapping(void)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableFlapping(Empty);
2013-10-08 11:57:35 +02:00
}
bool IcingaApplication::GetEnableChecks(void) const
{
2013-10-26 09:41:45 +02:00
if (!GetOverrideEnableChecks().IsEmpty())
return GetOverrideEnableChecks();
2013-10-08 11:57:35 +02:00
else
return ScriptVariable::Get("IcingaEnableChecks");
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::SetEnableChecks(bool enabled)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableChecks(enabled);
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::ClearEnableChecks(void)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnableChecks(Empty);
2013-10-08 11:57:35 +02:00
}
bool IcingaApplication::GetEnablePerfdata(void) const
{
2013-10-26 09:41:45 +02:00
if (!GetOverrideEnablePerfdata().IsEmpty())
return GetOverrideEnablePerfdata();
2013-10-08 11:57:35 +02:00
else
return ScriptVariable::Get("IcingaEnablePerfdata");
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::SetEnablePerfdata(bool enabled)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnablePerfdata(enabled);
2013-10-08 11:57:35 +02:00
}
void IcingaApplication::ClearEnablePerfdata(void)
{
2013-10-26 09:41:45 +02:00
SetOverrideEnablePerfdata(Empty);
2013-10-08 11:57:35 +02:00
}