Implement apply support for dependencies.

Refs #5880

diff --git a/lib/icinga/CMakeLists.txt b/lib/icinga/CMakeLists.txt
index 5242952b76f939a03dd385459da21dcda1860db5..ab9ed192ddd5772a9b483aa808955f1514f659d0 100644 (file)
--- a/lib/icinga/CMakeLists.txt
+++ b/lib/icinga/CMakeLists.txt
@@ -42,7 +42,7 @@ mkembedconfig_target(icinga-type.conf icinga-type.cpp)
 add_library(icinga SHARED
   api.cpp api.h checkcommand.cpp checkcommand.th checkresult.cpp checkresult.th
   cib.cpp command.cpp command.th comment.cpp comment.th compatutility.cpp dependency.cpp dependency.th
-  domain.cpp domain.th downtime.cpp downtime.th eventcommand.cpp eventcommand.th
+  dependency-apply.cpp domain.cpp domain.th downtime.cpp downtime.th eventcommand.cpp eventcommand.th
   externalcommandprocessor.cpp host.cpp host.th hostgroup.cpp hostgroup.th
   icingaapplication.cpp icingaapplication.th icingastatuswriter.cpp
   icingastatuswriter.th legacytimeperiod.cpp

diff --git a/lib/icinga/dependency-apply.cpp b/lib/icinga/dependency-apply.cpp
new file mode 100644 (file)
index 0000000..989090e
--- /dev/null
+++ b/lib/icinga/dependency-apply.cpp
@@ -0,0 +1,80 @@
+/******************************************************************************
+ * Icinga 2                                                                   *
+ * Copyright (C) 2012-present 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     *
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.             *
+ ******************************************************************************/
+
+#include "icinga/dependency.h"
+#include "config/configitembuilder.h"
+#include "base/initialize.h"
+#include "base/dynamictype.h"
+#include "base/convert.h"
+#include "base/logger_fwd.h"
+#include "base/context.h"
+#include <boost/foreach.hpp>
+
+using namespace icinga;
+
+INITIALIZE_ONCE(&Dependency::RegisterApplyRuleHandler);
+
+void Dependency::RegisterApplyRuleHandler(void)
+{
+       ApplyRule::RegisterType("Dependency", &Dependency::EvaluateApplyRules, 2);
+}
+
+void Dependency::EvaluateApplyRules(const std::vector<ApplyRule>& rules)
+{
+       BOOST_FOREACH(const Service::Ptr& service, DynamicType::GetObjects<Service>()) {
+               CONTEXT("Evaluating 'apply' rules for Service '" + service->GetName() + "'");
+
+               Dictionary::Ptr locals = make_shared<Dictionary>();
+               locals->Set("host", service->GetHost());
+               locals->Set("service", service);
+
+               BOOST_FOREACH(const ApplyRule& rule, rules) {
+                       DebugInfo di = rule.GetDebugInfo();
+
+                       std::ostringstream msgbuf;
+                       msgbuf << "Evaluating 'apply' rule (" << di << ")";
+                       CONTEXT(msgbuf.str());
+
+                       if (!rule.EvaluateFilter(locals))
+                               continue;
+
+                       std::ostringstream msgbuf2;
+                       msgbuf2 << "Applying dependency '" << rule.GetName() << "' to service '" << service->GetName() << "' for rule " << di;
+                       Log(LogDebug, "icinga", msgbuf2.str());
+
+                       std::ostringstream namebuf;
+                       namebuf << service->GetName() << "!" << rule.GetName();
+                       String name = namebuf.str();
+
+                       ConfigItemBuilder::Ptr builder = make_shared<ConfigItemBuilder>(di);
+                       builder->SetType("Dependency");
+                       builder->SetName(name);
+                       builder->SetScope(rule.GetScope());
+
+                       builder->AddExpression(make_shared<AExpression>(&AExpression::OpSet, "child_host", make_shared<AExpression>(&AExpression::OpLiteral, service->GetHost()->GetName(), di), di));
+                       builder->AddExpression(make_shared<AExpression>(&AExpression::OpSet, "child_service", make_shared<AExpression>(&AExpression::OpLiteral, service->GetShortName(), di), di));
+                       builder->AddExpression(rule.GetExpression());
+
+                       ConfigItem::Ptr serviceItem = builder->Compile();
+                       serviceItem->Register();
+                       DynamicObject::Ptr dobj = serviceItem->Commit();
+                       dobj->OnConfigLoaded();
+               }
+       }
+}

diff --git a/lib/icinga/dependency.h b/lib/icinga/dependency.h
index 481dedc1d4056bec8f1586479418244edb30f17e..1daccd72115bc4201d97f9fc9c28ab5dae2bb306 100644 (file)
--- a/lib/icinga/dependency.h
+++ b/lib/icinga/dependency.h
@@ -23,6 +23,7 @@
 #include "icinga/i2-icinga.h"
 #include "icinga/dependency.th"
 #include "icinga/service.h"
+#include "config/applyrule.h"
 #include "base/array.h"
 #include "base/dictionary.h"
 
@@ -47,6 +48,9 @@ public:
 
        bool IsAvailable(DependencyType dt) const;
 
+       static void RegisterApplyRuleHandler(void);
+       static void EvaluateApplyRules(const std::vector<ApplyRule>& rules);
+
 protected:
        virtual void OnStateLoaded(void);
        virtual void Stop(void);