livestatus/src/Triggers.h

   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 #ifndef Triggers_h
  26 #define Triggers_h
  27
  28 #include "config.h"  // IWYU pragma: keep
  29 #include <chrono>
  30 #include <condition_variable>
  31 #include <mutex>
  32 #include <string>
  33
  34 class Triggers {
  35 public:
  36     enum class Kind {
  37         all,
  38         check,
  39         state,
  40         log,
  41         downtime,
  42         comment,
  43         command,
  44         program
  45     };
  46
  47     Kind find(const std::string &name);
  48
  49     void notify_all(Kind trigger);
  50
  51     template <class Rep, class Period, class Predicate>
  52     void wait_for(Kind trigger,
  53                   const std::chrono::duration<Rep, Period> &rel_time,
  54                   Predicate pred) {
  55         std::unique_lock<std::mutex> ul(_mutex);
  56         auto &cond = condition_variable_for(trigger);
  57         if (rel_time == rel_time.zero()) {
  58             cond.wait(ul, pred);
  59         } else {
  60             cond.wait_for(ul, rel_time, pred);
  61         }
  62     }
  63
  64 private:
  65     std::mutex _mutex;
  66     std::condition_variable _cond_all;
  67     std::condition_variable _cond_check;
  68     std::condition_variable _cond_state;
  69     std::condition_variable _cond_log;
  70     std::condition_variable _cond_downtime;
  71     std::condition_variable _cond_comment;
  72     std::condition_variable _cond_command;
  73     std::condition_variable _cond_program;
  74
  75     std::condition_variable &condition_variable_for(Kind trigger);
  76 };
  77
  78 #endif  // Triggers_h