SignalQueue.hpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:3k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #ifndef __SIGNALQUEUE_HPP_INCLUDED__
  14. #define __SIGNALQUEUE_HPP_INCLUDED__
  15. #include <NdbApiSignal.hpp>
  16. #include <NdbMutex.h>
  17. #include <NdbCondition.h>
  18. #include <Vector.hpp>
  19. /* XXX Look for an already existing definition */
  20. #define DEFAULT_TIMEOUT 10000
  21. /**
  22.  * @class SignalQueue
  23.  * @brief
  24.  */
  25. class SignalQueue {
  26. public:
  27.   typedef void (* SignalHandler)(void *obj, int gsn, NdbApiSignal *signal);
  28.   SignalQueue();
  29.   ~SignalQueue();
  30.   /**
  31.    * Static wrapper making it possible to call receive without knowing the
  32.    * type of the receiver
  33.    */
  34.   static void receive(void *me, NdbApiSignal *signal);
  35.   /**
  36.    * Enqueues a signal, and notifies any thread waiting for signals.
  37.    */
  38.   void receive(NdbApiSignal *signal);
  39.   NdbApiSignal *waitFor(int gsn,
  40. NodeId nodeid = 0,
  41. Uint32 timeout = DEFAULT_TIMEOUT);
  42.   template<class T> bool waitFor(Vector<T> &t,
  43.  T *&handler,
  44.  NdbApiSignal *&signal,
  45.  Uint32 timeout);
  46.   /**
  47.    * size()
  48.    */
  49.   
  50.   Uint32 size() {return m_queueSize;};
  51. private:
  52.   NdbMutex *m_mutex; /* Locks all data in SignalQueue */
  53.   NdbCondition *m_cond; /* Notifies about new signal in the queue */
  54.   /**
  55.    * Returns the last recently received signal. 
  56.    * Must be called with m_mutex locked.
  57.    *
  58.    * The caller takes responsibility for deleting the returned object.
  59.    *
  60.    * @returns NULL if failed, or a received signal
  61.    */
  62.   NdbApiSignal *pop();
  63.   class QueueEntry {
  64.   public:
  65.     NdbApiSignal *signal;
  66.     QueueEntry *next;
  67.   };
  68.   QueueEntry *m_signalQueueHead; /** Head of the queue.
  69.   *  New entries added on the tail
  70.   */
  71.   Uint32 m_queueSize;
  72. };
  73. template<class T> bool
  74. SignalQueue::waitFor(Vector<T> &t,
  75.      T *&handler,
  76.      NdbApiSignal *&signal,
  77.      Uint32 timeout) {
  78.   Guard g(m_mutex);
  79.   if(m_signalQueueHead == NULL)
  80.     NdbCondition_WaitTimeout(m_cond, m_mutex, timeout);
  81.   if(m_signalQueueHead == NULL)
  82.     return false;
  83.   for(size_t i = 0; i < t.size(); i++) {
  84.     if(t[i].check(m_signalQueueHead->signal)) {
  85.       handler = &t[i];
  86.       signal = pop();
  87.       return true;
  88.     }
  89.   }
  90.   ndbout_c("SignalQueue: Queued signal without true check function (GSN: %d)",
  91.    m_signalQueueHead->signal->theVerId_signalNumber);
  92.   abort();
  93.   return false;
  94. }
  95. #endif /* !__SIGNALQUEUE_HPP_INCLUDED__ */