livestatus/src/ServiceSpecialIntColumn.cc

   1 // +------------------------------------------------------------------+
   2 // |             ____ _               _        __  __ _  __           |
   3 // |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
   4 // |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
   5 // |           | |___| | | |  __/ (__|   <    | |  | | . \            |
   6 // |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
   7 // |                                                                  |
   8 // | Copyright Mathias Kettner 2014             mk@mathias-kettner.de |
   9 // +------------------------------------------------------------------+
  10 //
  11 // This file is part of Check_MK.
  12 // The official homepage is at http://mathias-kettner.de/check_mk.
  13 //
  14 // check_mk is free software;  you can redistribute it and/or modify it
  15 // under the  terms of the  GNU General Public License  as published by
  16 // the Free Software Foundation in version 2.  check_mk is  distributed
  17 // in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
  18 // out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
  19 // PARTICULAR PURPOSE. See the  GNU General Public License for more de-
  20 // tails. You should have  received  a copy of the  GNU  General Public
  21 // License along with GNU Make; see the file  COPYING.  If  not,  write
  22 // to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
  23 // Boston, MA 02110-1301 USA.
  24
  25 #include "ServiceSpecialIntColumn.h"
  26 #include "Row.h"
  27
  28 #ifdef CMC
  29 #include "Core.h"
  30 #include "MonitoringCore.h"
  31 #include "Object.h"
  32 #include "RRDBackend.h"
  33 #include "RRDInfoCache.h"
  34 #include "State.h"
  35 #include "cmc.h"
  36 #else
  37 #include "nagios.h"
  38 #include "pnp4nagios.h"
  39 #endif
  40
  41 int32_t ServiceSpecialIntColumn::getValue(
  42     Row row, const contact* /* auth_user */) const {
  43 #ifdef CMC
  44     if (auto object = columnData<Object>(row)) {
  45         switch (_type) {
  46             case Type::real_hard_state: {
  47                 if (object->isCurrentStateOK()) {
  48                     return 0;
  49                 }
  50                 auto state = object->state();
  51                 return state->_state_type == StateType::hard
  52                            ? state->_current_state
  53                            : state->_last_hard_state;
  54             }
  55             case Type::pnp_graph_present:
  56                 return _mc->impl<Core>()
  57                                ->_rrd_backend.infoFor(object)
  58                                ._names.empty()
  59                            ? 0
  60                            : 1;
  61         }
  62     }
  63 #else
  64     if (auto svc = columnData<service>(row)) {
  65         switch (_type) {
  66             case Type::real_hard_state:
  67                 if (svc->current_state == 0) {
  68                     return 0;
  69                 }
  70                 if (svc->state_type == HARD_STATE) {
  71                     return svc->current_state;
  72                 }
  73                 return svc->last_hard_state;
  74
  75             case Type::pnp_graph_present:
  76                 return pnpgraph_present(_mc, svc->host_ptr->name,
  77                                         svc->description);
  78         }
  79     }
  80 #endif
  81     return 0;
  82 }