ACE/examples/APG/ThreadSafety/RW_Lock.cpp

   1 #include "ace/config-lite.h"
   2 #if defined (ACE_HAS_THREADS)
   3
   4 #include "ace/Synch.h"
   5 #include "ace/Containers.h"
   6 #include "ace/Task.h"
   7
   8 class Device
   9 {
  10 public:
  11   Device (int id) : id_(id)
  12   { }
  13
  14   int id_;
  15 };
  16
  17 typedef ACE_DLList<Device> DeviceList;
  18 typedef ACE_DLList_Iterator<Device> DeviceListIterator;
  19
  20 // Listing 1 code/ch14
  21 class HA_DiscoveryAgent
  22 {
  23 public:
  24   void add_device (Device *device)
  25   {
  26     ACE_WRITE_GUARD (ACE_RW_Thread_Mutex, mon, rwmutex_);
  27     list_add_item_i (device);
  28   }
  29
  30   void remove_device (Device *device)
  31   {
  32     ACE_READ_GUARD (ACE_RW_Thread_Mutex, mon, rwmutex_);
  33     list_remove_item_i(device);
  34   }
  35
  36   int contains_device (Device *device)
  37   {
  38     ACE_READ_GUARD_RETURN
  39       (ACE_RW_Thread_Mutex, mon, rwmutex_, -1);
  40     return list_contains_item_i (device);
  41   }
  42
  43 private:
  44   void list_add_item_i (Device * device);
  45   int list_contains_item_i (Device * device);
  46   void list_remove_item_i (Device* device);
  47
  48 private:
  49   DeviceList deviceList_;
  50   ACE_RW_Thread_Mutex rwmutex_;
  51 };
  52 // Listing 1
  53
  54 void
  55 HA_DiscoveryAgent::list_add_item_i (Device *device)
  56 {
  57   deviceList_.insert_tail (device);
  58 }
  59
  60 int
  61 HA_DiscoveryAgent::list_contains_item_i (Device *device)
  62 {
  63   DeviceListIterator iter (deviceList_);
  64   while (!iter.done ())
  65     {
  66       if (iter.next () == device)
  67         return 1;
  68       iter++;
  69     }
  70
  71   return 0;
  72 }
  73
  74 void
  75 HA_DiscoveryAgent::list_remove_item_i (Device *device)
  76 {
  77   DeviceListIterator iter (deviceList_);
  78   while (!iter.done ())
  79     {
  80       if (iter.next () == device)
  81         {
  82           iter.remove ();
  83           break;
  84         }
  85       iter++;
  86     }
  87 }
  88
  89 static Device *devices[100];
  90
  91 class Runner : public ACE_Task_Base
  92 {
  93 public:
  94   Runner(HA_DiscoveryAgent &agent) : agent_(agent)
  95   { }
  96
  97   virtual int svc ()
  98   {
  99     ACE_ASSERT(agent_.contains_device(devices[9]) == 1);
 100     agent_.remove_device (devices[9]);
 101     ACE_ASSERT(agent_.contains_device(devices[9]) == 0);
 102     return 0;
 103   }
 104
 105 private:
 106   HA_DiscoveryAgent &agent_;
 107 };
 108
 109 int ACE_TMAIN (int, ACE_TCHAR *[])
 110 {
 111   HA_DiscoveryAgent agent;
 112
 113   for (int i = 0; i < 100; i++)
 114     {
 115       devices[i]  = new Device (i);
 116       agent.add_device (devices[i]);
 117     }
 118
 119   Runner runner (agent);
 120   runner.activate ();
 121
 122   runner.wait ();
 123
 124   return 0;
 125 }
 126
 127 #else
 128 #include "ace/OS_main.h"
 129 #include "ace/OS_NS_stdio.h"
 130
 131 int ACE_TMAIN (int, ACE_TCHAR *[])
 132 {
 133   ACE_OS::puts (ACE_TEXT ("This example requires threads."));
 134   return 0;
 135 }
 136
 137 #endif /* ACE_HAS_THREADS */