icinga2/components/livestatus/table.cpp

149 lines
4.7 KiB
C++
Raw Normal View History

2013-03-10 03:09:01 +01:00
/******************************************************************************
* Icinga 2 *
2013-09-25 07:43:57 +02:00
* Copyright (C) 2012-2013 Icinga Development Team (http://www.icinga.org/) *
2013-03-10 03:09:01 +01:00
* *
* 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. *
******************************************************************************/
2013-03-17 20:19:29 +01:00
#include "livestatus/table.h"
#include "livestatus/statustable.h"
#include "livestatus/contactgroupstable.h"
#include "livestatus/contactstable.h"
#include "livestatus/hostgroupstable.h"
2013-03-17 20:19:29 +01:00
#include "livestatus/hoststable.h"
#include "livestatus/servicegroupstable.h"
2013-03-17 20:19:29 +01:00
#include "livestatus/servicestable.h"
#include "livestatus/commandstable.h"
2013-03-17 20:19:29 +01:00
#include "livestatus/commentstable.h"
#include "livestatus/downtimestable.h"
#include "livestatus/timeperiodstable.h"
#include "livestatus/logtable.h"
#include "livestatus/statehisttable.h"
2013-03-17 20:19:29 +01:00
#include "livestatus/filter.h"
#include "base/array.h"
#include "base/dictionary.h"
2013-03-15 18:21:29 +01:00
#include <boost/tuple/tuple.hpp>
2013-03-16 21:18:53 +01:00
#include <boost/foreach.hpp>
2013-03-17 20:19:29 +01:00
#include <boost/bind.hpp>
2013-03-10 03:09:01 +01:00
using namespace icinga;
Table::Table(void)
2013-03-10 03:09:01 +01:00
{ }
Table::Ptr Table::GetByName(const String& name, const String& compat_log_path, const unsigned long& from, const unsigned long& until)
2013-03-10 03:09:01 +01:00
{
if (name == "status")
return make_shared<StatusTable>();
else if (name == "contactgroups")
return make_shared<ContactGroupsTable>();
else if (name == "contacts")
return make_shared<ContactsTable>();
else if (name == "hostgroups")
return make_shared<HostGroupsTable>();
2013-03-10 18:49:14 +01:00
else if (name == "hosts")
return make_shared<HostsTable>();
else if (name == "servicegroups")
return make_shared<ServiceGroupsTable>();
2013-03-10 22:20:13 +01:00
else if (name == "services")
return make_shared<ServicesTable>();
else if (name == "commands")
return make_shared<CommandsTable>();
2013-03-10 22:20:13 +01:00
else if (name == "comments")
return make_shared<CommentsTable>();
2013-03-10 22:20:13 +01:00
else if (name == "downtimes")
return make_shared<DowntimesTable>();
else if (name == "timeperiods")
return make_shared<TimePeriodsTable>();
else if (name == "log")
return make_shared<LogTable>(compat_log_path, from, until);
else if (name == "statehist")
return make_shared<StateHistTable>(compat_log_path, from, until);
2013-03-10 03:09:01 +01:00
return Table::Ptr();
}
2013-03-10 18:49:14 +01:00
void Table::AddColumn(const String& name, const Column& column)
{
2013-03-16 21:18:53 +01:00
std::pair<String, Column> item = std::make_pair(name, column);
2013-03-10 18:49:14 +01:00
2013-03-16 21:18:53 +01:00
std::pair<std::map<String, Column>::iterator, bool> ret = m_Columns.insert(item);
2013-03-10 18:49:14 +01:00
if (!ret.second)
ret.first->second = column;
}
2013-03-10 18:49:14 +01:00
Column Table::GetColumn(const String& name) const
{
2013-03-16 21:18:53 +01:00
std::map<String, Column>::const_iterator it = m_Columns.find(name);
if (it == m_Columns.end())
BOOST_THROW_EXCEPTION(std::invalid_argument("Column '" + name + "' does not exist in table '" + GetName() + "'."));
2013-03-10 05:10:51 +01:00
return it->second;
}
2013-03-16 21:18:53 +01:00
std::vector<String> Table::GetColumnNames(void) const
{
2013-03-16 21:18:53 +01:00
std::vector<String> names;
String name;
2013-03-15 18:21:29 +01:00
BOOST_FOREACH(boost::tie(name, boost::tuples::ignore), m_Columns) {
names.push_back(name);
2013-03-10 05:10:51 +01:00
}
return names;
}
2013-03-10 03:09:01 +01:00
std::vector<Value> Table::FilterRows(const Filter::Ptr& filter)
{
std::vector<Value> rs;
2013-03-10 05:10:51 +01:00
2013-03-10 15:11:32 +01:00
FetchRows(boost::bind(&Table::FilteredAddRow, this, boost::ref(rs), filter, _1));
return rs;
}
2013-03-10 05:10:51 +01:00
void Table::FilteredAddRow(std::vector<Value>& rs, const Filter::Ptr& filter, const Value& row)
{
if (!filter || filter->Apply(GetSelf(), row))
rs.push_back(row);
}
Value Table::ZeroAccessor(const Value&)
{
return 0;
2013-03-10 03:09:01 +01:00
}
Value Table::OneAccessor(const Value&)
{
2013-10-03 03:48:12 +02:00
return 1;
}
Value Table::EmptyStringAccessor(const Value&)
{
return "";
}
Value Table::EmptyArrayAccessor(const Value&)
{
return make_shared<Array>();
}
Value Table::EmptyDictionaryAccessor(const Value&)
{
return make_shared<Dictionary>();
}