livestatus/src/CustomTimeperiodColumn.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 "CustomTimeperiodColumn.h"
  26 #include "Row.h"
  27 #include "TimeperiodsCache.h"
  28
  29 extern TimeperiodsCache *g_timeperiods_cache;
  30
  31 // Get the name of a timeperiod from a custom variable and lookup the current
  32 // state of that period
  33 int32_t CustomTimeperiodColumn::getValue(
  34     Row row, const contact * /* auth_user */) const {
  35     for (customvariablesmember *cvm = getCVM(row); cvm != nullptr;
  36          cvm = cvm->next) {
  37         if (cvm->variable_name == _varname) {
  38             return static_cast<int32_t>(
  39                 g_timeperiods_cache->inTimeperiod(cvm->variable_value));
  40         }
  41     }
  42     return 1;  // assume 24X7
  43 }
  44
  45 customvariablesmember *CustomTimeperiodColumn::getCVM(Row row) const {
  46     if (auto p = columnData<customvariablesmember *>(row)) {
  47         return *p;
  48     }
  49     return nullptr;
  50 }