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

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 MemoryChannel_H
  14. #define MemoryChannel_H
  15. //===========================================================================
  16. //
  17. // .DESCRIPTION
  18. //              Pointer based communication channel for communication between two 
  19. //              thread. It does not copy any data in or out the channel so the 
  20. //              item that is put in can not be used untill the other thread has 
  21. //              given it back. There is no support for detecting the return of a 
  22. //              item. The channel is half-duplex. 
  23. //              For comminication between 1 writer and 1 reader use the MemoryChannel
  24. //              class, for comminication between multiple writer and 1 reader use the
  25. //              MemoryChannelMultipleWriter. There is no support for multiple readers.
  26. //
  27. // .TYPICAL USE:
  28. //              to communicate between threads.
  29. //
  30. // .EXAMPLE:
  31. //              See AsyncFile.C
  32. //===========================================================================
  33. //
  34. //
  35. // MemoryChannel( int size= 256);
  36. //   Constuctor
  37. // Parameters:
  38. //      size : amount of pointer it can hold
  39. //
  40. // void operator ++ ();
  41. //   increments the index with one, if size is reached it is set to zero
  42. //
  43. // virtual void write( T *t);
  44. //   Puts the item in the channel if the channel is full an error is reported.
  45. // Parameters:
  46. //      t: pointer to item to put in the channel, after this the item
  47. //                        is shared with the other thread.
  48. // errors
  49. //                      AFS_ERROR_CHANNALFULL, channel is full
  50. //
  51. // T* read();
  52. //      Reads a itemn from the channel, if channel is empty it blocks untill
  53. //              an item can be read.
  54. // return
  55. //                      T : item from the channel
  56. //
  57. // T* tryRead();
  58. //      Reads a item from the channel, if channel is empty it returns zero.
  59. // return
  60. //                      T : item from the channel or zero if channel is empty.
  61. //
  62. #if defined NDB_OSE || defined NDB_SOFTOSE
  63. #include "MemoryChannelOSE.hpp"
  64. #else
  65. #include "ErrorHandlingMacros.hpp"
  66. #include "Error.hpp"
  67. #include "CircularIndex.hpp"
  68. #include "NdbMutex.h"
  69. #include "NdbCondition.h"
  70. #include <NdbOut.hpp>
  71. template <class T>
  72. class MemoryChannel
  73. {
  74. public:
  75.   MemoryChannel( int size= 256);
  76.   virtual ~MemoryChannel( );
  77.   void writeChannel( T *t);
  78.   void writeChannelNoSignal( T *t);
  79.   T* readChannel();
  80.   T* tryReadChannel();
  81. private:
  82.   int theSize;
  83.   T **theChannel;
  84.   CircularIndex theWriteIndex;
  85.   CircularIndex theReadIndex;
  86.   NdbMutex* theMutexPtr;
  87.   NdbCondition* theConditionPtr;
  88. };
  89. template <class T> MemoryChannel<T>::MemoryChannel( int size):
  90.         theSize(size),
  91.         theChannel(new T*[size] ),
  92.         theWriteIndex(0, size),
  93.         theReadIndex(0, size)
  94. {
  95.   theMutexPtr = NdbMutex_Create();
  96.   theConditionPtr = NdbCondition_Create();
  97. }
  98. template <class T> MemoryChannel<T>::~MemoryChannel( )
  99. {
  100.   NdbMutex_Destroy(theMutexPtr);
  101.   NdbCondition_Destroy(theConditionPtr);
  102.   delete [] theChannel;
  103. }
  104. template <class T> void MemoryChannel<T>::writeChannel( T *t)
  105. {
  106.   NdbMutex_Lock(theMutexPtr);
  107.   if(full(theWriteIndex, theReadIndex) || theChannel == NULL) abort();
  108.   theChannel[theWriteIndex]= t;
  109.   ++theWriteIndex;
  110.   NdbMutex_Unlock(theMutexPtr);
  111.   NdbCondition_Signal(theConditionPtr);
  112. }
  113. template <class T> void MemoryChannel<T>::writeChannelNoSignal( T *t)
  114. {
  115.   NdbMutex_Lock(theMutexPtr);
  116.   if(full(theWriteIndex, theReadIndex) || theChannel == NULL) abort();
  117.   theChannel[theWriteIndex]= t;
  118.   ++theWriteIndex;
  119.   NdbMutex_Unlock(theMutexPtr);
  120. }
  121. template <class T> T* MemoryChannel<T>::readChannel()
  122. {
  123.   T* tmp;
  124.   NdbMutex_Lock(theMutexPtr);
  125.   while ( empty(theWriteIndex, theReadIndex) )
  126.   {
  127.     NdbCondition_Wait(theConditionPtr,
  128.                         theMutexPtr);    
  129.   }
  130.         
  131.   tmp= theChannel[theReadIndex];
  132.   ++theReadIndex;
  133.   NdbMutex_Unlock(theMutexPtr);
  134.   return tmp;
  135. }
  136. template <class T> T* MemoryChannel<T>::tryReadChannel()
  137. {
  138.   T* tmp= 0;
  139.   NdbMutex_Lock(theMutexPtr);
  140.   if ( !empty(theWriteIndex, theReadIndex) )
  141.   {     
  142.     tmp= theChannel[theReadIndex];
  143.     ++theReadIndex;
  144.   }
  145.   NdbMutex_Unlock(theMutexPtr);
  146.   return tmp;
  147. }
  148. #endif
  149. #endif // MemoryChannel_H