Make sure the dependency graph is properly updated when adding and removing objects

[icinga2] / lib / icinga / downtime.ti

diff --git a/lib/icinga/downtime.ti b/lib/icinga/downtime.ti
index 8532ae6f507c4498e81378a7f0a945f6294e2142..5c7a19bdd55eca588fe944f9ee38b37ebf7d211f 100644 (file)
--- a/lib/icinga/downtime.ti
+++ b/lib/icinga/downtime.ti
@@ -1,6 +1,6 @@
 /******************************************************************************
  * Icinga 2                                                                   *
- * Copyright (C) 2012-2015 Icinga Development Team (http://www.icinga.org)    *
+ * Copyright (C) 2012-2016 Icinga Development Team (https://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                *
@@ -17,29 +17,73 @@
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.             *
  ******************************************************************************/
 
+#include "base/configobject.hpp"
+#include "base/utility.hpp"
+#impl_include "icinga/service.hpp"
+
+library icinga;
+
 namespace icinga
 {
 
-class Downtime
+code {{{
+class I2_ICINGA_API DowntimeNameComposer : public NameComposer
+{
+public:
+       virtual String MakeName(const String& shortName, const Object::Ptr& context) const;
+       virtual Dictionary::Ptr ParseName(const String& name) const;
+};
+}}}
+
+class Downtime : ConfigObject < DowntimeNameComposer
 {
-       [state] String id;
-       [state] double entry_time;
-       [state] String author;
-       [state] String comment;
-       [state] double start_time;
-       [state] double end_time;
+       load_after Host;
+       load_after Service;
+
+       [config, protected, required, navigation(host)] name(Host) host_name {
+               navigate {{{
+                       return Host::GetByName(GetHostName());
+               }}}
+       };
+       [config, protected, navigation(service)] String service_name {
+               track {{{
+                       if (!oldValue.IsEmpty()) {
+                               Service::Ptr service = Service::GetByNamePair(GetHostName(), oldValue);
+                               DependencyGraph::RemoveDependency(this, service.get());
+                       }
+
+                       if (!newValue.IsEmpty()) {
+                               Service::Ptr service = Service::GetByNamePair(GetHostName(), newValue);
+                               DependencyGraph::AddDependency(this, service.get());
+                       }
+               }}}
+               navigate {{{
+                       if (GetServiceName().IsEmpty())
+                               return Service::Ptr();
+
+                       Host::Ptr host = Host::GetByName(GetHostName());
+                       return host->GetServiceByShortName(GetServiceName());
+               }}}
+       };
+
+       [state] double entry_time {
+               default {{{ return Utility::GetTime(); }}}
+       };
+       [config, required] String author;
+       [config, required] String comment;
+       [config] double start_time;
+       [config] double end_time;
        [state] double trigger_time;
-       [state] bool fixed;
-       [state] double duration;
-       [state] int triggered_by_legacy_id;
-       [state] String triggered_by;
-       [state] String scheduled_by;
-       [state] Dictionary::Ptr triggers {
-               default {{{ return new Dictionary(); }}}
+       [config] bool fixed;
+       [config] double duration;
+       [config] name(Downtime) triggered_by;
+       [config] String scheduled_by;
+       [state] Array::Ptr triggers {
+               default {{{ return new Array(); }}}
        };
        [state] int legacy_id;
        [state] bool was_cancelled;
-       [state] String config_owner;
+       [config] String config_owner;
 };
 
 }