diff --git a/lib/icinga/commentprocessor.cpp b/lib/icinga/commentprocessor.cpp index 92a9b2946..d0485f0a1 100644 --- a/lib/icinga/commentprocessor.cpp +++ b/lib/icinga/commentprocessor.cpp @@ -47,10 +47,6 @@ String CommentProcessor::AddComment(const DynamicObject::Ptr& owner, if (!comments) comments = boost::make_shared(); - int legacy_id = m_NextCommentID; - m_NextCommentID++; - comment->Set("legacy_id", legacy_id); - String id = Utility::NewUUID(); comments->Set(id, comment); owner->Set("comments", comments); @@ -80,7 +76,12 @@ void CommentProcessor::RemoveComment(const String& id) String CommentProcessor::GetIDFromLegacyID(int id) { - return Convert::ToString(id); + map::iterator it = m_LegacyCommentCache.find(id); + + if (it == m_LegacyCommentCache.end()) + throw_exception(invalid_argument("Invalid legacy comment ID specified.")); + + return it->second; } DynamicObject::Ptr CommentProcessor::GetOwnerByCommentID(const String& id) @@ -124,15 +125,29 @@ void CommentProcessor::AddCommentsToCache(const DynamicObject::Ptr& owner) String id; Dictionary::Ptr comment; BOOST_FOREACH(tie(id, comment), comments) { - if (comment->Contains("legacy_id")) { - int legacy_id = comment->Get("legacy_id"); + int legacy_id; - m_LegacyCommentCache[legacy_id] = id; - - if (legacy_id > m_NextCommentID) - m_NextCommentID = legacy_id; + if (!comment->Contains("legacy_id")) { + legacy_id = m_NextCommentID; + m_NextCommentID++; + comment->Set("legacy_id", legacy_id); + } else { + legacy_id = comment->Get("legacy_id"); } + if (legacy_id >= m_NextCommentID) + m_NextCommentID = legacy_id + 1; + + if (m_LegacyCommentCache.find(legacy_id) != m_LegacyCommentCache.end()) { + /* The legacy_id is already in use by another comment; + * this shouldn't usually happen - assign it a new ID */ + + legacy_id = m_NextCommentID; + m_NextCommentID++; + comment->Set("legacy_id", legacy_id); + } + + m_LegacyCommentCache[legacy_id] = id; m_CommentCache[id] = owner; } } diff --git a/lib/icinga/commentprocessor.h b/lib/icinga/commentprocessor.h index 744e3e1a0..32ad57363 100644 --- a/lib/icinga/commentprocessor.h +++ b/lib/icinga/commentprocessor.h @@ -53,6 +53,7 @@ public: static Dictionary::Ptr GetCommentByID(const String& id); static void InvalidateCommentCache(void); + static void ValidateCommentCache(void); private: static int m_NextCommentID; @@ -64,7 +65,6 @@ private: CommentProcessor(void); static void AddCommentsToCache(const DynamicObject::Ptr& owner); - static void ValidateCommentCache(void); }; } diff --git a/lib/icinga/downtimeprocessor.cpp b/lib/icinga/downtimeprocessor.cpp index 08fdeb1c8..a62931f01 100644 --- a/lib/icinga/downtimeprocessor.cpp +++ b/lib/icinga/downtimeprocessor.cpp @@ -52,10 +52,6 @@ String DowntimeProcessor::AddDowntime(const DynamicObject::Ptr& owner, if (!downtimes) downtimes = boost::make_shared(); - int legacy_id = m_NextDowntimeID; - m_NextDowntimeID++; - downtime->Set("legacy_id", legacy_id); - String id = Utility::NewUUID(); downtimes->Set(id, downtime); owner->Set("downtimes", downtimes); @@ -77,7 +73,12 @@ void DowntimeProcessor::RemoveDowntime(const String& id) String DowntimeProcessor::GetIDFromLegacyID(int id) { - return Convert::ToString(id); + map::iterator it = m_LegacyDowntimeCache.find(id); + + if (it == m_LegacyDowntimeCache.end()) + throw_exception(invalid_argument("Invalid legacy downtime ID specified.")); + + return it->second; } DynamicObject::Ptr DowntimeProcessor::GetOwnerByDowntimeID(const String& id) @@ -140,15 +141,28 @@ void DowntimeProcessor::AddDowntimesToCache(const DynamicObject::Ptr& owner) String id; Dictionary::Ptr downtime; BOOST_FOREACH(tie(id, downtime), downtimes) { - if (downtime->Contains("legacy_id")) { - int legacy_id = downtime->Get("legacy_id"); + int legacy_id; - m_LegacyDowntimeCache[legacy_id] = id; - - if (legacy_id > m_NextDowntimeID) - m_NextDowntimeID = legacy_id; + if (!downtime->Contains("legacy_id")) { + legacy_id = m_NextDowntimeID; + m_NextDowntimeID++; + downtime->Set("legacy_id", legacy_id); + } else { + legacy_id = downtime->Get("legacy_id"); } + if (legacy_id >= m_NextDowntimeID) + m_NextDowntimeID = legacy_id + 1; + + if (m_LegacyDowntimeCache.find(legacy_id) != m_LegacyDowntimeCache.end()) { + /* The legacy_id is already in use by another downtime; + * this shouldn't usually happen - assign it a new ID. */ + legacy_id = m_NextDowntimeID; + m_NextDowntimeID++; + downtime->Set("legacy_id", legacy_id); + } + + m_LegacyDowntimeCache[legacy_id] = id; m_DowntimeCache[id] = owner; } } diff --git a/lib/icinga/downtimeprocessor.h b/lib/icinga/downtimeprocessor.h index 9c2090d57..6612a63ba 100644 --- a/lib/icinga/downtimeprocessor.h +++ b/lib/icinga/downtimeprocessor.h @@ -47,6 +47,7 @@ public: static bool IsDowntimeActive(const Dictionary::Ptr& downtime); static void InvalidateDowntimeCache(void); + static void ValidateDowntimeCache(void); private: static int m_NextDowntimeID; @@ -58,7 +59,6 @@ private: DowntimeProcessor(void); static void AddDowntimesToCache(const DynamicObject::Ptr& owner); - static void ValidateDowntimeCache(void); }; } diff --git a/lib/icinga/host.cpp b/lib/icinga/host.cpp index 6bf67d07f..ec0a1daeb 100644 --- a/lib/icinga/host.cpp +++ b/lib/icinga/host.cpp @@ -120,11 +120,15 @@ Dictionary::Ptr Host::GetMacros(void) const Dictionary::Ptr Host::GetDowntimes(void) const { + DowntimeProcessor::ValidateDowntimeCache(); + return Get("downtimes"); } Dictionary::Ptr Host::GetComments(void) const { + CommentProcessor::ValidateCommentCache(); + return Get("comments"); } diff --git a/lib/icinga/service.cpp b/lib/icinga/service.cpp index e7815f4cc..b9848bbd0 100644 --- a/lib/icinga/service.cpp +++ b/lib/icinga/service.cpp @@ -120,11 +120,15 @@ Dictionary::Ptr Service::GetMacros(void) const Dictionary::Ptr Service::GetDowntimes(void) const { + DowntimeProcessor::ValidateDowntimeCache(); + return Get("downtimes"); } Dictionary::Ptr Service::GetComments(void) const { + CommentProcessor::ValidateCommentCache(); + return Get("comments"); }