ACE/tests/MT_Reactor_Timer_Test.h

   1 /* -*- C++ -*- */
   2
   3 //=============================================================================
   4 /**
   5  *  @file    MT_Reactor_Timer_Test.h
   6  *
   7  *    This file contains class definitions needed for template
   8  *    instantiation in the MT_Reactor_Timer_Test.cpp file.
   9  *
  10  *  @author Steve Huston <shuston@riverace.com>
  11  */
  12 //=============================================================================
  13
  14
  15 #ifndef ACE_TESTS_MT_REACTOR_TIMER_TEST_H
  16 #define ACE_TESTS_MT_REACTOR_TIMER_TEST_H
  17
  18 #include "ace/Reactor.h"
  19
  20 #if !defined (ACE_LACKS_PRAGMA_ONCE)
  21 # pragma once
  22 #endif /* ACE_LACKS_PRAGMA_ONCE */
  23
  24 #include "ace/Task.h"
  25 #include "ace/Pipe.h"
  26
  27 /**
  28  * @class Time_Handler
  29  *
  30  * @brief Test out the multi-threading features of the Reactor's timer
  31  * mechanism.
  32  */
  33 class Time_Handler : public ACE_Task<ACE_SYNCH>
  34 {
  35 public:
  36   Time_Handler ();
  37
  38   void setup ();
  39
  40   int verify_results();
  41
  42   /// Run by a daemon thread to handle deferred processing.
  43   virtual int svc ();
  44
  45   virtual int handle_timeout (const ACE_Time_Value &tv,
  46                               const void *arg);
  47
  48 private:
  49   enum
  50   {
  51     TIMER_SLOTS = 10
  52   };
  53
  54   /**
  55    * The timer_id_ array holds timer IDs.  They also have some other values
  56    * that are specific to this test:
  57    *    -1  the timer has not been set
  58    *    -2  the timer was set, but has been cancelled
  59    *    -3  the timer was set, and it already fired
  60    */
  61   long timer_id_[TIMER_SLOTS];
  62   enum { TIMER_NOTSET = -1, TIMER_CANCELLED = -2, TIMER_FIRED = -3 };
  63
  64   long prev_timer_;
  65
  66 #if defined ACE_HAS_THREADS
  67   ACE_Thread_Mutex  lock_;
  68 #endif /* ACE_HAS_THREADS */
  69 };
  70
  71 /**
  72  * @class Dispatch_Count_Handler
  73  *
  74  * @brief A simple test to ensure that the Reactor counts the number of
  75  * dispatches correctly.
  76  */
  77 class Dispatch_Count_Handler : public ACE_Event_Handler
  78 {
  79 public:
  80   Dispatch_Count_Handler ();
  81
  82   /// Clean up resources from the Reactor.
  83   int handle_close (ACE_HANDLE h,
  84                     ACE_Reactor_Mask m);
  85
  86   /// Keep track of the number of timeouts.
  87   virtual int handle_timeout (const ACE_Time_Value &tv,
  88                               const void *arg);
  89
  90   /// Keep track of the number of I/O events.
  91   virtual int handle_input (ACE_HANDLE);
  92
  93   /// Keep track of the number of notifies.
  94   virtual int handle_exception (ACE_HANDLE);
  95
  96   /// Verify that the expected events did happen.
  97   int verify_results ();
  98
  99 private:
 100   /// Provide something to trigger I/O.
 101   ACE_Pipe pipe_;
 102
 103   int input_seen_;
 104   int notify_seen_;
 105   size_t timers_fired_;
 106 };
 107
 108 #endif /* ACE_TESTS_MT_REACTOR_TIMER_TEST_H */