Implement the external commands "SCHEDULE_FORCED_HOST_SVC_CHECKS" and "SCHEDULE_HOST_...

diff --git a/lib/icinga/externalcommand.cpp b/lib/icinga/externalcommand.cpp
index 56d3d232dd54c2534e3349c4b2351460ebdb8f25..aee5c9ca4a4b9a01f7ee1fe49d55872e1f0c2d6c 100644 (file)
--- a/lib/icinga/externalcommand.cpp
+++ b/lib/icinga/externalcommand.cpp
@@ -34,6 +34,8 @@ void ExternalCommand::Execute(double time, const String& command, const vector<S
                RegisterCommand("ENABLE_SVC_CHECK", &ExternalCommand::EnableSvcCheck);
                RegisterCommand("DISABLE_SVC_CHECK", &ExternalCommand::DisableSvcCheck);
                RegisterCommand("SHUTDOWN_PROCESS", &ExternalCommand::ShutdownProcess);
+               RegisterCommand("SCHEDULE_FORCED_HOST_SVC_CHECKS", &ExternalCommand::ScheduleForcedHostSvcChecks);
+               RegisterCommand("SCHEDULE_HOST_SVC_CHECKS", &ExternalCommand::ScheduleHostSvcChecks);
 
                m_Initialized = true;
        }
@@ -152,3 +154,58 @@ void ExternalCommand::ShutdownProcess(double time, const vector<String>& argumen
        Application::RequestShutdown();
 }
 
+void ExternalCommand::ScheduleForcedHostSvcChecks(double time, const vector<String>& arguments)
+{
+       if (arguments.size() < 2)
+               throw_exception(invalid_argument("Expected 2 arguments."));
+
+       if (!Host::Exists(arguments[0]))
+               throw_exception(invalid_argument("The host '" + arguments[0] + "' does not exist."));
+
+       double planned_check = arguments[1].ToDouble();
+
+       Host::Ptr host = Host::GetByName(arguments[0]);
+
+       DynamicObject::Ptr object;
+       BOOST_FOREACH(tie(tuples::ignore, object), DynamicType::GetByName("Service")->GetObjects()) {
+               Service::Ptr service = static_pointer_cast<Service>(object);
+
+               if (service->GetHost() != host)
+                       continue;
+
+               Logger::Write(LogInformation, "icinga", "Rescheduling next check for service '" + service->GetName() + "'");
+               service->SetNextCheck(planned_check);
+               service->SetForceNextCheck(true);
+       }
+}
+
+void ExternalCommand::ScheduleHostSvcChecks(double time, const vector<String>& arguments)
+{
+       if (arguments.size() < 2)
+               throw_exception(invalid_argument("Expected 2 arguments."));
+
+       if (!Host::Exists(arguments[0]))
+               throw_exception(invalid_argument("The host '" + arguments[0] + "' does not exist."));
+
+       double planned_check = arguments[1].ToDouble();
+
+       Host::Ptr host = Host::GetByName(arguments[0]);
+
+       DynamicObject::Ptr object;
+       BOOST_FOREACH(tie(tuples::ignore, object), DynamicType::GetByName("Service")->GetObjects()) {
+               Service::Ptr service = static_pointer_cast<Service>(object);
+
+               if (service->GetHost() != host)
+                       continue;
+
+               if (planned_check > service->GetNextCheck()) {
+                       Logger::Write(LogInformation, "icinga", "Ignoring reschedule request for service '" +
+                           arguments[1] + "' (next check is already sooner than requested check time)");
+                       return;
+               }
+
+               Logger::Write(LogInformation, "icinga", "Rescheduling next check for service '" + service->GetName() + "'");
+               service->SetNextCheck(planned_check);
+       }
+}
+

diff --git a/lib/icinga/externalcommand.h b/lib/icinga/externalcommand.h
index 4881ed143809d6f98a0e69f100ddd00ef1c87c5e..bbeb598ddd0e23c960d73a88de73af274efb67fc 100644 (file)
--- a/lib/icinga/externalcommand.h
+++ b/lib/icinga/externalcommand.h
@@ -35,6 +35,8 @@ public:
        static void EnableSvcCheck(double time, const vector<String>& arguments);
        static void DisableSvcCheck(double time, const vector<String>& arguments);
        static void ShutdownProcess(double time, const vector<String>& arguments);
+       static void ScheduleForcedHostSvcChecks(double time, const vector<String>& arguments);
+       static void ScheduleHostSvcChecks(double time, const vector<String>& arguments);
 
 private:
        typedef function<void (double time, const vector<String>& arguments)> Callback;