hphp/runtime/base/timeout_thread.h

   1 /*
   2    +----------------------------------------------------------------------+
   3    | HipHop for PHP                                                       |
   4    +----------------------------------------------------------------------+
   5    | Copyright (c) 2010- Facebook, Inc. (http://www.facebook.com)         |
   6    +----------------------------------------------------------------------+
   7    | This source file is subject to version 3.01 of the PHP license,      |
   8    | that is bundled with this package in the file LICENSE, and is        |
   9    | available through the world-wide-web at the following url:           |
  10    | http://www.php.net/license/3_01.txt                                  |
  11    | If you did not receive a copy of the PHP license and are unable to   |
  12    | obtain it through the world-wide-web, please send a note to          |
  13    | license@php.net so we can mail you a copy immediately.               |
  14    +----------------------------------------------------------------------+
  15 */
  16
  17 #ifndef incl_HPHP_TIMEOUT_THREAD_H_
  18 #define incl_HPHP_TIMEOUT_THREAD_H_
  19
  20 #include <queue>
  21
  22 #include "hphp/runtime/base/types.h"
  23 #include "hphp/util/base.h"
  24 #include "hphp/util/process.h"
  25 #include "hphp/util/synchronizable.h"
  26 #include <event.h>
  27
  28 namespace HPHP {
  29 ///////////////////////////////////////////////////////////////////////////////
  30
  31 class TimeoutThread : public Synchronizable {
  32 public:
  33   static void DeferTimeout(int seconds);
  34
  35 public:
  36   TimeoutThread(int timeoutSeconds);
  37   ~TimeoutThread();
  38
  39   void registerRequestThread(RequestInjectionData* data);
  40   void removeRequestThread(RequestInjectionData* data);
  41   void run();
  42   void stop();
  43
  44   void onTimer(int index);
  45
  46 private:
  47   void checkForNewWorkers();
  48   void drainPipe();
  49   void notifyPipe();
  50
  51   int m_nextId;
  52   // m_pendingIds contains ids of threads that were added or removed
  53   // and need to be processed by the worker thread
  54   std::queue<int> m_pendingIds;
  55   bool m_stopped;
  56
  57   event_base *m_eventBase;
  58
  59   struct ClientThread {
  60     RequestInjectionData* data;
  61     event e;
  62   };
  63   std::map<int, ClientThread> m_clients;
  64   int m_timeoutSeconds;
  65
  66   // signal to wake up the thread
  67   event m_eventPipe;
  68   CPipe m_pipe;
  69 };
  70
  71 ///////////////////////////////////////////////////////////////////////////////
  72 }
  73
  74 #endif // incl_HPHP_TIMEOUT_THREAD_H_
  75