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 5000
  21. class SignalQueue {
  22. public:
  23.   typedef void (* SignalHandler)(void *obj, int gsn, NdbApiSignal *signal);
  24.   SignalQueue();
  25.   ~SignalQueue();
  26.   /**
  27.    * Static wrapper making it possible to call receive without knowing the
  28.    * type of the receiver
  29.    */
  30.   static void receive(void *me, NdbApiSignal *signal);
  31.   /**
  32.    * Enqueues a signal, and notifies any thread waiting for signals.
  33.    */
  34.   void receive(NdbApiSignal *signal);
  35.   NdbApiSignal *waitFor(int gsn,
  36. NodeId nodeid = 0,
  37. Uint32 timeout = DEFAULT_TIMEOUT);
  38.   template<class T> bool waitFor(Vector<T> &t,
  39.  T **handler,
  40.  NdbApiSignal **signal,
  41.  Uint32 timeout = DEFAULT_TIMEOUT);
  42. private:
  43.   NdbMutex *m_mutex; /* Locks all data in SignalQueue */
  44.   NdbCondition *m_cond; /* Notifies about new signal in the queue */
  45.   /**
  46.    * Returns the last recently received signal. Must be called with
  47.    * m_mutex locked.
  48.    * The caller takes responsibility for deleting the returned object.
  49.    *
  50.    * @returns NULL if failed, or a received signal
  51.    */
  52.   NdbApiSignal *pop();
  53.   class QueueEntry {
  54.   public:
  55.     NdbApiSignal *signal;
  56.     QueueEntry *next;
  57.   };
  58.   QueueEntry *m_signalQueueHead; /** Head of the queue.
  59.   *  New entries added on the tail
  60.   */
  61. };
  62. template<class T> bool
  63. SignalQueue::waitFor(Vector<T> &t,
  64.      T **handler,
  65.      NdbApiSignal **signal,
  66.      Uint32 timeout) {
  67.   Guard g(m_mutex);
  68.   if(m_signalQueueHead == NULL)
  69.     NdbCondition_WaitTimeout(m_cond, m_mutex, timeout);
  70.   if(m_signalQueueHead == NULL)
  71.     return false;
  72.   for(size_t i = 0; i < t.size(); i++) {
  73.     if(t[i].check(m_signalQueueHead->signal)) {
  74.       * handler = &t[i];
  75.       * signal = pop();
  76.       return true;
  77.     }
  78.   }
  79.   return false;
  80. }
  81. #endif /* !__SIGNALQUEUE_HPP_INCLUDED__ */