TAO/orbsvcs/tests/Notify/Basic/Simple.cpp

   1 #include "ace/Arg_Shifter.h"
   2 #include "ace/Get_Opt.h"
   3 #include "tao/debug.h"
   4 #include "Simple.h"
   5
   6
   7
   8 //***************************************************************************
   9
  10 Event_AnyPushConsumer::Event_AnyPushConsumer (Simple_Test *test_client)
  11   : test_client_ (test_client)
  12 {
  13 }
  14
  15 void
  16 Event_AnyPushConsumer::push (const CORBA::Any & data)
  17 {
  18   int event_num;
  19   data >>= event_num;
  20
  21   if (TAO_debug_level)
  22     ACE_DEBUG ((LM_DEBUG,
  23                 "Received event# %d\n",
  24                 event_num));
  25
  26   this->test_client_->on_event_received ();
  27 }
  28
  29 //***************************************************************************
  30
  31 Event_AnyPushSupplier::Event_AnyPushSupplier (Simple_Test* test_client)
  32   : test_client_ (test_client)
  33 {
  34 }
  35
  36 Event_AnyPushSupplier::~Event_AnyPushSupplier (void)
  37 {
  38 }
  39
  40 //***************************************************************************
  41
  42 Simple_Test::Simple_Test (void)
  43   : event_count_ (5)
  44 {
  45 }
  46
  47 Simple_Test::~Simple_Test (void)
  48 {
  49 }
  50
  51 int
  52 Simple_Test::init (int argc,
  53                    ACE_TCHAR* argv [])
  54 {
  55   // Initialized the base class.
  56   Notify_Test_Client::init (argc,
  57                             argv);
  58
  59   // Create all participents.
  60   this->create_EC ();
  61
  62   CosNotifyChannelAdmin::AdminID adminid;
  63
  64   supplier_admin_ =
  65     this->ec_->new_for_suppliers (this->ifgop_,
  66                                   adminid);
  67
  68   ACE_ASSERT (!CORBA::is_nil (supplier_admin_.in ()));
  69
  70   consumer_admin_ =
  71     this->ec_->new_for_consumers (this->ifgop_,
  72                                   adminid);
  73
  74   ACE_ASSERT (!CORBA::is_nil (consumer_admin_.in ()));
  75
  76   ACE_NEW_RETURN (this->consumer_,
  77                   Event_AnyPushConsumer (this),
  78                   -1);
  79   this->consumer_->init (root_poa_.in ());
  80   this->consumer_->connect (this->consumer_admin_.in ());
  81
  82   Event_AnyPushConsumer* consumer2;
  83   ACE_NEW_RETURN (consumer2,
  84                   Event_AnyPushConsumer (this),
  85                   -1);
  86   consumer2->init (root_poa_.in ());
  87   consumer2->connect (this->consumer_admin_.in ());
  88
  89   ACE_NEW_RETURN (this->supplier_,
  90                   Event_AnyPushSupplier (this),
  91                   -1);
  92   this->supplier_->init (root_poa_.in ());
  93
  94   this->supplier_->connect (this->supplier_admin_.in ());
  95
  96   consumer_start( 0 );
  97
  98   return 0;
  99 }
 100
 101 int
 102 Simple_Test::parse_args (int argc,
 103                          ACE_TCHAR *argv[])
 104 {
 105     ACE_Arg_Shifter arg_shifter (argc,
 106                                  argv);
 107
 108     const ACE_TCHAR *current_arg = 0;
 109
 110     while (arg_shifter.is_anything_left ())
 111     {
 112       if (0 != (current_arg = arg_shifter.get_the_parameter (ACE_TEXT("-events"))))
 113         {
 114           this->event_count_ = ACE_OS::atoi (current_arg);
 115           // The number of events to send/receive.
 116           arg_shifter.consume_arg ();
 117         }
 118       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-?")) == 0)
 119         {
 120           ACE_DEBUG ((LM_DEBUG,
 121                       "usage: %s "
 122                       "-events event_count\n",
 123                       argv[0],
 124                       argv[0]));
 125
 126           arg_shifter.consume_arg ();
 127
 128           return -1;
 129         }
 130       else
 131         {
 132           arg_shifter.ignore_arg ();
 133         }
 134     }
 135     return 0;
 136 }
 137
 138 void
 139 Simple_Test::create_EC (void)
 140 {
 141   CosNotifyChannelAdmin::ChannelID id;
 142
 143   this->ec_ = notify_factory_->create_channel (this->initial_qos_,
 144                                                this->initial_admin_,
 145                                                id);
 146
 147   ACE_ASSERT (!CORBA::is_nil (ec_.in ()));
 148 }
 149
 150 void
 151 Simple_Test::on_event_received (void)
 152 {
 153   ++this->result_count_;
 154
 155   if (TAO_debug_level)
 156     ACE_DEBUG ((LM_DEBUG,
 157                 "event count = #%d\n",
 158                 this->result_count_.value ()));
 159
 160   if (this->result_count_ == 2 * this->event_count_)
 161     {
 162       this->end_test ();
 163     }
 164 }
 165
 166 void
 167 Simple_Test::run_test (void)
 168 {
 169   CORBA::Any data;
 170
 171   for (int i = 0; i < this->event_count_; ++i)
 172     {
 173       data <<= (CORBA::Long)i;
 174
 175       this->supplier_->send_event (data);
 176     }
 177 }
 178
 179 void
 180 Simple_Test::end_test (void)
 181 {
 182   consumer_done( 0 );
 183 }
 184
 185 int
 186 Simple_Test::check_results (void)
 187 {
 188   // Destroy the channel
 189   this->ec_->destroy ();
 190
 191   if (this->result_count_ == 2 * this->event_count_)
 192     {
 193       ACE_DEBUG ((LM_DEBUG,
 194                   "Events test success\n"));
 195       return 0;
 196     }
 197   else
 198     {
 199       ACE_DEBUG ((LM_DEBUG,
 200                   "Events test failed!\n"));
 201       return 1;
 202     }
 203 }
 204
 205 //***************************************************************************
 206
 207 int
 208 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 209 {
 210   Simple_Test events;
 211
 212   if (events.parse_args (argc, argv) == -1)
 213     {
 214       return 1;
 215     }
 216
 217   try
 218     {
 219       events.init (argc,
 220                    argv);
 221
 222       events.run_test ();
 223
 224       events.ORB_run( );
 225     }
 226   catch (const CORBA::Exception& se)
 227     {
 228       se._tao_print_exception ("Error: ");
 229       return 1;
 230     }
 231
 232   int status = 0;
 233
 234   try
 235     {
 236       status = events.check_results ();
 237     }
 238   catch (const CORBA::Exception& se)
 239     {
 240       se._tao_print_exception ("Error: ");
 241       status = 1;
 242     }
 243
 244   return status;
 245 }