git.m6w6.name Git - awesomized/libmemcached/blob - util/signal.hpp

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  libtest
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *
   7  *  This library is free software; you can redistribute it and/or
   8  *  modify it under the terms of the GNU Lesser General Public
   9  *  License as published by the Free Software Foundation; either
  10  *  version 3 of the License, or (at your option) any later version.
  11  *
  12  *  This library is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  *  Lesser General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU Lesser General Public
  18  *  License along with this library; if not, write to the Free Software
  19  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  20  */
  21
  22
  23
  24 #pragma once
  25
  26 #include <pthread.h>
  27 #include <semaphore.h>
  28
  29 namespace datadifferential {
  30 namespace util {
  31
  32 enum shutdown_t {
  33   SHUTDOWN_RUNNING,
  34   SHUTDOWN_GRACEFUL,
  35   SHUTDOWN_FORCED
  36 };
  37
  38 class SignalThread {
  39   bool _exit_on_signal;
  40   sigset_t set;
  41   sem_t lock;
  42   uint64_t magic_memory;
  43   volatile shutdown_t __shutdown;
  44   pthread_mutex_t shutdown_mutex;
  45   pthread_t thread;
  46
  47 public:
  48
  49   SignalThread(bool exit_on_signal_arg= false);
  50
  51   void test();
  52   void post();
  53   bool setup();
  54
  55   bool exit_on_signal()
  56   {
  57     return _exit_on_signal;
  58   }
  59
  60   int wait(int& sig)
  61   {
  62     return sigwait(&set, &sig);
  63   }
  64
  65   ~SignalThread();
  66
  67   void set_shutdown(shutdown_t arg);
  68   bool is_shutdown();
  69   shutdown_t get_shutdown();
  70 };
  71
  72 } /* namespace util */
  73 } /* namespace datadifferential */