/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */ #include "icingadb/icingadb.hpp" #include "icingadb/redisconnection.hpp" #include "base/configtype.hpp" #include "base/configobject.hpp" #include "base/defer.hpp" #include "base/json.hpp" #include "base/logger.hpp" #include "base/serializer.hpp" #include "base/shared.hpp" #include "base/tlsutility.hpp" #include "base/initialize.hpp" #include "base/convert.hpp" #include "base/array.hpp" #include "base/exception.hpp" #include "base/utility.hpp" #include "base/object-packer.hpp" #include "icinga/command.hpp" #include "icinga/compatutility.hpp" #include "icinga/customvarobject.hpp" #include "icinga/host.hpp" #include "icinga/service.hpp" #include "icinga/hostgroup.hpp" #include "icinga/servicegroup.hpp" #include "icinga/usergroup.hpp" #include "icinga/checkcommand.hpp" #include "icinga/eventcommand.hpp" #include "icinga/notificationcommand.hpp" #include "icinga/timeperiod.hpp" #include "icinga/pluginutility.hpp" #include "remote/zone.hpp" #include #include #include #include #include #include #include #include #include using namespace icinga; using Prio = RedisConnection::QueryPriority; std::unordered_set IcingaDB::m_IndexedTypes; INITIALIZE_ONCE(&IcingaDB::ConfigStaticInitialize); std::vector IcingaDB::GetTypes() { // The initial config sync will queue the types in the following order. return { // Sync them first to get their states ASAP. Host::TypeInstance, Service::TypeInstance, // Then sync them for similar reasons. Downtime::TypeInstance, Comment::TypeInstance, HostGroup::TypeInstance, ServiceGroup::TypeInstance, CheckCommand::TypeInstance, Endpoint::TypeInstance, EventCommand::TypeInstance, Notification::TypeInstance, NotificationCommand::TypeInstance, TimePeriod::TypeInstance, User::TypeInstance, UserGroup::TypeInstance, Zone::TypeInstance }; } void IcingaDB::ConfigStaticInitialize() { for (auto& type : GetTypes()) { m_IndexedTypes.emplace(type.get()); } /* triggered in ProcessCheckResult(), requires UpdateNextCheck() to be called before */ Checkable::OnStateChange.connect([](const Checkable::Ptr& checkable, const CheckResult::Ptr& cr, StateType type, const MessageOrigin::Ptr&) { IcingaDB::StateChangeHandler(checkable, cr, type); }); Checkable::OnAcknowledgementSet.connect([](const Checkable::Ptr& checkable, const String& author, const String& comment, AcknowledgementType type, bool, bool persistent, double changeTime, double expiry, const MessageOrigin::Ptr&) { AcknowledgementSetHandler(checkable, author, comment, type, persistent, changeTime, expiry); }); Checkable::OnAcknowledgementCleared.connect([](const Checkable::Ptr& checkable, const String& removedBy, double changeTime, const MessageOrigin::Ptr&) { AcknowledgementClearedHandler(checkable, removedBy, changeTime); }); Checkable::OnReachabilityChanged.connect([](const Checkable::Ptr&, const CheckResult::Ptr&, std::set children, const MessageOrigin::Ptr&) { IcingaDB::ReachabilityChangeHandler(children); }); /* triggered on create, update and delete objects */ ConfigObject::OnActiveChanged.connect([](const ConfigObject::Ptr& object, const Value&) { IcingaDB::VersionChangedHandler(object); }); ConfigObject::OnVersionChanged.connect([](const ConfigObject::Ptr& object, const Value&) { IcingaDB::VersionChangedHandler(object); }); /* downtime start */ Downtime::OnDowntimeTriggered.connect(&IcingaDB::DowntimeStartedHandler); /* fixed/flexible downtime end or remove */ Downtime::OnDowntimeRemoved.connect(&IcingaDB::DowntimeRemovedHandler); Checkable::OnNotificationSentToAllUsers.connect([]( const Notification::Ptr& notification, const Checkable::Ptr& checkable, const std::set& users, const NotificationType& type, const CheckResult::Ptr& cr, const String& author, const String& text, const MessageOrigin::Ptr& ) { IcingaDB::NotificationSentToAllUsersHandler(notification, checkable, users, type, cr, author, text); }); Comment::OnCommentAdded.connect(&IcingaDB::CommentAddedHandler); Comment::OnCommentRemoved.connect(&IcingaDB::CommentRemovedHandler); Checkable::OnFlappingChange.connect(&IcingaDB::FlappingChangeHandler); Checkable::OnNewCheckResult.connect([](const Checkable::Ptr& checkable, const CheckResult::Ptr&, const MessageOrigin::Ptr&) { IcingaDB::NewCheckResultHandler(checkable); }); Checkable::OnNextCheckUpdated.connect([](const Checkable::Ptr& checkable) { IcingaDB::NextCheckUpdatedHandler(checkable); }); Service::OnHostProblemChanged.connect([](const Service::Ptr& service, const CheckResult::Ptr&, const MessageOrigin::Ptr&) { IcingaDB::HostProblemChangedHandler(service); }); Notification::OnUsersRawChangedWithOldValue.connect([](const Notification::Ptr& notification, const Value& oldValues, const Value& newValues) { IcingaDB::NotificationUsersChangedHandler(notification, oldValues, newValues); }); Notification::OnUserGroupsRawChangedWithOldValue.connect([](const Notification::Ptr& notification, const Value& oldValues, const Value& newValues) { IcingaDB::NotificationUserGroupsChangedHandler(notification, oldValues, newValues); }); TimePeriod::OnRangesChangedWithOldValue.connect([](const TimePeriod::Ptr& timeperiod, const Value& oldValues, const Value& newValues) { IcingaDB::TimePeriodRangesChangedHandler(timeperiod, oldValues, newValues); }); TimePeriod::OnIncludesChangedWithOldValue.connect([](const TimePeriod::Ptr& timeperiod, const Value& oldValues, const Value& newValues) { IcingaDB::TimePeriodIncludesChangedHandler(timeperiod, oldValues, newValues); }); TimePeriod::OnExcludesChangedWithOldValue.connect([](const TimePeriod::Ptr& timeperiod, const Value& oldValues, const Value& newValues) { IcingaDB::TimePeriodExcludesChangedHandler(timeperiod, oldValues, newValues); }); User::OnGroupsChangedWithOldValue.connect([](const User::Ptr& user, const Value& oldValues, const Value& newValues) { IcingaDB::UserGroupsChangedHandler(user, oldValues, newValues); }); Host::OnGroupsChangedWithOldValue.connect([](const Host::Ptr& host, const Value& oldValues, const Value& newValues) { IcingaDB::HostGroupsChangedHandler(host, oldValues, newValues); }); Service::OnGroupsChangedWithOldValue.connect([](const Service::Ptr& service, const Value& oldValues, const Value& newValues) { IcingaDB::ServiceGroupsChangedHandler(service, oldValues, newValues); }); Command::OnEnvChangedWithOldValue.connect([](const ConfigObject::Ptr& command, const Value& oldValues, const Value& newValues) { IcingaDB::CommandEnvChangedHandler(command, oldValues, newValues); }); Command::OnArgumentsChangedWithOldValue.connect([](const ConfigObject::Ptr& command, const Value& oldValues, const Value& newValues) { IcingaDB::CommandArgumentsChangedHandler(command, oldValues, newValues); }); CustomVarObject::OnVarsChangedWithOldValue.connect([](const ConfigObject::Ptr& object, const Value& oldValues, const Value& newValues) { IcingaDB::CustomVarsChangedHandler(object, oldValues, newValues); }); } void IcingaDB::UpdateAllConfigObjects() { m_Rcon->Sync(); m_Rcon->FireAndForgetQuery({"XADD", "icinga:schema", "MAXLEN", "1", "*", "version", "5"}, Prio::Heartbeat); Log(LogInformation, "IcingaDB") << "Starting initial config/status dump"; double startTime = Utility::GetTime(); SetOngoingDumpStart(startTime); Defer resetOngoingDumpStart ([this]() { SetOngoingDumpStart(0); }); // Use a Workqueue to pack objects in parallel WorkQueue upq(25000, Configuration::Concurrency, LogNotice); upq.SetName("IcingaDB:ConfigDump"); std::vector types = GetTypes(); m_Rcon->SuppressQueryKind(Prio::CheckResult); m_Rcon->SuppressQueryKind(Prio::RuntimeStateSync); Defer unSuppress ([this]() { m_Rcon->UnsuppressQueryKind(Prio::RuntimeStateSync); m_Rcon->UnsuppressQueryKind(Prio::CheckResult); }); // Add a new type=* state=wip entry to the stream and remove all previous entries (MAXLEN 1). m_Rcon->FireAndForgetQuery({"XADD", "icinga:dump", "MAXLEN", "1", "*", "key", "*", "state", "wip"}, Prio::Config); const std::vector globalKeys = { m_PrefixConfigObject + "customvar", m_PrefixConfigObject + "action:url", m_PrefixConfigObject + "notes:url", m_PrefixConfigObject + "icon:image", }; DeleteKeys(m_Rcon, globalKeys, Prio::Config); DeleteKeys(m_Rcon, {"icinga:nextupdate:host", "icinga:nextupdate:service"}, Prio::Config); m_Rcon->Sync(); Defer resetDumpedGlobals ([this]() { m_DumpedGlobals.CustomVar.Reset(); m_DumpedGlobals.ActionUrl.Reset(); m_DumpedGlobals.NotesUrl.Reset(); m_DumpedGlobals.IconImage.Reset(); }); upq.ParallelFor(types, false, [this](const Type::Ptr& type) { String lcType = type->GetName().ToLower(); ConfigType *ctype = dynamic_cast(type.get()); if (!ctype) return; auto& rcon (m_Rcons.at(ctype)); std::vector keys = GetTypeOverwriteKeys(lcType); DeleteKeys(rcon, keys, Prio::Config); WorkQueue upqObjectType(25000, Configuration::Concurrency, LogNotice); upqObjectType.SetName("IcingaDB:ConfigDump:" + lcType); std::map redisCheckSums; String configCheckSum = m_PrefixConfigCheckSum + lcType; upqObjectType.Enqueue([&rcon, &configCheckSum, &redisCheckSums]() { String cursor = "0"; do { Array::Ptr res = rcon->GetResultOfQuery({ "HSCAN", configCheckSum, cursor, "COUNT", "1000" }, Prio::Config); AddKvsToMap(res->Get(1), redisCheckSums); cursor = res->Get(0); } while (cursor != "0"); }); auto objectChunks (ChunkObjects(ctype->GetObjects(), 500)); String configObject = m_PrefixConfigObject + lcType; // Skimmed away attributes and checksums HMSETs' keys and values by Redis key. std::map>> ourContentRaw {{configCheckSum, {}}, {configObject, {}}}; std::mutex ourContentMutex; upqObjectType.ParallelFor(objectChunks, [&](decltype(objectChunks)::const_reference chunk) { std::map> hMSets; // Two values are appended per object: Object ID (Hash encoded) and Object State (IcingaDB::SerializeState() -> JSON encoded) std::vector states = {"HMSET", m_PrefixConfigObject + lcType + ":state"}; // Two values are appended per object: Object ID (Hash encoded) and State Checksum ({ "checksum": checksum } -> JSON encoded) std::vector statesChksms = {"HMSET", m_PrefixConfigCheckSum + lcType + ":state"}; std::vector > transaction = {{"MULTI"}}; std::vector hostZAdds = {"ZADD", "icinga:nextupdate:host"}, serviceZAdds = {"ZADD", "icinga:nextupdate:service"}; auto skimObjects ([&]() { std::lock_guard l (ourContentMutex); for (auto& kv : ourContentRaw) { auto pos (hMSets.find(kv.first)); if (pos != hMSets.end()) { kv.second.emplace_back(std::move(pos->second)); hMSets.erase(pos); } } }); bool dumpState = (lcType == "host" || lcType == "service"); size_t bulkCounter = 0; for (const ConfigObject::Ptr& object : chunk) { if (lcType != GetLowerCaseTypeNameDB(object)) continue; std::vector runtimeUpdates; CreateConfigUpdate(object, lcType, hMSets, runtimeUpdates, false); // Write out inital state for checkables if (dumpState) { String objectKey = GetObjectIdentifier(object); Dictionary::Ptr state = SerializeState(dynamic_pointer_cast(object)); states.emplace_back(objectKey); states.emplace_back(JsonEncode(state)); statesChksms.emplace_back(objectKey); statesChksms.emplace_back(JsonEncode(new Dictionary({{"checksum", HashValue(state)}}))); } bulkCounter++; if (!(bulkCounter % 100)) { skimObjects(); for (auto& kv : hMSets) { if (!kv.second.empty()) { kv.second.insert(kv.second.begin(), {"HMSET", kv.first}); transaction.emplace_back(std::move(kv.second)); } } if (states.size() > 2) { transaction.emplace_back(std::move(states)); transaction.emplace_back(std::move(statesChksms)); states = {"HMSET", m_PrefixConfigObject + lcType + ":state"}; statesChksms = {"HMSET", m_PrefixConfigCheckSum + lcType + ":state"}; } hMSets = decltype(hMSets)(); if (transaction.size() > 1) { transaction.push_back({"EXEC"}); rcon->FireAndForgetQueries(std::move(transaction), Prio::Config); transaction = {{"MULTI"}}; } } auto checkable (dynamic_pointer_cast(object)); if (checkable && checkable->GetEnableActiveChecks()) { auto zAdds (dynamic_pointer_cast(checkable) ? &serviceZAdds : &hostZAdds); zAdds->emplace_back(Convert::ToString(checkable->GetNextUpdate())); zAdds->emplace_back(GetObjectIdentifier(checkable)); if (zAdds->size() >= 102u) { std::vector header (zAdds->begin(), zAdds->begin() + 2u); rcon->FireAndForgetQuery(std::move(*zAdds), Prio::CheckResult); *zAdds = std::move(header); } } } skimObjects(); for (auto& kv : hMSets) { if (!kv.second.empty()) { kv.second.insert(kv.second.begin(), {"HMSET", kv.first}); transaction.emplace_back(std::move(kv.second)); } } if (states.size() > 2) { transaction.emplace_back(std::move(states)); transaction.emplace_back(std::move(statesChksms)); } if (transaction.size() > 1) { transaction.push_back({"EXEC"}); rcon->FireAndForgetQueries(std::move(transaction), Prio::Config); } for (auto zAdds : {&hostZAdds, &serviceZAdds}) { if (zAdds->size() > 2u) { rcon->FireAndForgetQuery(std::move(*zAdds), Prio::CheckResult); } } Log(LogNotice, "IcingaDB") << "Dumped " << bulkCounter << " objects of type " << lcType; }); upqObjectType.Join(); if (upqObjectType.HasExceptions()) { for (boost::exception_ptr exc : upqObjectType.GetExceptions()) { if (exc) { boost::rethrow_exception(exc); } } } std::map> ourContent; for (auto& source : ourContentRaw) { auto& dest (ourContent[source.first]); upqObjectType.Enqueue([&]() { for (auto& hMSet : source.second) { for (decltype(hMSet.size()) i = 0, stop = hMSet.size() - 1u; i < stop; i += 2u) { dest.emplace(std::move(hMSet[i]), std::move(hMSet[i + 1u])); } hMSet.clear(); } source.second.clear(); }); } upqObjectType.Join(); ourContentRaw.clear(); auto& ourCheckSums (ourContent[configCheckSum]); auto& ourObjects (ourContent[configObject]); std::vector setChecksum, setObject, delChecksum, delObject; auto redisCurrent (redisCheckSums.begin()); auto redisEnd (redisCheckSums.end()); auto ourCurrent (ourCheckSums.begin()); auto ourEnd (ourCheckSums.end()); auto flushSets ([&]() { auto affectedConfig (setObject.size() / 2u); setChecksum.insert(setChecksum.begin(), {"HMSET", configCheckSum}); setObject.insert(setObject.begin(), {"HMSET", configObject}); std::vector> transaction; transaction.emplace_back(std::vector{"MULTI"}); transaction.emplace_back(std::move(setChecksum)); transaction.emplace_back(std::move(setObject)); transaction.emplace_back(std::vector{"EXEC"}); setChecksum.clear(); setObject.clear(); rcon->FireAndForgetQueries(std::move(transaction), Prio::Config, {affectedConfig}); }); auto flushDels ([&]() { auto affectedConfig (delObject.size()); delChecksum.insert(delChecksum.begin(), {"HDEL", configCheckSum}); delObject.insert(delObject.begin(), {"HDEL", configObject}); std::vector> transaction; transaction.emplace_back(std::vector{"MULTI"}); transaction.emplace_back(std::move(delChecksum)); transaction.emplace_back(std::move(delObject)); transaction.emplace_back(std::vector{"EXEC"}); delChecksum.clear(); delObject.clear(); rcon->FireAndForgetQueries(std::move(transaction), Prio::Config, {affectedConfig}); }); auto setOne ([&]() { setChecksum.emplace_back(ourCurrent->first); setChecksum.emplace_back(ourCurrent->second); setObject.emplace_back(ourCurrent->first); setObject.emplace_back(ourObjects[ourCurrent->first]); if (setChecksum.size() == 100u) { flushSets(); } }); auto delOne ([&]() { delChecksum.emplace_back(redisCurrent->first); delObject.emplace_back(redisCurrent->first); if (delChecksum.size() == 100u) { flushDels(); } }); for (;;) { if (redisCurrent == redisEnd) { for (; ourCurrent != ourEnd; ++ourCurrent) { setOne(); } break; } else if (ourCurrent == ourEnd) { for (; redisCurrent != redisEnd; ++redisCurrent) { delOne(); } break; } else if (redisCurrent->first < ourCurrent->first) { delOne(); ++redisCurrent; } else if (redisCurrent->first > ourCurrent->first) { setOne(); ++ourCurrent; } else { if (redisCurrent->second != ourCurrent->second) { setOne(); } ++redisCurrent; ++ourCurrent; } } if (delChecksum.size()) { flushDels(); } if (setChecksum.size()) { flushSets(); } for (auto& key : GetTypeDumpSignalKeys(type)) { rcon->FireAndForgetQuery({"XADD", "icinga:dump", "*", "key", key, "state", "done"}, Prio::Config); } rcon->Sync(); }); upq.Join(); if (upq.HasExceptions()) { for (boost::exception_ptr exc : upq.GetExceptions()) { try { if (exc) { boost::rethrow_exception(exc); } } catch(const std::exception& e) { Log(LogCritical, "IcingaDB") << "Exception during ConfigDump: " << e.what(); } } } for (auto& key : globalKeys) { m_Rcon->FireAndForgetQuery({"XADD", "icinga:dump", "*", "key", key, "state", "done"}, Prio::Config); } m_Rcon->FireAndForgetQuery({"XADD", "icinga:dump", "*", "key", "*", "state", "done"}, Prio::Config); // enqueue a callback that will notify us once all previous queries were executed and wait for this event std::promise p; m_Rcon->EnqueueCallback([&p](boost::asio::yield_context& yc) { p.set_value(); }, Prio::Config); p.get_future().wait(); auto endTime (Utility::GetTime()); auto took (endTime - startTime); SetLastdumpTook(took); SetLastdumpEnd(endTime); Log(LogInformation, "IcingaDB") << "Initial config/status dump finished in " << took << " seconds."; } std::vector>> IcingaDB::ChunkObjects(std::vector> objects, size_t chunkSize) { std::vector>> chunks; auto offset (objects.begin()); auto end (objects.end()); chunks.reserve((std::distance(offset, end) + chunkSize - 1) / chunkSize); while (std::distance(offset, end) >= chunkSize) { auto until (offset + chunkSize); chunks.emplace_back(offset, until); offset = until; } if (offset != end) { chunks.emplace_back(offset, end); } return std::move(chunks); } void IcingaDB::DeleteKeys(const RedisConnection::Ptr& conn, const std::vector& keys, RedisConnection::QueryPriority priority) { std::vector query = {"DEL"}; for (auto& key : keys) { query.emplace_back(key); } conn->FireAndForgetQuery(std::move(query), priority); } std::vector IcingaDB::GetTypeOverwriteKeys(const String& type) { std::vector keys = { m_PrefixConfigObject + type + ":customvar", }; if (type == "host" || type == "service" || type == "user") { keys.emplace_back(m_PrefixConfigObject + type + "group:member"); keys.emplace_back(m_PrefixConfigObject + type + ":state"); keys.emplace_back(m_PrefixConfigCheckSum + type + ":state"); } else if (type == "timeperiod") { keys.emplace_back(m_PrefixConfigObject + type + ":override:include"); keys.emplace_back(m_PrefixConfigObject + type + ":override:exclude"); keys.emplace_back(m_PrefixConfigObject + type + ":range"); } else if (type == "notification") { keys.emplace_back(m_PrefixConfigObject + type + ":user"); keys.emplace_back(m_PrefixConfigObject + type + ":usergroup"); keys.emplace_back(m_PrefixConfigObject + type + ":recipient"); } else if (type == "checkcommand" || type == "notificationcommand" || type == "eventcommand") { keys.emplace_back(m_PrefixConfigObject + type + ":envvar"); keys.emplace_back(m_PrefixConfigCheckSum + type + ":envvar"); keys.emplace_back(m_PrefixConfigObject + type + ":argument"); keys.emplace_back(m_PrefixConfigCheckSum + type + ":argument"); } return std::move(keys); } std::vector IcingaDB::GetTypeDumpSignalKeys(const Type::Ptr& type) { String lcType = type->GetName().ToLower(); std::vector keys = {m_PrefixConfigObject + lcType}; if (CustomVarObject::TypeInstance->IsAssignableFrom(type)) { keys.emplace_back(m_PrefixConfigObject + lcType + ":customvar"); } if (type == Host::TypeInstance || type == Service::TypeInstance) { keys.emplace_back(m_PrefixConfigObject + lcType + "group:member"); keys.emplace_back(m_PrefixConfigObject + lcType + ":state"); } else if (type == User::TypeInstance) { keys.emplace_back(m_PrefixConfigObject + lcType + "group:member"); } else if (type == TimePeriod::TypeInstance) { keys.emplace_back(m_PrefixConfigObject + lcType + ":override:include"); keys.emplace_back(m_PrefixConfigObject + lcType + ":override:exclude"); keys.emplace_back(m_PrefixConfigObject + lcType + ":range"); } else if (type == Notification::TypeInstance) { keys.emplace_back(m_PrefixConfigObject + lcType + ":user"); keys.emplace_back(m_PrefixConfigObject + lcType + ":usergroup"); keys.emplace_back(m_PrefixConfigObject + lcType + ":recipient"); } else if (type == CheckCommand::TypeInstance || type == NotificationCommand::TypeInstance || type == EventCommand::TypeInstance) { keys.emplace_back(m_PrefixConfigObject + lcType + ":envvar"); keys.emplace_back(m_PrefixConfigObject + lcType + ":argument"); } return std::move(keys); } template static ConfigObject::Ptr GetObjectByName(const String& name) { return ConfigObject::GetObject(name); } void IcingaDB::InsertObjectDependencies(const ConfigObject::Ptr& object, const String typeName, std::map>& hMSets, std::vector& runtimeUpdates, bool runtimeUpdate) { String objectKey = GetObjectIdentifier(object); String objectKeyName = typeName + "_id"; Type::Ptr type = object->GetReflectionType(); CustomVarObject::Ptr customVarObject = dynamic_pointer_cast(object); if (customVarObject) { auto vars(SerializeVars(customVarObject->GetVars())); if (vars) { auto& typeCvs (hMSets[m_PrefixConfigObject + typeName + ":customvar"]); auto& allCvs (hMSets[m_PrefixConfigObject + "customvar"]); ObjectLock varsLock(vars); Array::Ptr varsArray(new Array); varsArray->Reserve(vars->GetLength()); for (auto& kv : vars) { if (runtimeUpdate || m_DumpedGlobals.CustomVar.IsNew(kv.first)) { allCvs.emplace_back(kv.first); allCvs.emplace_back(JsonEncode(kv.second)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, kv.first, m_PrefixConfigObject + "customvar", kv.second); } } String id = HashValue(new Array({m_EnvironmentId, kv.first, object->GetName()})); typeCvs.emplace_back(id); Dictionary::Ptr data = new Dictionary({{objectKeyName, objectKey}, {"environment_id", m_EnvironmentId}, {"customvar_id", kv.first}}); typeCvs.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":customvar", data); } } } } if (type == Host::TypeInstance || type == Service::TypeInstance) { Checkable::Ptr checkable = static_pointer_cast(object); String actionUrl = checkable->GetActionUrl(); String notesUrl = checkable->GetNotesUrl(); String iconImage = checkable->GetIconImage(); if (!actionUrl.IsEmpty()) { auto& actionUrls (hMSets[m_PrefixConfigObject + "action:url"]); auto id (HashValue(new Array({m_EnvironmentId, actionUrl}))); if (runtimeUpdate || m_DumpedGlobals.ActionUrl.IsNew(id)) { actionUrls.emplace_back(std::move(id)); Dictionary::Ptr data = new Dictionary({{"environment_id", m_EnvironmentId}, {"action_url", actionUrl}}); actionUrls.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, actionUrls.at(actionUrls.size() - 2u), m_PrefixConfigObject + "action:url", data); } } } if (!notesUrl.IsEmpty()) { auto& notesUrls (hMSets[m_PrefixConfigObject + "notes:url"]); auto id (HashValue(new Array({m_EnvironmentId, notesUrl}))); if (runtimeUpdate || m_DumpedGlobals.NotesUrl.IsNew(id)) { notesUrls.emplace_back(std::move(id)); Dictionary::Ptr data = new Dictionary({{"environment_id", m_EnvironmentId}, {"notes_url", notesUrl}}); notesUrls.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, notesUrls.at(notesUrls.size() - 2u), m_PrefixConfigObject + "notes:url", data); } } } if (!iconImage.IsEmpty()) { auto& iconImages (hMSets[m_PrefixConfigObject + "icon:image"]); auto id (HashValue(new Array({m_EnvironmentId, iconImage}))); if (runtimeUpdate || m_DumpedGlobals.IconImage.IsNew(id)) { iconImages.emplace_back(std::move(id)); Dictionary::Ptr data = new Dictionary({{"environment_id", m_EnvironmentId}, {"icon_image", iconImage}}); iconImages.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, iconImages.at(iconImages.size() - 2u), m_PrefixConfigObject + "icon:image", data); } } } Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); ConfigObject::Ptr (*getGroup)(const String& name); Array::Ptr groups; if (service) { groups = service->GetGroups(); getGroup = &::GetObjectByName; } else { groups = host->GetGroups(); getGroup = &::GetObjectByName; } if (groups) { ObjectLock groupsLock(groups); Array::Ptr groupIds(new Array); groupIds->Reserve(groups->GetLength()); auto& members (hMSets[m_PrefixConfigObject + typeName + "group:member"]); for (auto& group : groups) { auto groupObj ((*getGroup)(group)); String groupId = GetObjectIdentifier(groupObj); String id = HashValue(new Array({m_EnvironmentId, groupObj->GetName(), object->GetName()})); members.emplace_back(id); Dictionary::Ptr data = new Dictionary({{objectKeyName, objectKey}, {"environment_id", m_EnvironmentId}, {typeName + "group_id", groupId}}); members.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + "group:member", data); } groupIds->Add(groupId); } } return; } if (type == TimePeriod::TypeInstance) { TimePeriod::Ptr timeperiod = static_pointer_cast(object); Dictionary::Ptr ranges = timeperiod->GetRanges(); if (ranges) { ObjectLock rangesLock(ranges); Array::Ptr rangeIds(new Array); auto& typeRanges (hMSets[m_PrefixConfigObject + typeName + ":range"]); rangeIds->Reserve(ranges->GetLength()); for (auto& kv : ranges) { String rangeId = HashValue(new Array({m_EnvironmentId, kv.first, kv.second})); rangeIds->Add(rangeId); String id = HashValue(new Array({m_EnvironmentId, kv.first, kv.second, object->GetName()})); typeRanges.emplace_back(id); Dictionary::Ptr data = new Dictionary({{"environment_id", m_EnvironmentId}, {"timeperiod_id", objectKey}, {"range_key", kv.first}, {"range_value", kv.second}}); typeRanges.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":range", data); } } } Array::Ptr includes; ConfigObject::Ptr (*getInclude)(const String& name); includes = timeperiod->GetIncludes(); getInclude = &::GetObjectByName; Array::Ptr includeChecksums = new Array(); ObjectLock includesLock(includes); ObjectLock includeChecksumsLock(includeChecksums); includeChecksums->Reserve(includes->GetLength()); auto& includs (hMSets[m_PrefixConfigObject + typeName + ":override:include"]); for (auto include : includes) { auto includeTp ((*getInclude)(include.Get())); String includeId = GetObjectIdentifier(includeTp); includeChecksums->Add(includeId); String id = HashValue(new Array({m_EnvironmentId, includeTp->GetName(), object->GetName()})); includs.emplace_back(id); Dictionary::Ptr data = new Dictionary({{"environment_id", m_EnvironmentId}, {"timeperiod_id", objectKey}, {"include_id", includeId}}); includs.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":override:include", data); } } Array::Ptr excludes; ConfigObject::Ptr (*getExclude)(const String& name); excludes = timeperiod->GetExcludes(); getExclude = &::GetObjectByName; Array::Ptr excludeChecksums = new Array(); ObjectLock excludesLock(excludes); ObjectLock excludeChecksumsLock(excludeChecksums); excludeChecksums->Reserve(excludes->GetLength()); auto& excluds (hMSets[m_PrefixConfigObject + typeName + ":override:exclude"]); for (auto exclude : excludes) { auto excludeTp ((*getExclude)(exclude.Get())); String excludeId = GetObjectIdentifier(excludeTp); excludeChecksums->Add(excludeId); String id = HashValue(new Array({m_EnvironmentId, excludeTp->GetName(), object->GetName()})); excluds.emplace_back(id); Dictionary::Ptr data = new Dictionary({{"environment_id", m_EnvironmentId}, {"timeperiod_id", objectKey}, {"exclude_id", excludeId}}); excluds.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":override:exclude", data); } } return; } if (type == User::TypeInstance) { User::Ptr user = static_pointer_cast(object); Array::Ptr groups = user->GetGroups(); if (groups) { ObjectLock groupsLock(groups); Array::Ptr groupIds(new Array); groupIds->Reserve(groups->GetLength()); auto& members (hMSets[m_PrefixConfigObject + typeName + "group:member"]); auto& notificationRecipients (hMSets[m_PrefixConfigObject + "notification:recipient"]); for (auto& group : groups) { UserGroup::Ptr groupObj = UserGroup::GetByName(group); String groupId = GetObjectIdentifier(groupObj); String id = HashValue(new Array({m_EnvironmentId, groupObj->GetName(), object->GetName()})); members.emplace_back(id); Dictionary::Ptr data = new Dictionary({{"user_id", objectKey}, {"environment_id", m_EnvironmentId}, {"usergroup_id", groupId}}); members.emplace_back(JsonEncode(data)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + "group:member", data); // Recipients are handled by notifications during initial dumps and only need to be handled here during runtime (e.g. User creation). for (auto& notification : groupObj->GetNotifications()) { String recipientId = HashValue(new Array({m_EnvironmentId, "usergroupuser", user->GetName(), groupObj->GetName(), notification->GetName()})); notificationRecipients.emplace_back(recipientId); Dictionary::Ptr recipientData = new Dictionary({{"notification_id", GetObjectIdentifier(notification)}, {"environment_id", m_EnvironmentId}, {"user_id", objectKey}, {"usergroup_id", groupId}}); notificationRecipients.emplace_back(JsonEncode(recipientData)); AddObjectDataToRuntimeUpdates(runtimeUpdates, recipientId, m_PrefixConfigObject + "notification:recipient", recipientData); } } groupIds->Add(groupId); } } return; } if (type == Notification::TypeInstance) { Notification::Ptr notification = static_pointer_cast(object); std::set users = notification->GetUsers(); Array::Ptr userIds = new Array(); auto usergroups(notification->GetUserGroups()); Array::Ptr usergroupIds = new Array(); userIds->Reserve(users.size()); auto& usrs (hMSets[m_PrefixConfigObject + typeName + ":user"]); auto& notificationRecipients (hMSets[m_PrefixConfigObject + typeName + ":recipient"]); for (auto& user : users) { String userId = GetObjectIdentifier(user); String id = HashValue(new Array({m_EnvironmentId, "user", user->GetName(), object->GetName()})); usrs.emplace_back(id); notificationRecipients.emplace_back(id); Dictionary::Ptr data = new Dictionary({{"notification_id", objectKey}, {"environment_id", m_EnvironmentId}, {"user_id", userId}}); String dataJson = JsonEncode(data); usrs.emplace_back(dataJson); notificationRecipients.emplace_back(dataJson); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":user", data); AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":recipient", data); } userIds->Add(userId); } usergroupIds->Reserve(usergroups.size()); auto& groups (hMSets[m_PrefixConfigObject + typeName + ":usergroup"]); for (auto& usergroup : usergroups) { String usergroupId = GetObjectIdentifier(usergroup); String id = HashValue(new Array({m_EnvironmentId, "usergroup", usergroup->GetName(), object->GetName()})); groups.emplace_back(id); notificationRecipients.emplace_back(id); Dictionary::Ptr groupData = new Dictionary({{"notification_id", objectKey}, {"environment_id", m_EnvironmentId}, {"usergroup_id", usergroupId}}); String groupDataJson = JsonEncode(groupData); groups.emplace_back(groupDataJson); notificationRecipients.emplace_back(groupDataJson); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":usergroup", groupData); AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":recipient", groupData); } for (const User::Ptr& user : usergroup->GetMembers()) { String userId = GetObjectIdentifier(user); String recipientId = HashValue(new Array({m_EnvironmentId, "usergroupuser", user->GetName(), usergroup->GetName(), notification->GetName()})); notificationRecipients.emplace_back(recipientId); Dictionary::Ptr userData = new Dictionary({{"notification_id", objectKey}, {"environment_id", m_EnvironmentId}, {"user_id", userId}, {"usergroup_id", usergroupId}}); notificationRecipients.emplace_back(JsonEncode(userData)); if (runtimeUpdate) { AddObjectDataToRuntimeUpdates(runtimeUpdates, recipientId, m_PrefixConfigObject + typeName + ":recipient", userData); } } usergroupIds->Add(usergroupId); } return; } if (type == CheckCommand::TypeInstance || type == NotificationCommand::TypeInstance || type == EventCommand::TypeInstance) { Command::Ptr command = static_pointer_cast(object); Dictionary::Ptr arguments = command->GetArguments(); if (arguments) { ObjectLock argumentsLock(arguments); auto& typeArgs (hMSets[m_PrefixConfigObject + typeName + ":argument"]); auto& argChksms (hMSets[m_PrefixConfigCheckSum + typeName + ":argument"]); for (auto& kv : arguments) { Dictionary::Ptr values; if (kv.second.IsObjectType()) { values = kv.second; values = values->ShallowClone(); } else if (kv.second.IsObjectType()) { values = new Dictionary({{"value", JsonEncode(kv.second)}}); } else { values = new Dictionary({{"value", kv.second}}); } for (const char *attr : {"value", "set_if", "separator"}) { Value value; // Stringify if set. if (values->Get(attr, &value)) { switch (value.GetType()) { case ValueEmpty: case ValueString: break; case ValueObject: values->Set(attr, value.Get()->ToString()); break; default: values->Set(attr, JsonEncode(value)); } } } { Value order; // Intify if set. if (values->Get("order", &order)) { values->Set("order", (int)order); } } values->Set(objectKeyName, objectKey); values->Set("argument_key", kv.first); values->Set("environment_id", m_EnvironmentId); String id = HashValue(new Array({m_EnvironmentId, kv.first, object->GetName()})); typeArgs.emplace_back(id); typeArgs.emplace_back(JsonEncode(values)); argChksms.emplace_back(id); String checksum = HashValue(kv.second); argChksms.emplace_back(JsonEncode(new Dictionary({{"checksum", checksum}}))); if (runtimeUpdate) { values->Set("checksum", checksum); AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":argument", values); } } } Dictionary::Ptr envvars = command->GetEnv(); if (envvars) { ObjectLock envvarsLock(envvars); Array::Ptr envvarIds(new Array); auto& typeVars (hMSets[m_PrefixConfigObject + typeName + ":envvar"]); auto& varChksms (hMSets[m_PrefixConfigCheckSum + typeName + ":envvar"]); envvarIds->Reserve(envvars->GetLength()); for (auto& kv : envvars) { Dictionary::Ptr values; if (kv.second.IsObjectType()) { values = kv.second; values = values->ShallowClone(); } else if (kv.second.IsObjectType()) { values = new Dictionary({{"value", JsonEncode(kv.second)}}); } else { values = new Dictionary({{"value", kv.second}}); } { Value value; // JsonEncode() the value if it's set. if (values->Get("value", &value)) { values->Set("value", JsonEncode(value)); } } values->Set(objectKeyName, objectKey); values->Set("envvar_key", kv.first); values->Set("environment_id", m_EnvironmentId); String id = HashValue(new Array({m_EnvironmentId, kv.first, object->GetName()})); typeVars.emplace_back(id); typeVars.emplace_back(JsonEncode(values)); varChksms.emplace_back(id); String checksum = HashValue(kv.second); varChksms.emplace_back(JsonEncode(new Dictionary({{"checksum", checksum}}))); if (runtimeUpdate) { values->Set("checksum", checksum); AddObjectDataToRuntimeUpdates(runtimeUpdates, id, m_PrefixConfigObject + typeName + ":envvar", values); } } } return; } } /** * Update the state information of a checkable in Redis. * * What is updated exactly depends on the mode parameter: * - Volatile: Update the volatile state information stored in icinga:host:state or icinga:service:state as well as * the corresponding checksum stored in icinga:checksum:host:state or icinga:checksum:service:state. * - RuntimeOnly: Write a runtime update to the icinga:runtime:state stream. It is up to the caller to ensure that * identical volatile state information was already written before to avoid inconsistencies. This mode is only * useful to upgrade a previous Volatile to a Full operation, otherwise Full should be used. * - Full: Perform an update of all state information in Redis, that is updating the volatile information and sending * a corresponding runtime update so that this state update gets written through to the persistent database by a * running icingadb process. * * @param checkable State of this checkable is updated in Redis * @param mode Mode of operation (StateUpdate::Volatile, StateUpdate::RuntimeOnly, or StateUpdate::Full) */ void IcingaDB::UpdateState(const Checkable::Ptr& checkable, StateUpdate mode) { if (!m_Rcon || !m_Rcon->IsConnected()) return; String objectType = GetLowerCaseTypeNameDB(checkable); String objectKey = GetObjectIdentifier(checkable); Dictionary::Ptr stateAttrs = SerializeState(checkable); String redisStateKey = m_PrefixConfigObject + objectType + ":state"; String redisChecksumKey = m_PrefixConfigCheckSum + objectType + ":state"; String checksum = HashValue(stateAttrs); if (mode & StateUpdate::Volatile) { m_Rcon->FireAndForgetQueries({ {"HSET", redisStateKey, objectKey, JsonEncode(stateAttrs)}, {"HSET", redisChecksumKey, objectKey, JsonEncode(new Dictionary({{"checksum", checksum}}))}, }, Prio::RuntimeStateSync); } if (mode & StateUpdate::RuntimeOnly) { ObjectLock olock(stateAttrs); std::vector streamadd({ "XADD", "icinga:runtime:state", "MAXLEN", "~", "1000000", "*", "runtime_type", "upsert", "redis_key", redisStateKey, "checksum", checksum, }); for (const Dictionary::Pair& kv : stateAttrs) { streamadd.emplace_back(kv.first); streamadd.emplace_back(IcingaToStreamValue(kv.second)); } m_Rcon->FireAndForgetQuery(std::move(streamadd), Prio::RuntimeStateStream, {0, 1}); } } // Used to update a single object, used for runtime updates void IcingaDB::SendConfigUpdate(const ConfigObject::Ptr& object, bool runtimeUpdate) { if (!m_Rcon || !m_Rcon->IsConnected()) return; String typeName = GetLowerCaseTypeNameDB(object); std::map> hMSets; std::vector runtimeUpdates; CreateConfigUpdate(object, typeName, hMSets, runtimeUpdates, runtimeUpdate); Checkable::Ptr checkable = dynamic_pointer_cast(object); if (checkable) { UpdateState(checkable, runtimeUpdate ? StateUpdate::Full : StateUpdate::Volatile); } std::vector > transaction = {{"MULTI"}}; for (auto& kv : hMSets) { if (!kv.second.empty()) { kv.second.insert(kv.second.begin(), {"HMSET", kv.first}); transaction.emplace_back(std::move(kv.second)); } } for (auto& objectAttributes : runtimeUpdates) { std::vector xAdd({"XADD", "icinga:runtime", "MAXLEN", "~", "1000000", "*"}); ObjectLock olock(objectAttributes); for (const Dictionary::Pair& kv : objectAttributes) { String value = IcingaToStreamValue(kv.second); if (!value.IsEmpty()) { xAdd.emplace_back(kv.first); xAdd.emplace_back(value); } } transaction.emplace_back(std::move(xAdd)); } if (transaction.size() > 1) { transaction.push_back({"EXEC"}); m_Rcon->FireAndForgetQueries(std::move(transaction), Prio::Config, {1}); } if (checkable) { SendNextUpdate(checkable); } } void IcingaDB::AddObjectDataToRuntimeUpdates(std::vector& runtimeUpdates, const String& objectKey, const String& redisKey, const Dictionary::Ptr& data) { Dictionary::Ptr dataClone = data->ShallowClone(); dataClone->Set("id", objectKey); dataClone->Set("redis_key", redisKey); dataClone->Set("runtime_type", "upsert"); runtimeUpdates.emplace_back(dataClone); } // Takes object and collects IcingaDB relevant attributes and computes checksums. Returns whether the object is relevant // for IcingaDB. bool IcingaDB::PrepareObject(const ConfigObject::Ptr& object, Dictionary::Ptr& attributes, Dictionary::Ptr& checksums) { attributes->Set("name_checksum", SHA1(object->GetName())); attributes->Set("environment_id", m_EnvironmentId); attributes->Set("name", object->GetName()); Zone::Ptr ObjectsZone; Type::Ptr type = object->GetReflectionType(); if (type == Endpoint::TypeInstance) { ObjectsZone = static_cast(object.get())->GetZone(); } else { ObjectsZone = static_pointer_cast(object->GetZone()); } if (ObjectsZone) { attributes->Set("zone_id", GetObjectIdentifier(ObjectsZone)); attributes->Set("zone_name", ObjectsZone->GetName()); } if (type == Endpoint::TypeInstance) { return true; } if (type == Zone::TypeInstance) { Zone::Ptr zone = static_pointer_cast(object); attributes->Set("is_global", zone->GetGlobal()); Zone::Ptr parent = zone->GetParent(); if (parent) { attributes->Set("parent_id", GetObjectIdentifier(parent)); } auto parentsRaw (zone->GetAllParentsRaw()); attributes->Set("depth", parentsRaw.size()); return true; } if (type == Host::TypeInstance || type == Service::TypeInstance) { Checkable::Ptr checkable = static_pointer_cast(object); attributes->Set("checkcommand_name", checkable->GetCheckCommand()->GetName()); attributes->Set("max_check_attempts", checkable->GetMaxCheckAttempts()); attributes->Set("check_timeout", checkable->GetCheckTimeout()); attributes->Set("check_interval", checkable->GetCheckInterval()); attributes->Set("check_retry_interval", checkable->GetRetryInterval()); attributes->Set("active_checks_enabled", checkable->GetEnableActiveChecks()); attributes->Set("passive_checks_enabled", checkable->GetEnablePassiveChecks()); attributes->Set("event_handler_enabled", checkable->GetEnableEventHandler()); attributes->Set("notifications_enabled", checkable->GetEnableNotifications()); attributes->Set("flapping_enabled", checkable->GetEnableFlapping()); attributes->Set("flapping_threshold_low", checkable->GetFlappingThresholdLow()); attributes->Set("flapping_threshold_high", checkable->GetFlappingThresholdHigh()); attributes->Set("perfdata_enabled", checkable->GetEnablePerfdata()); attributes->Set("is_volatile", checkable->GetVolatile()); attributes->Set("notes", checkable->GetNotes()); attributes->Set("icon_image_alt", checkable->GetIconImageAlt()); attributes->Set("checkcommand_id", GetObjectIdentifier(checkable->GetCheckCommand())); Endpoint::Ptr commandEndpoint = checkable->GetCommandEndpoint(); if (commandEndpoint) { attributes->Set("command_endpoint_id", GetObjectIdentifier(commandEndpoint)); attributes->Set("command_endpoint_name", commandEndpoint->GetName()); } TimePeriod::Ptr timePeriod = checkable->GetCheckPeriod(); if (timePeriod) { attributes->Set("check_timeperiod_id", GetObjectIdentifier(timePeriod)); attributes->Set("check_timeperiod_name", timePeriod->GetName()); } EventCommand::Ptr eventCommand = checkable->GetEventCommand(); if (eventCommand) { attributes->Set("eventcommand_id", GetObjectIdentifier(eventCommand)); attributes->Set("eventcommand_name", eventCommand->GetName()); } String actionUrl = checkable->GetActionUrl(); String notesUrl = checkable->GetNotesUrl(); String iconImage = checkable->GetIconImage(); if (!actionUrl.IsEmpty()) attributes->Set("action_url_id", HashValue(new Array({m_EnvironmentId, actionUrl}))); if (!notesUrl.IsEmpty()) attributes->Set("notes_url_id", HashValue(new Array({m_EnvironmentId, notesUrl}))); if (!iconImage.IsEmpty()) attributes->Set("icon_image_id", HashValue(new Array({m_EnvironmentId, iconImage}))); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); if (service) { attributes->Set("host_id", GetObjectIdentifier(service->GetHost())); attributes->Set("display_name", service->GetDisplayName()); // Overwrite name here, `object->name` is 'HostName!ServiceName' but we only want the name of the Service attributes->Set("name", service->GetShortName()); } else { attributes->Set("display_name", host->GetDisplayName()); attributes->Set("address", host->GetAddress()); attributes->Set("address6", host->GetAddress6()); } return true; } if (type == User::TypeInstance) { User::Ptr user = static_pointer_cast(object); attributes->Set("display_name", user->GetDisplayName()); attributes->Set("email", user->GetEmail()); attributes->Set("pager", user->GetPager()); attributes->Set("notifications_enabled", user->GetEnableNotifications()); attributes->Set("states", user->GetStates()); attributes->Set("types", user->GetTypes()); if (user->GetPeriod()) attributes->Set("timeperiod_id", GetObjectIdentifier(user->GetPeriod())); return true; } if (type == TimePeriod::TypeInstance) { TimePeriod::Ptr timeperiod = static_pointer_cast(object); attributes->Set("display_name", timeperiod->GetDisplayName()); attributes->Set("prefer_includes", timeperiod->GetPreferIncludes()); return true; } if (type == Notification::TypeInstance) { Notification::Ptr notification = static_pointer_cast(object); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(notification->GetCheckable()); attributes->Set("notificationcommand_id", GetObjectIdentifier(notification->GetCommand())); attributes->Set("host_id", GetObjectIdentifier(host)); if (service) attributes->Set("service_id", GetObjectIdentifier(service)); TimePeriod::Ptr timeperiod = notification->GetPeriod(); if (timeperiod) attributes->Set("timeperiod_id", GetObjectIdentifier(timeperiod)); if (notification->GetTimes()) { attributes->Set("times_begin", notification->GetTimes()->Get("begin")); attributes->Set("times_end",notification->GetTimes()->Get("end")); } attributes->Set("notification_interval", notification->GetInterval()); attributes->Set("states", notification->GetStates()); attributes->Set("types", notification->GetTypes()); return true; } if (type == Comment::TypeInstance) { Comment::Ptr comment = static_pointer_cast(object); attributes->Set("author", comment->GetAuthor()); attributes->Set("text", comment->GetText()); attributes->Set("entry_type", comment->GetEntryType()); attributes->Set("entry_time", TimestampToMilliseconds(comment->GetEntryTime())); attributes->Set("is_persistent", comment->GetPersistent()); attributes->Set("is_sticky", comment->GetSticky()); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(comment->GetCheckable()); attributes->Set("host_id", GetObjectIdentifier(host)); if (service) { attributes->Set("object_type", "service"); attributes->Set("service_id", GetObjectIdentifier(service)); } else attributes->Set("object_type", "host"); auto expireTime (comment->GetExpireTime()); if (expireTime > 0) { attributes->Set("expire_time", TimestampToMilliseconds(expireTime)); } return true; } if (type == Downtime::TypeInstance) { Downtime::Ptr downtime = static_pointer_cast(object); attributes->Set("author", downtime->GetAuthor()); attributes->Set("comment", downtime->GetComment()); attributes->Set("entry_time", TimestampToMilliseconds(downtime->GetEntryTime())); attributes->Set("scheduled_start_time", TimestampToMilliseconds(downtime->GetStartTime())); attributes->Set("scheduled_end_time", TimestampToMilliseconds(downtime->GetEndTime())); attributes->Set("scheduled_duration", TimestampToMilliseconds(downtime->GetEndTime() - downtime->GetStartTime())); attributes->Set("flexible_duration", TimestampToMilliseconds(downtime->GetDuration())); attributes->Set("is_flexible", !downtime->GetFixed()); attributes->Set("is_in_effect", downtime->IsInEffect()); if (downtime->IsInEffect()) { attributes->Set("start_time", TimestampToMilliseconds(downtime->GetTriggerTime())); attributes->Set("end_time", TimestampToMilliseconds(downtime->GetFixed() ? downtime->GetEndTime() : (downtime->GetTriggerTime() + downtime->GetDuration()))); } auto duration = downtime->GetDuration(); if (downtime->GetFixed()) { duration = downtime->GetEndTime() - downtime->GetStartTime(); } attributes->Set("duration", TimestampToMilliseconds(duration)); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(downtime->GetCheckable()); attributes->Set("host_id", GetObjectIdentifier(host)); if (service) { attributes->Set("object_type", "service"); attributes->Set("service_id", GetObjectIdentifier(service)); } else attributes->Set("object_type", "host"); auto triggeredBy (Downtime::GetByName(downtime->GetTriggeredBy())); if (triggeredBy) { attributes->Set("triggered_by_id", GetObjectIdentifier(triggeredBy)); } auto scheduledBy (downtime->GetScheduledBy()); if (!scheduledBy.IsEmpty()) { attributes->Set("scheduled_by", scheduledBy); } auto parent (Downtime::GetByName(downtime->GetParent())); if (parent) { attributes->Set("parent_id", GetObjectIdentifier(parent)); } return true; } if (type == UserGroup::TypeInstance) { UserGroup::Ptr userGroup = static_pointer_cast(object); attributes->Set("display_name", userGroup->GetDisplayName()); return true; } if (type == HostGroup::TypeInstance) { HostGroup::Ptr hostGroup = static_pointer_cast(object); attributes->Set("display_name", hostGroup->GetDisplayName()); return true; } if (type == ServiceGroup::TypeInstance) { ServiceGroup::Ptr serviceGroup = static_pointer_cast(object); attributes->Set("display_name", serviceGroup->GetDisplayName()); return true; } if (type == CheckCommand::TypeInstance || type == NotificationCommand::TypeInstance || type == EventCommand::TypeInstance) { Command::Ptr command = static_pointer_cast(object); attributes->Set("command", JsonEncode(command->GetCommandLine())); attributes->Set("timeout", command->GetTimeout()); return true; } return false; } /* Creates a config update with computed checksums etc. * Writes attributes, customVars and checksums into the respective supplied vectors. Adds two values to each vector * (if applicable), first the key then the value. To use in a Redis command the command (e.g. HSET) and the key (e.g. * icinga:config:object:downtime) need to be prepended. There is nothing to indicate success or failure. */ void IcingaDB::CreateConfigUpdate(const ConfigObject::Ptr& object, const String typeName, std::map>& hMSets, std::vector& runtimeUpdates, bool runtimeUpdate) { /* TODO: This isn't essentially correct as we don't keep track of config objects ourselves. This would avoid duplicated config updates at startup. if (!runtimeUpdate && m_ConfigDumpInProgress) return; */ if (m_Rcon == nullptr) return; Dictionary::Ptr attr = new Dictionary; Dictionary::Ptr chksm = new Dictionary; if (!PrepareObject(object, attr, chksm)) return; InsertObjectDependencies(object, typeName, hMSets, runtimeUpdates, runtimeUpdate); String objectKey = GetObjectIdentifier(object); auto& attrs (hMSets[m_PrefixConfigObject + typeName]); auto& chksms (hMSets[m_PrefixConfigCheckSum + typeName]); attrs.emplace_back(objectKey); attrs.emplace_back(JsonEncode(attr)); String checksum = HashValue(attr); chksms.emplace_back(objectKey); chksms.emplace_back(JsonEncode(new Dictionary({{"checksum", checksum}}))); /* Send an update event to subscribers. */ if (runtimeUpdate) { attr->Set("checksum", checksum); AddObjectDataToRuntimeUpdates(runtimeUpdates, objectKey, m_PrefixConfigObject + typeName, attr); } } void IcingaDB::SendConfigDelete(const ConfigObject::Ptr& object) { Type::Ptr type = object->GetReflectionType(); String typeName = type->GetName().ToLower(); String objectKey = GetObjectIdentifier(object); m_Rcon->FireAndForgetQueries({ {"HDEL", m_PrefixConfigObject + typeName, objectKey}, {"HDEL", m_PrefixConfigCheckSum + typeName, objectKey}, { "XADD", "icinga:runtime", "MAXLEN", "~", "1000000", "*", "redis_key", m_PrefixConfigObject + typeName, "id", objectKey, "runtime_type", "delete" } }, Prio::Config); CustomVarObject::Ptr customVarObject = dynamic_pointer_cast(object); if (customVarObject) { Dictionary::Ptr vars = customVarObject->GetVars(); SendCustomVarsChanged(object, vars, nullptr); } if (type == Host::TypeInstance || type == Service::TypeInstance) { Checkable::Ptr checkable = static_pointer_cast(object); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); m_Rcon->FireAndForgetQuery({ "ZREM", service ? "icinga:nextupdate:service" : "icinga:nextupdate:host", GetObjectIdentifier(checkable) }, Prio::CheckResult); m_Rcon->FireAndForgetQueries({ {"HDEL", m_PrefixConfigObject + typeName + ":state", objectKey}, {"HDEL", m_PrefixConfigCheckSum + typeName + ":state", objectKey} }, Prio::RuntimeStateSync); if (service) { SendGroupsChanged(checkable, service->GetGroups(), nullptr); } else { SendGroupsChanged(checkable, host->GetGroups(), nullptr); } return; } if (type == TimePeriod::TypeInstance) { TimePeriod::Ptr timeperiod = static_pointer_cast(object); SendTimePeriodRangesChanged(timeperiod, timeperiod->GetRanges(), nullptr); SendTimePeriodIncludesChanged(timeperiod, timeperiod->GetIncludes(), nullptr); SendTimePeriodExcludesChanged(timeperiod, timeperiod->GetExcludes(), nullptr); return; } if (type == User::TypeInstance) { User::Ptr user = static_pointer_cast(object); SendGroupsChanged(user, user->GetGroups(), nullptr); return; } if (type == Notification::TypeInstance) { Notification::Ptr notification = static_pointer_cast(object); SendNotificationUsersChanged(notification, notification->GetUsersRaw(), nullptr); SendNotificationUserGroupsChanged(notification, notification->GetUserGroupsRaw(), nullptr); return; } if (type == CheckCommand::TypeInstance || type == NotificationCommand::TypeInstance || type == EventCommand::TypeInstance) { Command::Ptr command = static_pointer_cast(object); SendCommandArgumentsChanged(command, command->GetArguments(), nullptr); SendCommandEnvChanged(command, command->GetEnv(), nullptr); return; } } static inline unsigned short GetPreviousState(const Checkable::Ptr& checkable, const Service::Ptr& service, StateType type) { auto phs ((type == StateTypeHard ? checkable->GetLastHardStatesRaw() : checkable->GetLastSoftStatesRaw()) % 100u); if (service) { return phs; } else { return phs == 99 ? phs : Host::CalculateState(ServiceState(phs)); } } void IcingaDB::SendStateChange(const ConfigObject::Ptr& object, const CheckResult::Ptr& cr, StateType type) { if (!GetActive()) { return; } Checkable::Ptr checkable = dynamic_pointer_cast(object); if (!checkable) return; if (!cr) return; Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); UpdateState(checkable, StateUpdate::RuntimeOnly); int hard_state; if (!cr) { hard_state = 99; } else { hard_state = service ? Convert::ToLong(service->GetLastHardState()) : Convert::ToLong(host->GetLastHardState()); } auto eventTime (cr->GetExecutionEnd()); auto eventTs (TimestampToMilliseconds(eventTime)); Array::Ptr rawId = new Array({m_EnvironmentId, object->GetName()}); rawId->Add(eventTs); std::vector xAdd ({ "XADD", "icinga:history:stream:state", "*", "id", HashValue(rawId), "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "state_type", Convert::ToString(type), "soft_state", Convert::ToString(cr ? service ? Convert::ToLong(cr->GetState()) : Convert::ToLong(Host::CalculateState(cr->GetState())) : 99), "hard_state", Convert::ToString(hard_state), "check_attempt", Convert::ToString(checkable->GetCheckAttempt()), "previous_soft_state", Convert::ToString(GetPreviousState(checkable, service, StateTypeSoft)), "previous_hard_state", Convert::ToString(GetPreviousState(checkable, service, StateTypeHard)), "max_check_attempts", Convert::ToString(checkable->GetMaxCheckAttempts()), "event_time", Convert::ToString(eventTs), "event_id", CalcEventID("state_change", object, eventTime), "event_type", "state_change" }); if (cr) { auto output (cr->GetOutput()); auto pos (output.Find("\n")); if (pos != String::NPos) { auto longOutput (output.SubStr(pos + 1u)); output.erase(output.Begin() + pos, output.End()); xAdd.emplace_back("long_output"); xAdd.emplace_back(Utility::ValidateUTF8(std::move(longOutput))); } xAdd.emplace_back("output"); xAdd.emplace_back(Utility::ValidateUTF8(std::move(output))); xAdd.emplace_back("check_source"); xAdd.emplace_back(cr->GetCheckSource()); xAdd.emplace_back("scheduling_source"); xAdd.emplace_back(cr->GetSchedulingSource()); } if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } m_HistoryBulker.ProduceOne(std::move(xAdd)); } void IcingaDB::SendSentNotification( const Notification::Ptr& notification, const Checkable::Ptr& checkable, const std::set& users, NotificationType type, const CheckResult::Ptr& cr, const String& author, const String& text, double sendTime ) { if (!GetActive()) { return; } Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); auto finalText = text; if (finalText == "" && cr) { finalText = cr->GetOutput(); } auto usersAmount (users.size()); auto sendTs (TimestampToMilliseconds(sendTime)); Array::Ptr rawId = new Array({m_EnvironmentId, notification->GetName()}); rawId->Add(GetNotificationTypeByEnum(type)); rawId->Add(sendTs); auto notificationHistoryId (HashValue(rawId)); std::vector xAdd ({ "XADD", "icinga:history:stream:notification", "*", "id", notificationHistoryId, "environment_id", m_EnvironmentId, "notification_id", GetObjectIdentifier(notification), "host_id", GetObjectIdentifier(host), "type", Convert::ToString(type), "state", Convert::ToString(cr ? service ? Convert::ToLong(cr->GetState()) : Convert::ToLong(Host::CalculateState(cr->GetState())) : 99), "previous_hard_state", Convert::ToString(cr ? Convert::ToLong(service ? cr->GetPreviousHardState() : Host::CalculateState(cr->GetPreviousHardState())) : 99), "author", Utility::ValidateUTF8(author), "text", Utility::ValidateUTF8(finalText), "users_notified", Convert::ToString(usersAmount), "send_time", Convert::ToString(sendTs), "event_id", CalcEventID("notification", notification, sendTime, type), "event_type", "notification" }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } if (!users.empty()) { Array::Ptr users_notified = new Array(); for (const User::Ptr& user : users) { users_notified->Add(GetObjectIdentifier(user)); } xAdd.emplace_back("users_notified_ids"); xAdd.emplace_back(JsonEncode(users_notified)); } m_HistoryBulker.ProduceOne(std::move(xAdd)); } void IcingaDB::SendStartedDowntime(const Downtime::Ptr& downtime) { if (!GetActive()) { return; } SendConfigUpdate(downtime, true); auto checkable (downtime->GetCheckable()); auto triggeredBy (Downtime::GetByName(downtime->GetTriggeredBy())); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); /* Update checkable state as in_downtime may have changed. */ UpdateState(checkable, StateUpdate::Full); std::vector xAdd ({ "XADD", "icinga:history:stream:downtime", "*", "downtime_id", GetObjectIdentifier(downtime), "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "entry_time", Convert::ToString(TimestampToMilliseconds(downtime->GetEntryTime())), "author", Utility::ValidateUTF8(downtime->GetAuthor()), "comment", Utility::ValidateUTF8(downtime->GetComment()), "is_flexible", Convert::ToString((unsigned short)!downtime->GetFixed()), "flexible_duration", Convert::ToString(TimestampToMilliseconds(downtime->GetDuration())), "scheduled_start_time", Convert::ToString(TimestampToMilliseconds(downtime->GetStartTime())), "scheduled_end_time", Convert::ToString(TimestampToMilliseconds(downtime->GetEndTime())), "has_been_cancelled", Convert::ToString((unsigned short)downtime->GetWasCancelled()), "trigger_time", Convert::ToString(TimestampToMilliseconds(downtime->GetTriggerTime())), "event_id", CalcEventID("downtime_start", downtime), "event_type", "downtime_start" }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } if (triggeredBy) { xAdd.emplace_back("triggered_by_id"); xAdd.emplace_back(GetObjectIdentifier(triggeredBy)); } if (downtime->GetFixed()) { xAdd.emplace_back("start_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetStartTime()))); xAdd.emplace_back("end_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetEndTime()))); } else { xAdd.emplace_back("start_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetTriggerTime()))); xAdd.emplace_back("end_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetTriggerTime() + downtime->GetDuration()))); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } auto parent (Downtime::GetByName(downtime->GetParent())); if (parent) { xAdd.emplace_back("parent_id"); xAdd.emplace_back(GetObjectIdentifier(parent)); } auto scheduledBy (downtime->GetScheduledBy()); if (!scheduledBy.IsEmpty()) { xAdd.emplace_back("scheduled_by"); xAdd.emplace_back(scheduledBy); } m_HistoryBulker.ProduceOne(std::move(xAdd)); } void IcingaDB::SendRemovedDowntime(const Downtime::Ptr& downtime) { if (!GetActive()) { return; } auto checkable (downtime->GetCheckable()); auto triggeredBy (Downtime::GetByName(downtime->GetTriggeredBy())); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); // Downtime never got triggered (didn't send "downtime_start") so we don't want to send "downtime_end" if (downtime->GetTriggerTime() == 0) return; /* Update checkable state as in_downtime may have changed. */ UpdateState(checkable, StateUpdate::Full); std::vector xAdd ({ "XADD", "icinga:history:stream:downtime", "*", "downtime_id", GetObjectIdentifier(downtime), "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "entry_time", Convert::ToString(TimestampToMilliseconds(downtime->GetEntryTime())), "author", Utility::ValidateUTF8(downtime->GetAuthor()), "cancelled_by", Utility::ValidateUTF8(downtime->GetRemovedBy()), "comment", Utility::ValidateUTF8(downtime->GetComment()), "is_flexible", Convert::ToString((unsigned short)!downtime->GetFixed()), "flexible_duration", Convert::ToString(TimestampToMilliseconds(downtime->GetDuration())), "scheduled_start_time", Convert::ToString(TimestampToMilliseconds(downtime->GetStartTime())), "scheduled_end_time", Convert::ToString(TimestampToMilliseconds(downtime->GetEndTime())), "has_been_cancelled", Convert::ToString((unsigned short)downtime->GetWasCancelled()), "trigger_time", Convert::ToString(TimestampToMilliseconds(downtime->GetTriggerTime())), "cancel_time", Convert::ToString(TimestampToMilliseconds(downtime->GetRemoveTime())), "event_id", CalcEventID("downtime_end", downtime), "event_type", "downtime_end" }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } if (triggeredBy) { xAdd.emplace_back("triggered_by_id"); xAdd.emplace_back(GetObjectIdentifier(triggeredBy)); } if (downtime->GetFixed()) { xAdd.emplace_back("start_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetStartTime()))); xAdd.emplace_back("end_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetEndTime()))); } else { xAdd.emplace_back("start_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetTriggerTime()))); xAdd.emplace_back("end_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(downtime->GetTriggerTime() + downtime->GetDuration()))); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } auto parent (Downtime::GetByName(downtime->GetParent())); if (parent) { xAdd.emplace_back("parent_id"); xAdd.emplace_back(GetObjectIdentifier(parent)); } auto scheduledBy (downtime->GetScheduledBy()); if (!scheduledBy.IsEmpty()) { xAdd.emplace_back("scheduled_by"); xAdd.emplace_back(scheduledBy); } m_HistoryBulker.ProduceOne(std::move(xAdd)); } void IcingaDB::SendAddedComment(const Comment::Ptr& comment) { if (comment->GetEntryType() != CommentUser || !GetActive()) return; auto checkable (comment->GetCheckable()); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); std::vector xAdd ({ "XADD", "icinga:history:stream:comment", "*", "comment_id", GetObjectIdentifier(comment), "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "entry_time", Convert::ToString(TimestampToMilliseconds(comment->GetEntryTime())), "author", Utility::ValidateUTF8(comment->GetAuthor()), "comment", Utility::ValidateUTF8(comment->GetText()), "entry_type", Convert::ToString(comment->GetEntryType()), "is_persistent", Convert::ToString((unsigned short)comment->GetPersistent()), "is_sticky", Convert::ToString((unsigned short)comment->GetSticky()), "event_id", CalcEventID("comment_add", comment), "event_type", "comment_add" }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } { auto expireTime (comment->GetExpireTime()); if (expireTime > 0) { xAdd.emplace_back("expire_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(expireTime))); } } m_HistoryBulker.ProduceOne(std::move(xAdd)); UpdateState(checkable, StateUpdate::Full); } void IcingaDB::SendRemovedComment(const Comment::Ptr& comment) { if (comment->GetEntryType() != CommentUser || !GetActive()) { return; } double removeTime = comment->GetRemoveTime(); bool wasRemoved = removeTime > 0; double expireTime = comment->GetExpireTime(); bool hasExpireTime = expireTime > 0; bool isExpired = hasExpireTime && expireTime <= Utility::GetTime(); if (!wasRemoved && !isExpired) { /* The comment object disappeared for no apparent reason, most likely because it simply was deleted instead * of using the proper remove-comment API action. In this case, information that should normally be set is * missing and a proper history event cannot be generated. */ return; } auto checkable (comment->GetCheckable()); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); std::vector xAdd ({ "XADD", "icinga:history:stream:comment", "*", "comment_id", GetObjectIdentifier(comment), "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "entry_time", Convert::ToString(TimestampToMilliseconds(comment->GetEntryTime())), "author", Utility::ValidateUTF8(comment->GetAuthor()), "comment", Utility::ValidateUTF8(comment->GetText()), "entry_type", Convert::ToString(comment->GetEntryType()), "is_persistent", Convert::ToString((unsigned short)comment->GetPersistent()), "is_sticky", Convert::ToString((unsigned short)comment->GetSticky()), "event_id", CalcEventID("comment_remove", comment), "event_type", "comment_remove" }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } if (wasRemoved) { xAdd.emplace_back("remove_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(removeTime))); xAdd.emplace_back("has_been_removed"); xAdd.emplace_back("1"); xAdd.emplace_back("removed_by"); xAdd.emplace_back(Utility::ValidateUTF8(comment->GetRemovedBy())); } else { xAdd.emplace_back("has_been_removed"); xAdd.emplace_back("0"); } if (hasExpireTime) { xAdd.emplace_back("expire_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(expireTime))); } m_HistoryBulker.ProduceOne(std::move(xAdd)); UpdateState(checkable, StateUpdate::Full); } void IcingaDB::SendFlappingChange(const Checkable::Ptr& checkable, double changeTime, double flappingLastChange) { if (!GetActive()) { return; } Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); std::vector xAdd ({ "XADD", "icinga:history:stream:flapping", "*", "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "flapping_threshold_low", Convert::ToString(checkable->GetFlappingThresholdLow()), "flapping_threshold_high", Convert::ToString(checkable->GetFlappingThresholdHigh()) }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } long long startTime; if (checkable->IsFlapping()) { startTime = TimestampToMilliseconds(changeTime); xAdd.emplace_back("event_type"); xAdd.emplace_back("flapping_start"); xAdd.emplace_back("percent_state_change_start"); xAdd.emplace_back(Convert::ToString(checkable->GetFlappingCurrent())); } else { startTime = TimestampToMilliseconds(flappingLastChange); xAdd.emplace_back("event_type"); xAdd.emplace_back("flapping_end"); xAdd.emplace_back("end_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(changeTime))); xAdd.emplace_back("percent_state_change_end"); xAdd.emplace_back(Convert::ToString(checkable->GetFlappingCurrent())); } xAdd.emplace_back("start_time"); xAdd.emplace_back(Convert::ToString(startTime)); xAdd.emplace_back("event_id"); xAdd.emplace_back(CalcEventID(checkable->IsFlapping() ? "flapping_start" : "flapping_end", checkable, startTime)); xAdd.emplace_back("id"); xAdd.emplace_back(HashValue(new Array({m_EnvironmentId, checkable->GetName(), startTime}))); m_HistoryBulker.ProduceOne(std::move(xAdd)); } void IcingaDB::SendNextUpdate(const Checkable::Ptr& checkable) { if (!m_Rcon || !m_Rcon->IsConnected()) return; if (checkable->GetEnableActiveChecks()) { m_Rcon->FireAndForgetQuery( { "ZADD", dynamic_pointer_cast(checkable) ? "icinga:nextupdate:service" : "icinga:nextupdate:host", Convert::ToString(checkable->GetNextUpdate()), GetObjectIdentifier(checkable) }, Prio::CheckResult ); } else { m_Rcon->FireAndForgetQuery( { "ZREM", dynamic_pointer_cast(checkable) ? "icinga:nextupdate:service" : "icinga:nextupdate:host", GetObjectIdentifier(checkable) }, Prio::CheckResult ); } } void IcingaDB::SendAcknowledgementSet(const Checkable::Ptr& checkable, const String& author, const String& comment, AcknowledgementType type, bool persistent, double changeTime, double expiry) { if (!GetActive()) { return; } Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); /* Update checkable state as is_acknowledged may have changed. */ UpdateState(checkable, StateUpdate::Full); std::vector xAdd ({ "XADD", "icinga:history:stream:acknowledgement", "*", "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "event_type", "ack_set", "author", author, "comment", comment, "is_sticky", Convert::ToString((unsigned short)(type == AcknowledgementSticky)), "is_persistent", Convert::ToString((unsigned short)persistent) }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } if (expiry > 0) { xAdd.emplace_back("expire_time"); xAdd.emplace_back(Convert::ToString(TimestampToMilliseconds(expiry))); } long long setTime = TimestampToMilliseconds(changeTime); xAdd.emplace_back("set_time"); xAdd.emplace_back(Convert::ToString(setTime)); xAdd.emplace_back("event_id"); xAdd.emplace_back(CalcEventID("ack_set", checkable, setTime)); xAdd.emplace_back("id"); xAdd.emplace_back(HashValue(new Array({m_EnvironmentId, checkable->GetName(), setTime}))); m_HistoryBulker.ProduceOne(std::move(xAdd)); } void IcingaDB::SendAcknowledgementCleared(const Checkable::Ptr& checkable, const String& removedBy, double changeTime, double ackLastChange) { if (!GetActive()) { return; } Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); /* Update checkable state as is_acknowledged may have changed. */ UpdateState(checkable, StateUpdate::Full); std::vector xAdd ({ "XADD", "icinga:history:stream:acknowledgement", "*", "environment_id", m_EnvironmentId, "host_id", GetObjectIdentifier(host), "clear_time", Convert::ToString(TimestampToMilliseconds(changeTime)), "event_type", "ack_clear" }); if (service) { xAdd.emplace_back("object_type"); xAdd.emplace_back("service"); xAdd.emplace_back("service_id"); xAdd.emplace_back(GetObjectIdentifier(checkable)); } else { xAdd.emplace_back("object_type"); xAdd.emplace_back("host"); } auto endpoint (Endpoint::GetLocalEndpoint()); if (endpoint) { xAdd.emplace_back("endpoint_id"); xAdd.emplace_back(GetObjectIdentifier(endpoint)); } long long setTime = TimestampToMilliseconds(ackLastChange); xAdd.emplace_back("set_time"); xAdd.emplace_back(Convert::ToString(setTime)); xAdd.emplace_back("event_id"); xAdd.emplace_back(CalcEventID("ack_clear", checkable, setTime)); xAdd.emplace_back("id"); xAdd.emplace_back(HashValue(new Array({m_EnvironmentId, checkable->GetName(), setTime}))); if (!removedBy.IsEmpty()) { xAdd.emplace_back("cleared_by"); xAdd.emplace_back(removedBy); } m_HistoryBulker.ProduceOne(std::move(xAdd)); } void IcingaDB::ForwardHistoryEntries() { using clock = std::chrono::steady_clock; const std::chrono::seconds logInterval (10); auto nextLog (clock::now() + logInterval); auto logPeriodically ([this, logInterval, &nextLog]() { if (clock::now() > nextLog) { nextLog += logInterval; auto size (m_HistoryBulker.Size()); Log(size > m_HistoryBulker.GetBulkSize() ? LogInformation : LogNotice, "IcingaDB") << "Pending history queries: " << size; } }); for (;;) { logPeriodically(); auto haystack (m_HistoryBulker.ConsumeMany()); if (haystack.empty()) { if (!GetActive()) { break; } continue; } uintmax_t attempts = 0; auto logFailure ([&haystack, &attempts](const char* err = nullptr) { Log msg (LogNotice, "IcingaDB"); msg << "history: " << haystack.size() << " queries failed temporarily (attempt #" << ++attempts << ")"; if (err) { msg << ": " << err; } }); for (;;) { logPeriodically(); if (m_Rcon && m_Rcon->IsConnected()) { try { m_Rcon->GetResultsOfQueries(haystack, Prio::History, {0, 0, haystack.size()}); break; } catch (const std::exception& ex) { logFailure(ex.what()); } catch (...) { logFailure(); } } else { logFailure("not connected to Redis"); } if (!GetActive()) { Log(LogCritical, "IcingaDB") << "history: " << haystack.size() << " queries failed (attempt #" << attempts << ") while we're about to shut down. Giving up and discarding additional " << m_HistoryBulker.Size() << " queued history queries."; return; } Utility::Sleep(2); } } } void IcingaDB::SendNotificationUsersChanged(const Notification::Ptr& notification, const Array::Ptr& oldValues, const Array::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedUsers = GetArrayDeletedValues(oldValues, newValues); for (const auto& userName : deletedUsers) { String id = HashValue(new Array({m_EnvironmentId, "user", userName, notification->GetName()})); DeleteRelationship(id, "notification:user"); DeleteRelationship(id, "notification:recipient"); } } void IcingaDB::SendNotificationUserGroupsChanged(const Notification::Ptr& notification, const Array::Ptr& oldValues, const Array::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedUserGroups = GetArrayDeletedValues(oldValues, newValues); for (const auto& userGroupName : deletedUserGroups) { UserGroup::Ptr userGroup = UserGroup::GetByName(userGroupName); String id = HashValue(new Array({m_EnvironmentId, "usergroup", userGroupName, notification->GetName()})); DeleteRelationship(id, "notification:usergroup"); DeleteRelationship(id, "notification:recipient"); for (const User::Ptr& user : userGroup->GetMembers()) { String userId = HashValue(new Array({m_EnvironmentId, "usergroupuser", user->GetName(), userGroupName, notification->GetName()})); DeleteRelationship(userId, "notification:recipient"); } } } void IcingaDB::SendTimePeriodRangesChanged(const TimePeriod::Ptr& timeperiod, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedKeys = GetDictionaryDeletedKeys(oldValues, newValues); String typeName = GetLowerCaseTypeNameDB(timeperiod); for (const auto& rangeKey : deletedKeys) { String id = HashValue(new Array({m_EnvironmentId, rangeKey, oldValues->Get(rangeKey), timeperiod->GetName()})); DeleteRelationship(id, "timeperiod:range"); } } void IcingaDB::SendTimePeriodIncludesChanged(const TimePeriod::Ptr& timeperiod, const Array::Ptr& oldValues, const Array::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedIncludes = GetArrayDeletedValues(oldValues, newValues); for (const auto& includeName : deletedIncludes) { String id = HashValue(new Array({m_EnvironmentId, includeName, timeperiod->GetName()})); DeleteRelationship(id, "timeperiod:override:include"); } } void IcingaDB::SendTimePeriodExcludesChanged(const TimePeriod::Ptr& timeperiod, const Array::Ptr& oldValues, const Array::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedExcludes = GetArrayDeletedValues(oldValues, newValues); for (const auto& excludeName : deletedExcludes) { String id = HashValue(new Array({m_EnvironmentId, excludeName, timeperiod->GetName()})); DeleteRelationship(id, "timeperiod:override:exclude"); } } template void IcingaDB::SendGroupsChanged(const ConfigObject::Ptr& object, const Array::Ptr& oldValues, const Array::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedGroups = GetArrayDeletedValues(oldValues, newValues); String typeName = GetLowerCaseTypeNameDB(object); for (const auto& groupName : deletedGroups) { typename T::Ptr group = ConfigObject::GetObject(groupName); String id = HashValue(new Array({m_EnvironmentId, group->GetName(), object->GetName()})); DeleteRelationship(id, typeName + "group:member"); if (std::is_same::value) { UserGroup::Ptr userGroup = dynamic_pointer_cast(group); for (const auto& notification : userGroup->GetNotifications()) { String userId = HashValue(new Array({m_EnvironmentId, "usergroupuser", object->GetName(), groupName, notification->GetName()})); DeleteRelationship(userId, "notification:recipient"); } } } } void IcingaDB::SendCommandEnvChanged(const ConfigObject::Ptr& command, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedKeys = GetDictionaryDeletedKeys(oldValues, newValues); String typeName = GetLowerCaseTypeNameDB(command); for (const auto& envvarKey : deletedKeys) { String id = HashValue(new Array({m_EnvironmentId, envvarKey, command->GetName()})); DeleteRelationship(id, typeName + ":envvar", true); } } void IcingaDB::SendCommandArgumentsChanged(const ConfigObject::Ptr& command, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } std::vector deletedKeys = GetDictionaryDeletedKeys(oldValues, newValues); String typeName = GetLowerCaseTypeNameDB(command); for (const auto& argumentKey : deletedKeys) { String id = HashValue(new Array({m_EnvironmentId, argumentKey, command->GetName()})); DeleteRelationship(id, typeName + ":argument", true); } } void IcingaDB::SendCustomVarsChanged(const ConfigObject::Ptr& object, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { if (m_IndexedTypes.find(object->GetReflectionType().get()) == m_IndexedTypes.end()) { return; } if (!m_Rcon || !m_Rcon->IsConnected() || oldValues == newValues) { return; } Dictionary::Ptr oldVars = SerializeVars(oldValues); Dictionary::Ptr newVars = SerializeVars(newValues); std::vector deletedVars = GetDictionaryDeletedKeys(oldVars, newVars); String typeName = GetLowerCaseTypeNameDB(object); for (const auto& varId : deletedVars) { String id = HashValue(new Array({m_EnvironmentId, varId, object->GetName()})); DeleteRelationship(id, typeName + ":customvar"); } } Dictionary::Ptr IcingaDB::SerializeState(const Checkable::Ptr& checkable) { Dictionary::Ptr attrs = new Dictionary(); Host::Ptr host; Service::Ptr service; tie(host, service) = GetHostService(checkable); String id = GetObjectIdentifier(checkable); /* * As there is a 1:1 relationship between host and host state, the host ID ('host_id') * is also used as the host state ID ('id'). These are duplicated to 1) avoid having * special handling for this in Icinga DB and 2) to have both a primary key and a foreign key * in the SQL database in the end. In the database 'host_id' ends up as foreign key 'host_state.host_id' * referring to 'host.id' while 'id' ends up as the primary key 'host_state.id'. This also applies for service. */ attrs->Set("id", id); attrs->Set("environment_id", m_EnvironmentId); attrs->Set("state_type", checkable->HasBeenChecked() ? checkable->GetStateType() : StateTypeHard); // TODO: last_hard/soft_state should be "previous". if (service) { attrs->Set("service_id", id); auto state = service->HasBeenChecked() ? service->GetState() : 99; attrs->Set("soft_state", state); attrs->Set("hard_state", service->HasBeenChecked() ? service->GetLastHardState() : 99); attrs->Set("severity", service->GetSeverity()); attrs->Set("host_id", GetObjectIdentifier(host)); } else { attrs->Set("host_id", id); auto state = host->HasBeenChecked() ? host->GetState() : 99; attrs->Set("soft_state", state); attrs->Set("hard_state", host->HasBeenChecked() ? host->GetLastHardState() : 99); attrs->Set("severity", host->GetSeverity()); } attrs->Set("previous_soft_state", GetPreviousState(checkable, service, StateTypeSoft)); attrs->Set("previous_hard_state", GetPreviousState(checkable, service, StateTypeHard)); attrs->Set("check_attempt", checkable->GetCheckAttempt()); attrs->Set("is_active", checkable->IsActive()); CheckResult::Ptr cr = checkable->GetLastCheckResult(); if (cr) { String rawOutput = cr->GetOutput(); if (!rawOutput.IsEmpty()) { size_t lineBreak = rawOutput.Find("\n"); String output = rawOutput.SubStr(0, lineBreak); if (!output.IsEmpty()) attrs->Set("output", rawOutput.SubStr(0, lineBreak)); if (lineBreak > 0 && lineBreak != String::NPos) { String longOutput = rawOutput.SubStr(lineBreak+1, rawOutput.GetLength()); if (!longOutput.IsEmpty()) attrs->Set("long_output", longOutput); } } String perfData = PluginUtility::FormatPerfdata(cr->GetPerformanceData()); if (!perfData.IsEmpty()) attrs->Set("performance_data", perfData); String normedPerfData = PluginUtility::FormatPerfdata(cr->GetPerformanceData(), true); if (!normedPerfData.IsEmpty()) attrs->Set("normalized_performance_data", normedPerfData); if (!cr->GetCommand().IsEmpty()) attrs->Set("check_commandline", FormatCommandLine(cr->GetCommand())); attrs->Set("execution_time", TimestampToMilliseconds(fmax(0.0, cr->CalculateExecutionTime()))); attrs->Set("latency", TimestampToMilliseconds(cr->CalculateLatency())); attrs->Set("check_source", cr->GetCheckSource()); attrs->Set("scheduling_source", cr->GetSchedulingSource()); } attrs->Set("is_problem", checkable->GetProblem()); attrs->Set("is_handled", checkable->GetHandled()); attrs->Set("is_reachable", checkable->IsReachable()); attrs->Set("is_flapping", checkable->IsFlapping()); attrs->Set("is_acknowledged", checkable->GetAcknowledgement()); if (checkable->IsAcknowledged()) { Timestamp entry = 0; Comment::Ptr AckComment; for (const Comment::Ptr& c : checkable->GetComments()) { if (c->GetEntryType() == CommentAcknowledgement) { if (c->GetEntryTime() > entry) { entry = c->GetEntryTime(); AckComment = c; } } } if (AckComment != nullptr) { attrs->Set("acknowledgement_comment_id", GetObjectIdentifier(AckComment)); } } { auto lastComment (checkable->GetLastComment()); if (lastComment) { attrs->Set("last_comment_id", GetObjectIdentifier(lastComment)); } } attrs->Set("in_downtime", checkable->IsInDowntime()); if (checkable->GetCheckTimeout().IsEmpty()) attrs->Set("check_timeout", TimestampToMilliseconds(checkable->GetCheckCommand()->GetTimeout())); else attrs->Set("check_timeout", TimestampToMilliseconds(checkable->GetCheckTimeout())); long long lastCheck = TimestampToMilliseconds(checkable->GetLastCheck()); if (lastCheck > 0) attrs->Set("last_update", lastCheck); attrs->Set("last_state_change", TimestampToMilliseconds(checkable->GetLastStateChange())); attrs->Set("next_check", TimestampToMilliseconds(checkable->GetNextCheck())); attrs->Set("next_update", TimestampToMilliseconds(checkable->GetNextUpdate())); return attrs; } std::vector IcingaDB::UpdateObjectAttrs(const ConfigObject::Ptr& object, int fieldType, const String& typeNameOverride) { Type::Ptr type = object->GetReflectionType(); Dictionary::Ptr attrs(new Dictionary); for (int fid = 0; fid < type->GetFieldCount(); fid++) { Field field = type->GetFieldInfo(fid); if ((field.Attributes & fieldType) == 0) continue; Value val = object->GetField(fid); /* hide attributes which shouldn't be user-visible */ if (field.Attributes & FANoUserView) continue; /* hide internal navigation fields */ if (field.Attributes & FANavigation && !(field.Attributes & (FAConfig | FAState))) continue; attrs->Set(field.Name, Serialize(val)); } /* Downtimes require in_effect, which is not an attribute */ Downtime::Ptr downtime = dynamic_pointer_cast(object); if (downtime) { attrs->Set("in_effect", Serialize(downtime->IsInEffect())); attrs->Set("trigger_time", Serialize(TimestampToMilliseconds(downtime->GetTriggerTime()))); } /* Use the name checksum as unique key. */ String typeName = type->GetName().ToLower(); if (!typeNameOverride.IsEmpty()) typeName = typeNameOverride.ToLower(); return {GetObjectIdentifier(object), JsonEncode(attrs)}; //m_Rcon->FireAndForgetQuery({"HSET", keyPrefix + typeName, GetObjectIdentifier(object), JsonEncode(attrs)}); } void IcingaDB::StateChangeHandler(const ConfigObject::Ptr& object, const CheckResult::Ptr& cr, StateType type) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendStateChange(object, cr, type); } } void IcingaDB::ReachabilityChangeHandler(const std::set& children) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { for (auto& checkable : children) { rw->UpdateState(checkable, StateUpdate::Full); } } } void IcingaDB::VersionChangedHandler(const ConfigObject::Ptr& object) { Type::Ptr type = object->GetReflectionType(); if (m_IndexedTypes.find(type.get()) == m_IndexedTypes.end()) { return; } if (object->IsActive()) { // Create or update the object config for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { if (rw) rw->SendConfigUpdate(object, true); } } else if (!object->IsActive() && object->GetExtension("ConfigObjectDeleted")) { // same as in apilistener-configsync.cpp // Delete object config for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { if (rw) rw->SendConfigDelete(object); } } } void IcingaDB::DowntimeStartedHandler(const Downtime::Ptr& downtime) { for (auto& rw : ConfigType::GetObjectsByType()) { rw->SendStartedDowntime(downtime); } } void IcingaDB::DowntimeRemovedHandler(const Downtime::Ptr& downtime) { for (auto& rw : ConfigType::GetObjectsByType()) { rw->SendRemovedDowntime(downtime); } } void IcingaDB::NotificationSentToAllUsersHandler( const Notification::Ptr& notification, const Checkable::Ptr& checkable, const std::set& users, NotificationType type, const CheckResult::Ptr& cr, const String& author, const String& text ) { auto rws (ConfigType::GetObjectsByType()); auto sendTime (notification->GetLastNotification()); if (!rws.empty()) { for (auto& rw : rws) { rw->SendSentNotification(notification, checkable, users, type, cr, author, text, sendTime); } } } void IcingaDB::CommentAddedHandler(const Comment::Ptr& comment) { for (auto& rw : ConfigType::GetObjectsByType()) { rw->SendAddedComment(comment); } } void IcingaDB::CommentRemovedHandler(const Comment::Ptr& comment) { for (auto& rw : ConfigType::GetObjectsByType()) { rw->SendRemovedComment(comment); } } void IcingaDB::FlappingChangeHandler(const Checkable::Ptr& checkable, double changeTime) { auto flappingLastChange (checkable->GetFlappingLastChange()); for (auto& rw : ConfigType::GetObjectsByType()) { rw->SendFlappingChange(checkable, changeTime, flappingLastChange); } } void IcingaDB::NewCheckResultHandler(const Checkable::Ptr& checkable) { for (auto& rw : ConfigType::GetObjectsByType()) { rw->UpdateState(checkable, StateUpdate::Volatile); rw->SendNextUpdate(checkable); } } void IcingaDB::NextCheckUpdatedHandler(const Checkable::Ptr& checkable) { for (auto& rw : ConfigType::GetObjectsByType()) { rw->UpdateState(checkable, StateUpdate::Volatile); rw->SendNextUpdate(checkable); } } void IcingaDB::HostProblemChangedHandler(const Service::Ptr& service) { for (auto& rw : ConfigType::GetObjectsByType()) { /* Host state changes affect is_handled and severity of services. */ rw->UpdateState(service, StateUpdate::Full); } } void IcingaDB::AcknowledgementSetHandler(const Checkable::Ptr& checkable, const String& author, const String& comment, AcknowledgementType type, bool persistent, double changeTime, double expiry) { auto rws (ConfigType::GetObjectsByType()); if (!rws.empty()) { for (auto& rw : rws) { rw->SendAcknowledgementSet(checkable, author, comment, type, persistent, changeTime, expiry); } } } void IcingaDB::AcknowledgementClearedHandler(const Checkable::Ptr& checkable, const String& removedBy, double changeTime) { auto rws (ConfigType::GetObjectsByType()); if (!rws.empty()) { auto rb (Shared::Make(removedBy)); auto ackLastChange (checkable->GetAcknowledgementLastChange()); for (auto& rw : rws) { rw->SendAcknowledgementCleared(checkable, *rb, changeTime, ackLastChange); } } } void IcingaDB::NotificationUsersChangedHandler(const Notification::Ptr& notification, const Array::Ptr& oldValues, const Array::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendNotificationUsersChanged(notification, oldValues, newValues); } } void IcingaDB::NotificationUserGroupsChangedHandler(const Notification::Ptr& notification, const Array::Ptr& oldValues, const Array::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendNotificationUserGroupsChanged(notification, oldValues, newValues); } } void IcingaDB::TimePeriodRangesChangedHandler(const TimePeriod::Ptr& timeperiod, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendTimePeriodRangesChanged(timeperiod, oldValues, newValues); } } void IcingaDB::TimePeriodIncludesChangedHandler(const TimePeriod::Ptr& timeperiod, const Array::Ptr& oldValues, const Array::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendTimePeriodIncludesChanged(timeperiod, oldValues, newValues); } } void IcingaDB::TimePeriodExcludesChangedHandler(const TimePeriod::Ptr& timeperiod, const Array::Ptr& oldValues, const Array::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendTimePeriodExcludesChanged(timeperiod, oldValues, newValues); } } void IcingaDB::UserGroupsChangedHandler(const User::Ptr& user, const Array::Ptr& oldValues, const Array::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendGroupsChanged(user, oldValues, newValues); } } void IcingaDB::HostGroupsChangedHandler(const Host::Ptr& host, const Array::Ptr& oldValues, const Array::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendGroupsChanged(host, oldValues, newValues); } } void IcingaDB::ServiceGroupsChangedHandler(const Service::Ptr& service, const Array::Ptr& oldValues, const Array::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendGroupsChanged(service, oldValues, newValues); } } void IcingaDB::CommandEnvChangedHandler(const ConfigObject::Ptr& command, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendCommandEnvChanged(command, oldValues, newValues); } } void IcingaDB::CommandArgumentsChangedHandler(const ConfigObject::Ptr& command, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendCommandArgumentsChanged(command, oldValues, newValues); } } void IcingaDB::CustomVarsChangedHandler(const ConfigObject::Ptr& object, const Dictionary::Ptr& oldValues, const Dictionary::Ptr& newValues) { for (const IcingaDB::Ptr& rw : ConfigType::GetObjectsByType()) { rw->SendCustomVarsChanged(object, oldValues, newValues); } } void IcingaDB::DeleteRelationship(const String& id, const String& redisKeyWithoutPrefix, bool hasChecksum) { Log(LogNotice, "IcingaDB") << "Deleting relationship '" << redisKeyWithoutPrefix << " -> '" << id << "'"; String redisKey = m_PrefixConfigObject + redisKeyWithoutPrefix; std::vector> queries; if (hasChecksum) { queries.push_back({"HDEL", m_PrefixConfigCheckSum + redisKeyWithoutPrefix, id}); } queries.push_back({"HDEL", redisKey, id}); queries.push_back({ "XADD", "icinga:runtime", "MAXLEN", "~", "1000000", "*", "redis_key", redisKey, "id", id, "runtime_type", "delete" }); m_Rcon->FireAndForgetQueries(queries, Prio::Config); }