icinga2/components/livestatus/hostgroupstable.cpp

374 lines
12 KiB
C++

/******************************************************************************
* Icinga 2 *
* Copyright (C) 2012 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 "livestatus/hostgroupstable.h"
#include "icinga/hostgroup.h"
#include "icinga/host.h"
#include "icinga/service.h"
#include "base/dynamictype.h"
#include <boost/foreach.hpp>
using namespace icinga;
using namespace livestatus;
HostGroupsTable::HostGroupsTable(void)
{
AddColumns(this);
}
void HostGroupsTable::AddColumns(Table *table, const String& prefix,
const Column::ObjectAccessor& objectAccessor)
{
table->AddColumn(prefix + "name", Column(&HostGroupsTable::NameAccessor, objectAccessor));
table->AddColumn(prefix + "alias", Column(&HostGroupsTable::AliasAccessor, objectAccessor));
table->AddColumn(prefix + "notes", Column(&HostGroupsTable::NotesAccessor, objectAccessor));
table->AddColumn(prefix + "notes_url", Column(&HostGroupsTable::NotesUrlAccessor, objectAccessor));
table->AddColumn(prefix + "action_url", Column(&HostGroupsTable::ActionUrlAccessor, objectAccessor));
table->AddColumn(prefix + "members", Column(&HostGroupsTable::MembersAccessor, objectAccessor));
table->AddColumn(prefix + "members_with_state", Column(&HostGroupsTable::MembersWithStateAccessor, objectAccessor));
table->AddColumn(prefix + "worst_host_state", Column(&HostGroupsTable::WorstHostStateAccessor, objectAccessor));
table->AddColumn(prefix + "num_hosts", Column(&HostGroupsTable::NumHostsAccessor, objectAccessor));
table->AddColumn(prefix + "num_hosts_pending", Column(&HostGroupsTable::NumHostsPendingAccessor, objectAccessor));
table->AddColumn(prefix + "num_hosts_up", Column(&HostGroupsTable::NumHostsUpAccessor, objectAccessor));
table->AddColumn(prefix + "num_hosts_down", Column(&HostGroupsTable::NumHostsDownAccessor, objectAccessor));
table->AddColumn(prefix + "num_hosts_unreach", Column(&HostGroupsTable::NumHostsUnreachAccessor, objectAccessor));
table->AddColumn(prefix + "num_services", Column(&HostGroupsTable::NumServicesAccessor, objectAccessor));
table->AddColumn(prefix + "worst_services_state", Column(&HostGroupsTable::WorstServicesStateAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_pending", Column(&HostGroupsTable::NumServicesPendingAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_ok", Column(&HostGroupsTable::NumServicesOkAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_warn", Column(&HostGroupsTable::NumServicesWarnAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_crit", Column(&HostGroupsTable::NumServicesCritAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_unknown", Column(&HostGroupsTable::NumServicesUnknownAccessor, objectAccessor));
table->AddColumn(prefix + "worst_service_hard_state", Column(&HostGroupsTable::WorstServiceHardStateAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_hard_ok", Column(&HostGroupsTable::NumServicesHardOkAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_hard_warn", Column(&HostGroupsTable::NumServicesHardWarnAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_hard_crit", Column(&HostGroupsTable::NumServicesHardCritAccessor, objectAccessor));
table->AddColumn(prefix + "num_services_hard_unknown", Column(&HostGroupsTable::NumServicesHardUnknownAccessor, objectAccessor));
}
String HostGroupsTable::GetName(void) const
{
return "hostgroups";
}
void HostGroupsTable::FetchRows(const AddRowFunction& addRowFn)
{
BOOST_FOREACH(const DynamicObject::Ptr& object, DynamicType::GetObjects("HostGroup")) {
addRowFn(object);
}
}
Value HostGroupsTable::NameAccessor(const Value& row)
{
return static_cast<HostGroup::Ptr>(row)->GetName();
}
Value HostGroupsTable::AliasAccessor(const Value& row)
{
return static_cast<HostGroup::Ptr>(row)->GetDisplayName();
}
Value HostGroupsTable::NotesAccessor(const Value& row)
{
Dictionary::Ptr custom = static_cast<HostGroup::Ptr>(row)->GetCustom();
if (!custom)
return Empty;
return custom->Get("notes");
}
Value HostGroupsTable::NotesUrlAccessor(const Value& row)
{
Dictionary::Ptr custom = static_cast<HostGroup::Ptr>(row)->GetCustom();
if (!custom)
return Empty;
return custom->Get("notes_url");
}
Value HostGroupsTable::ActionUrlAccessor(const Value& row)
{
Dictionary::Ptr custom = static_cast<HostGroup::Ptr>(row)->GetCustom();
if (!custom)
return Empty;
return custom->Get("action_url");
}
Value HostGroupsTable::MembersAccessor(const Value& row)
{
Array::Ptr members = boost::make_shared<Array>();
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
members->Add(host->GetName());
}
return members;
}
Value HostGroupsTable::MembersWithStateAccessor(const Value& row)
{
/* TODO */
return Empty;
}
Value HostGroupsTable::WorstHostStateAccessor(const Value& row)
{
int worst_host = HostUp;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
if (host->GetState() > worst_host)
worst_host = host->GetState();
}
return worst_host;
}
Value HostGroupsTable::NumHostsAccessor(const Value& row)
{
return static_cast<HostGroup::Ptr>(row)->GetMembers().size();
}
Value HostGroupsTable::NumHostsPendingAccessor(const Value& row)
{
int num_hosts = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
Service::Ptr hc = host->GetHostCheckService();
/* no hostcheck service or no checkresult */
if (!hc || !hc->GetLastCheckResult())
num_hosts++;
}
return num_hosts;
}
Value HostGroupsTable::NumHostsUpAccessor(const Value& row)
{
int num_hosts = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
if (host->GetState() == HostUp)
num_hosts++;
}
return num_hosts;
}
Value HostGroupsTable::NumHostsDownAccessor(const Value& row)
{
int num_hosts = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
if (host->GetState() == HostDown)
num_hosts++;
}
return num_hosts;
}
Value HostGroupsTable::NumHostsUnreachAccessor(const Value& row)
{
int num_hosts = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
if (host->GetState() == HostUnreachable)
num_hosts++;
}
return num_hosts;
}
Value HostGroupsTable::NumServicesAccessor(const Value& row)
{
int num_services = 0;
HostGroup::Ptr hg = static_cast<HostGroup::Ptr>(row);
if (hg->GetMembers().size() == 0)
return 0;
BOOST_FOREACH(const Host::Ptr& host, hg->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::WorstServicesStateAccessor(const Value& row)
{
Value worst_service = StateOK;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetState() > worst_service)
worst_service = service->GetState();
}
}
return worst_service;
}
Value HostGroupsTable::NumServicesPendingAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (!service->GetLastCheckResult())
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::NumServicesOkAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetState() == StateOK)
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::NumServicesWarnAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetState() == StateWarning)
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::NumServicesCritAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetState() == StateCritical)
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::NumServicesUnknownAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetState() == StateUnknown)
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::WorstServiceHardStateAccessor(const Value& row)
{
Value worst_service = StateOK;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetStateType() == StateTypeHard) {
if (service->GetState() > worst_service)
worst_service = service->GetState();
}
}
}
return worst_service;
}
Value HostGroupsTable::NumServicesHardOkAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetStateType() == StateTypeHard && service->GetState() == StateOK)
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::NumServicesHardWarnAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetStateType() == StateTypeHard && service->GetState() == StateWarning)
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::NumServicesHardCritAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetStateType() == StateTypeHard && service->GetState() == StateCritical)
num_services++;
}
}
return num_services;
}
Value HostGroupsTable::NumServicesHardUnknownAccessor(const Value& row)
{
int num_services = 0;
BOOST_FOREACH(const Host::Ptr& host, static_cast<HostGroup::Ptr>(row)->GetMembers()) {
BOOST_FOREACH(const Service::Ptr& service, host->GetServices()) {
if (service->GetStateType() == StateTypeHard && service->GetState() == StateUnknown)
num_services++;
}
}
return num_services;
}