TAO/orbsvcs/tests/CosEvent/lib/Counting_Consumer.h

   1 /* -*- C++ -*- */
   2 //=============================================================================
   3 /**
   4  *  @file   Counting_Consumer.h
   5  *
   6  *  @author Carlos O'Ryan (coryan@cs.wustl.edu)
   7  */
   8 //=============================================================================
   9
  10
  11 #ifndef CEC_COUNTING_CONSUMER_H
  12 #define CEC_COUNTING_CONSUMER_H
  13
  14 #include "cectest_export.h"
  15
  16 #if !defined (ACE_LACKS_PRAGMA_ONCE)
  17 # pragma once
  18 #endif /* ACE_LACKS_PRAGMA_ONCE */
  19
  20 #include "orbsvcs/CosEventCommS.h"
  21 #include "orbsvcs/CosEventChannelAdminC.h"
  22 #include "ace/Task.h"
  23
  24 /**
  25  * @class CEC_Counting_Consumer
  26  *
  27  * @brief Simple consumer object to implement EC tests.
  28  *
  29  * This is a simple consumer that counts the events it receives.
  30  */
  31 class CEC_Test_Export CEC_Counting_Consumer : public POA_CosEventComm::PushConsumer
  32 {
  33 public:
  34   /// Constructor
  35   CEC_Counting_Consumer (const char* name);
  36
  37   /// Simple connect/disconnect methods..
  38   void connect (CosEventChannelAdmin::ConsumerAdmin_ptr consumer_admin);
  39   void disconnect ();
  40
  41   /// Print out an error message if the event count is too far from the
  42   /// expected count.
  43   void dump_results (int expected_count, int tolerance);
  44
  45   // = The CosEventComm::PushConsumer methods
  46
  47   /// The skeleton methods.
  48   virtual void push (const CORBA::Any& events);
  49   virtual void disconnect_push_consumer ();
  50
  51   /// Keep track of the number of events received.
  52   CORBA::ULong event_count;
  53
  54   /// Keep track of the number of disconnect calls received.
  55   CORBA::ULong disconnect_count;
  56
  57 protected:
  58   /// The proxy
  59   CosEventChannelAdmin::ProxyPushSupplier_var supplier_proxy_;
  60
  61   /// The name
  62   const char* name_;
  63 };
  64
  65 // ****************************************************************
  66
  67 /**
  68  * @class CEC_Pull_Counting_Consumer
  69  *
  70  * @brief Simple consumer object to implement EC tests.
  71  *
  72  * This is a simple consumer that counts the events it receives.
  73  */
  74 class CEC_Test_Export CEC_Pull_Counting_Consumer : public POA_CosEventComm::PullConsumer
  75 {
  76 public:
  77   /// Constructor
  78   CEC_Pull_Counting_Consumer (const char* name);
  79
  80   /// Simple connect/disconnect methods..
  81   void connect (CosEventChannelAdmin::ConsumerAdmin_ptr consumer_admin);
  82   void disconnect ();
  83
  84   /// Print out an error message if the event count is too far from the
  85   /// expected count.
  86   void dump_results (int expected_count, int tolerance);
  87
  88   CORBA::Any *pull ();
  89   CORBA::Any *try_pull (CORBA::Boolean_out has_event);
  90
  91   // = The CosEventComm::PullConsumer methods
  92   /// The skeleton methods.
  93   virtual void disconnect_pull_consumer ();
  94
  95   /// Keep track of the number of events received.
  96   CORBA::ULong event_count;
  97
  98   /// Keep track of the number of disconnect calls received.
  99   CORBA::ULong disconnect_count;
 100
 101 protected:
 102   /// The proxy
 103   CosEventChannelAdmin::ProxyPullSupplier_var supplier_proxy_;
 104
 105   /// The name
 106   const char* name_;
 107 };
 108
 109 // ****************************************************************
 110
 111 class CEC_Test_Export CEC_Counting_Consumer_Task : public ACE_Task_Base
 112 {
 113 public:
 114   /// Create the task...
 115   CEC_Counting_Consumer_Task (CEC_Pull_Counting_Consumer *consumer,
 116                               int milliseconds = 0);
 117
 118   // = Check the ACE_Task_Base documentation.
 119   int svc ();
 120
 121   void stop ();
 122   CORBA::ULong pull_count ();
 123
 124   /// Run a single iteration of the test
 125   void run ();
 126
 127 private:
 128   /// The consumer we are turning into an active object
 129   CEC_Pull_Counting_Consumer *consumer_;
 130
 131   /// Set to 1 when the test must stop
 132   int stop_flag_;
 133
 134   /// Count the number of pull() calls
 135   CORBA::ULong pull_count_;
 136
 137   /// If not zero then pause for <milliseconds> before sending each
 138   /// event.
 139   int milliseconds_;
 140
 141   TAO_SYNCH_MUTEX lock_;
 142 };
 143
 144 #endif /* ECT_CONSUMER_H */