mirror of https://github.com/Icinga/icinga2.git
439 lines
11 KiB
C++
439 lines
11 KiB
C++
/******************************************************************************
|
|
* Icinga 2 *
|
|
* Copyright (C) 2012-2014 Icinga Development Team (http://www.icinga.org) *
|
|
* *
|
|
* This program is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU General Public License *
|
|
* as published by the Free Software Foundation; either version 2 *
|
|
* of the License, or (at your option) any later version. *
|
|
* *
|
|
* This program is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU General Public License *
|
|
* along with this program; if not, write to the Free Software Foundation *
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. *
|
|
******************************************************************************/
|
|
|
|
#include "config/configitem.hpp"
|
|
#include "config/configcompilercontext.hpp"
|
|
#include "config/applyrule.hpp"
|
|
#include "config/objectrule.hpp"
|
|
#include "config/configtype.hpp"
|
|
#include "base/application.hpp"
|
|
#include "base/dynamictype.hpp"
|
|
#include "base/objectlock.hpp"
|
|
#include "base/convert.hpp"
|
|
#include "base/logger.hpp"
|
|
#include "base/debug.hpp"
|
|
#include "base/workqueue.hpp"
|
|
#include "base/exception.hpp"
|
|
#include "base/stdiostream.hpp"
|
|
#include "base/netstring.hpp"
|
|
#include "base/serializer.hpp"
|
|
#include "base/json.hpp"
|
|
#include "base/exception.hpp"
|
|
#include "base/scriptfunction.hpp"
|
|
#include <sstream>
|
|
#include <fstream>
|
|
#include <boost/foreach.hpp>
|
|
|
|
using namespace icinga;
|
|
|
|
boost::mutex ConfigItem::m_Mutex;
|
|
ConfigItem::TypeMap ConfigItem::m_Items;
|
|
ConfigItem::ItemList ConfigItem::m_UnnamedItems;
|
|
ConfigItem::ItemList ConfigItem::m_CommittedItems;
|
|
|
|
REGISTER_SCRIPTFUNCTION(__commit, &ConfigItem::ScriptCommit);
|
|
|
|
/**
|
|
* Constructor for the ConfigItem class.
|
|
*
|
|
* @param type The object type.
|
|
* @param name The name of the item.
|
|
* @param unit The unit of the item.
|
|
* @param abstract Whether the item is a template.
|
|
* @param exprl Expression list for the item.
|
|
* @param debuginfo Debug information.
|
|
*/
|
|
ConfigItem::ConfigItem(const String& type, const String& name,
|
|
bool abstract, const boost::shared_ptr<Expression>& exprl,
|
|
const boost::shared_ptr<Expression>& filter,
|
|
const DebugInfo& debuginfo, const Dictionary::Ptr& scope,
|
|
const String& zone)
|
|
: m_Type(type), m_Name(name), m_Abstract(abstract),
|
|
m_Expression(exprl), m_Filter(filter),
|
|
m_DebugInfo(debuginfo), m_Scope(scope), m_Zone(zone)
|
|
{
|
|
}
|
|
|
|
/**
|
|
* Retrieves the type of the configuration item.
|
|
*
|
|
* @returns The type.
|
|
*/
|
|
String ConfigItem::GetType(void) const
|
|
{
|
|
return m_Type;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the name of the configuration item.
|
|
*
|
|
* @returns The name.
|
|
*/
|
|
String ConfigItem::GetName(void) const
|
|
{
|
|
return m_Name;
|
|
}
|
|
|
|
/**
|
|
* Checks whether the item is abstract.
|
|
*
|
|
* @returns true if the item is abstract, false otherwise.
|
|
*/
|
|
bool ConfigItem::IsAbstract(void) const
|
|
{
|
|
return m_Abstract;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the debug information for the configuration item.
|
|
*
|
|
* @returns The debug information.
|
|
*/
|
|
DebugInfo ConfigItem::GetDebugInfo(void) const
|
|
{
|
|
return m_DebugInfo;
|
|
}
|
|
|
|
Dictionary::Ptr ConfigItem::GetScope(void) const
|
|
{
|
|
return m_Scope;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the expression list for the configuration item.
|
|
*
|
|
* @returns The expression list.
|
|
*/
|
|
boost::shared_ptr<Expression> ConfigItem::GetExpression(void) const
|
|
{
|
|
return m_Expression;
|
|
}
|
|
|
|
/**
|
|
* Retrieves the object filter for the configuration item.
|
|
*
|
|
* @returns The filter expression.
|
|
*/
|
|
boost::shared_ptr<Expression> ConfigItem::GetFilter(void) const
|
|
{
|
|
return m_Filter;
|
|
}
|
|
|
|
/**
|
|
* Commits the configuration item by creating a DynamicObject
|
|
* object.
|
|
*
|
|
* @returns The DynamicObject that was created/updated.
|
|
*/
|
|
DynamicObject::Ptr ConfigItem::Commit(bool discard)
|
|
{
|
|
ASSERT(!OwnsLock());
|
|
|
|
#ifdef I2_DEBUG
|
|
Log(LogDebug, "ConfigItem")
|
|
<< "Commit called for ConfigItem Type=" << GetType() << ", Name=" << GetName();
|
|
#endif /* I2_DEBUG */
|
|
|
|
/* Make sure the type is valid. */
|
|
Type::Ptr type = Type::GetByName(GetType());
|
|
ASSERT(type && Type::GetByName("DynamicObject")->IsAssignableFrom(type));
|
|
|
|
if (IsAbstract())
|
|
return DynamicObject::Ptr();
|
|
|
|
DynamicObject::Ptr dobj = static_pointer_cast<DynamicObject>(type->Instantiate());
|
|
|
|
dobj->SetDebugInfo(m_DebugInfo);
|
|
dobj->SetTypeName(m_Type);
|
|
dobj->SetZone(m_Zone);
|
|
dobj->SetName(m_Name);
|
|
|
|
DebugHint debugHints;
|
|
|
|
ScriptFrame frame(dobj);
|
|
if (m_Scope)
|
|
m_Scope->CopyTo(frame.Locals);
|
|
m_Expression->Evaluate(frame, &debugHints);
|
|
|
|
if (discard)
|
|
m_Expression.reset();
|
|
|
|
String item_name;
|
|
String short_name = dobj->GetShortName();
|
|
|
|
if (!short_name.IsEmpty()) {
|
|
item_name = short_name;
|
|
dobj->SetName(short_name);
|
|
} else
|
|
item_name = m_Name;
|
|
|
|
String name = item_name;
|
|
|
|
NameComposer *nc = dynamic_cast<NameComposer *>(type.get());
|
|
|
|
if (nc) {
|
|
name = nc->MakeName(name, dobj);
|
|
|
|
if (name.IsEmpty())
|
|
BOOST_THROW_EXCEPTION(std::runtime_error("Could not determine name for object"));
|
|
}
|
|
|
|
if (name != item_name)
|
|
dobj->SetShortName(item_name);
|
|
|
|
dobj->SetName(name);
|
|
dobj->OnConfigLoaded();
|
|
|
|
{
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
m_CommittedItems.push_back(this);
|
|
}
|
|
|
|
Dictionary::Ptr attrs = Serialize(dobj, FAConfig);
|
|
|
|
Dictionary::Ptr persistentItem = new Dictionary();
|
|
|
|
persistentItem->Set("type", GetType());
|
|
persistentItem->Set("name", GetName());
|
|
persistentItem->Set("properties", attrs);
|
|
persistentItem->Set("debug_hints", debugHints.ToDictionary());
|
|
|
|
ConfigCompilerContext::GetInstance()->WriteObject(persistentItem);
|
|
persistentItem.reset();
|
|
|
|
ConfigType::Ptr ctype = ConfigType::GetByName(GetType());
|
|
|
|
if (ctype) {
|
|
TypeRuleUtilities utils;
|
|
ctype->ValidateItem(GetName(), dobj, GetDebugInfo(), &utils);
|
|
}
|
|
|
|
dobj->Register();
|
|
|
|
m_Object = dobj;
|
|
|
|
return dobj;
|
|
}
|
|
|
|
/**
|
|
* Registers the configuration item.
|
|
*/
|
|
void ConfigItem::Register(void)
|
|
{
|
|
Type::Ptr type = Type::GetByName(m_Type);
|
|
|
|
/* If this is a non-abstract object with a composite name
|
|
* we register it in m_UnnamedItems instead of m_Items. */
|
|
if (!m_Abstract && dynamic_cast<NameComposer *>(type.get())) {
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
m_UnnamedItems.push_back(this);
|
|
} else {
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
m_Items[m_Type][m_Name] = this;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Retrieves a configuration item by type and name.
|
|
*
|
|
* @param type The type of the ConfigItem that is to be looked up.
|
|
* @param name The name of the ConfigItem that is to be looked up.
|
|
* @returns The configuration item.
|
|
*/
|
|
ConfigItem::Ptr ConfigItem::GetObject(const String& type, const String& name)
|
|
{
|
|
{
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
|
|
ConfigItem::TypeMap::const_iterator it = m_Items.find(type);
|
|
|
|
if (it == m_Items.end())
|
|
return ConfigItem::Ptr();
|
|
|
|
ConfigItem::ItemMap::const_iterator it2 = it->second.find(name);
|
|
|
|
if (it2 == it->second.end())
|
|
return ConfigItem::Ptr();
|
|
|
|
return it2->second;
|
|
}
|
|
}
|
|
|
|
bool ConfigItem::CommitNewItems(WorkQueue& upq)
|
|
{
|
|
typedef std::pair<ConfigItem::Ptr, bool> ItemPair;
|
|
std::vector<ItemPair> items;
|
|
|
|
do {
|
|
items.clear();
|
|
|
|
{
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
|
|
BOOST_FOREACH(const TypeMap::value_type& kv, m_Items) {
|
|
BOOST_FOREACH(const ItemMap::value_type& kv2, kv.second)
|
|
{
|
|
if (!kv2.second->m_Abstract && !kv2.second->m_Object)
|
|
items.push_back(std::make_pair(kv2.second, false));
|
|
}
|
|
}
|
|
|
|
BOOST_FOREACH(const ConfigItem::Ptr& item, m_UnnamedItems) {
|
|
if (!item->m_Abstract && !item->m_Object)
|
|
items.push_back(std::make_pair(item, true));
|
|
}
|
|
|
|
m_UnnamedItems.clear();
|
|
}
|
|
|
|
BOOST_FOREACH(const ItemPair& ip, items) {
|
|
upq.Enqueue(boost::bind(&ConfigItem::Commit, ip.first, ip.second));
|
|
}
|
|
|
|
upq.Join();
|
|
|
|
if (upq.HasExceptions())
|
|
return false;
|
|
|
|
std::vector<ConfigItem::Ptr> new_items;
|
|
|
|
{
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
new_items.swap(m_CommittedItems);
|
|
}
|
|
|
|
BOOST_FOREACH(const ConfigItem::Ptr& item, new_items) {
|
|
upq.Enqueue(boost::bind(&DynamicObject::OnAllConfigLoaded, item->m_Object));
|
|
}
|
|
|
|
upq.Join();
|
|
} while (!items.empty());
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ConfigItem::CommitItems(void)
|
|
{
|
|
WorkQueue upq(25000, Application::GetConcurrency());
|
|
|
|
Log(LogInformation, "ConfigItem", "Committing config items");
|
|
|
|
if (!CommitNewItems(upq)) {
|
|
upq.ReportExceptions("config");
|
|
return false;
|
|
}
|
|
|
|
ApplyRule::CheckMatches();
|
|
ConfigType::DiscardTypes();
|
|
|
|
/* log stats for external parsers */
|
|
BOOST_FOREACH(const DynamicType::Ptr& type, DynamicType::GetTypes()) {
|
|
int count = std::distance(type->GetObjects().first, type->GetObjects().second);
|
|
if (count > 0)
|
|
Log(LogInformation, "ConfigItem")
|
|
<< "Checked " << count << " " << type->GetName() << "(s).";
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ConfigItem::ActivateItems(void)
|
|
{
|
|
/* restore the previous program state */
|
|
try {
|
|
DynamicObject::RestoreObjects(Application::GetStatePath());
|
|
} catch (const std::exception& ex) {
|
|
Log(LogCritical, "ConfigItem")
|
|
<< "Failed to restore state file: " << DiagnosticInformation(ex);
|
|
}
|
|
|
|
Log(LogInformation, "ConfigItem", "Triggering Start signal for config items");
|
|
|
|
WorkQueue upq(25000, Application::GetConcurrency());
|
|
|
|
BOOST_FOREACH(const DynamicType::Ptr& type, DynamicType::GetTypes()) {
|
|
BOOST_FOREACH(const DynamicObject::Ptr& object, type->GetObjects()) {
|
|
if (object->IsActive())
|
|
continue;
|
|
|
|
#ifdef I2_DEBUG
|
|
Log(LogDebug, "ConfigItem")
|
|
<< "Activating object '" << object->GetName() << "' of type '" << object->GetType()->GetName() << "'";
|
|
#endif /* I2_DEBUG */
|
|
upq.Enqueue(boost::bind(&DynamicObject::Activate, object));
|
|
}
|
|
}
|
|
|
|
upq.Join();
|
|
|
|
#ifdef I2_DEBUG
|
|
BOOST_FOREACH(const DynamicType::Ptr& type, DynamicType::GetTypes()) {
|
|
BOOST_FOREACH(const DynamicObject::Ptr& object, type->GetObjects()) {
|
|
ASSERT(object->IsActive());
|
|
}
|
|
}
|
|
#endif /* I2_DEBUG */
|
|
|
|
Log(LogInformation, "ConfigItem", "Activated all objects.");
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ConfigItem::ScriptCommit(void)
|
|
{
|
|
WorkQueue upq(25000, Application::GetConcurrency());
|
|
|
|
if (!CommitNewItems(upq))
|
|
return false;
|
|
|
|
BOOST_FOREACH(const DynamicType::Ptr& type, DynamicType::GetTypes()) {
|
|
BOOST_FOREACH(const DynamicObject::Ptr& object, type->GetObjects()) {
|
|
if (object->IsActive())
|
|
continue;
|
|
|
|
#ifdef I2_DEBUG
|
|
Log(LogDebug, "ConfigItem")
|
|
<< "Activating object '" << object->GetName() << "' of type '" << object->GetType()->GetName() << "'";
|
|
#endif /* I2_DEBUG */
|
|
upq.Enqueue(boost::bind(&DynamicObject::Activate, object));
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
std::vector<ConfigItem::Ptr> ConfigItem::GetItems(const String& type)
|
|
{
|
|
std::vector<ConfigItem::Ptr> items;
|
|
|
|
boost::mutex::scoped_lock lock(m_Mutex);
|
|
|
|
TypeMap::const_iterator it = m_Items.find(type);
|
|
|
|
if (it == m_Items.end())
|
|
return items;
|
|
|
|
BOOST_FOREACH(const ItemMap::value_type& kv, it->second)
|
|
{
|
|
items.push_back(kv.second);
|
|
}
|
|
|
|
return items;
|
|
}
|