TAO/orbsvcs/tests/Notify/lib/Application_Command.cpp

   1 #include "Application_Command.h"
   2
   3
   4 #include "tao/PortableServer/PortableServer.h"
   5 #include "LookupManager.h"
   6 #include "Name.h"
   7 #include "Activation_Manager.h"
   8 #include "Driver_Base.h"
   9 #include "Priority_Mapping.h"
  10
  11 TAO_Notify_Tests_Application_Command::TAO_Notify_Tests_Application_Command ()
  12   : dump_samples_ (0)
  13 {
  14 }
  15
  16 TAO_Notify_Tests_Application_Command::~TAO_Notify_Tests_Application_Command ()
  17 {
  18 }
  19
  20 const char*
  21 TAO_Notify_Tests_Application_Command::get_name ()
  22 {
  23   return TAO_Notify_Tests_Application_Command::name ();
  24 }
  25
  26 const char*
  27 TAO_Notify_Tests_Application_Command::name ()
  28 {
  29   return TAO_Notify_Tests_Name::application_command;
  30 }
  31
  32 void
  33 TAO_Notify_Tests_Application_Command::init (ACE_Arg_Shifter& arg_shifter)
  34 {
  35   if (arg_shifter.is_anything_left ())
  36     {
  37       /// -Init | Run | Shutdown
  38       if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-Init")) == 0)
  39         {
  40           this->command_ = INIT;
  41
  42           arg_shifter.consume_arg ();
  43         }
  44       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-Run")) == 0)
  45         {
  46           this->command_ = RUN;
  47
  48           arg_shifter.consume_arg ();
  49         }
  50       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-WaitForEvents")) == 0)
  51         {
  52           this->command_ = WAIT_FOR_EVENTS;
  53
  54           arg_shifter.consume_arg ();
  55         }
  56       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-Shutdown")) == 0)
  57         {
  58           this->command_ = SHUTDOWN;
  59
  60           arg_shifter.consume_arg ();
  61         }
  62       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-DumpStats")) == 0)
  63         {
  64           this->command_ = DUMP_STATE;
  65
  66           arg_shifter.consume_arg ();
  67
  68           if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-Samples")) == 0)
  69             {
  70               this->dump_samples_ = 1;
  71
  72               arg_shifter.consume_arg ();
  73             }
  74         }
  75       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-SignalPeer")) == 0)
  76         {
  77           this->command_ = SIGNAL_PEER;
  78
  79           arg_shifter.consume_arg ();
  80         }
  81       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-WaitToStart")) == 0)
  82         {
  83           this->command_ = WAIT_TO_START;
  84
  85           arg_shifter.consume_arg ();
  86         }
  87
  88     }
  89 }
  90
  91 void
  92 TAO_Notify_Tests_Application_Command::handle_init ()
  93 {
  94   /// Fetch the root poa.
  95   PortableServer::POA_var root_poa;
  96   LOOKUP_MANAGER->resolve (root_poa);
  97
  98   PortableServer::POAManager_var poa_manager =
  99     root_poa->the_POAManager ();
 100
 101   /// Activate the root POA.
 102   poa_manager->activate ();
 103
 104   TAO_Notify_Tests_Priority_Mapping* mapping = new TAO_Notify_Tests_Priority_Mapping ();
 105
 106   LOOKUP_MANAGER->_register (mapping);
 107 }
 108
 109 void
 110 TAO_Notify_Tests_Application_Command::handle_wait_for_completion ()
 111 {
 112   ACE_DEBUG ((LM_DEBUG, "(%P, %t) Waiting for suppliers and consumers to finish...\n"));
 113
 114   TAO_Notify_Tests_Activation_Manager* act_mgr = 0;
 115   LOOKUP_MANAGER->resolve (act_mgr);
 116
 117   act_mgr->wait_for_completion ();
 118 }
 119
 120 void
 121 TAO_Notify_Tests_Application_Command::handle_shutdown ()
 122 {
 123   ACE_DEBUG ((LM_DEBUG, "(%P, %t)Shutting down the Application...\n"));
 124
 125   TAO_Notify_Tests_Driver_Base* driver = 0;
 126   LOOKUP_MANAGER->resolve (driver);
 127
 128   driver->shutdown ();
 129 }
 130
 131 void
 132 TAO_Notify_Tests_Application_Command::handle_dump_stats ()
 133 {
 134   ACE_DEBUG ((LM_DEBUG, "(%P, %t)Dumpimg stats...\n"));
 135
 136   TAO_Notify_Tests_Activation_Manager* act_mgr = 0;
 137   LOOKUP_MANAGER->resolve (act_mgr);
 138
 139   act_mgr->dump_stats (this->dump_samples_);
 140 }
 141
 142 void
 143 TAO_Notify_Tests_Application_Command::handle_run ()
 144 {
 145   // Run the Consumers
 146
 147
 148   //= Run the Suppliers
 149   TAO_Notify_Tests_Activation_Manager* act_mgr = 0;
 150   LOOKUP_MANAGER->resolve (act_mgr);
 151
 152   if (act_mgr->activate_suppliers () == 0)
 153     ACE_DEBUG ((LM_DEBUG, "Suppliers activated...\n"));
 154   else
 155     ACE_DEBUG ((LM_DEBUG, "Suppliers activation failed!...\n"));
 156 }
 157
 158 void
 159 TAO_Notify_Tests_Application_Command::handle_signal_peer ()
 160 {
 161   TAO_Notify_Tests_Activation_Manager* act_mgr = 0;
 162   LOOKUP_MANAGER->resolve (act_mgr);
 163
 164   act_mgr->signal_peer ();
 165 }
 166
 167 void
 168 TAO_Notify_Tests_Application_Command::handle_wait_to_start ()
 169 {
 170   TAO_Notify_Tests_Activation_Manager* act_mgr = 0;
 171   LOOKUP_MANAGER->resolve (act_mgr);
 172
 173   act_mgr->write_ior ();
 174
 175   act_mgr->wait_for_start_signal ();
 176 }
 177
 178 void
 179 TAO_Notify_Tests_Application_Command::execute_i ()
 180 {
 181   if (this->command_ == INIT)
 182     {
 183       this->handle_init ();
 184     }
 185   else if (this->command_ == RUN)
 186     {
 187       this->handle_run ();
 188     }
 189   else if (this->command_ == WAIT_FOR_EVENTS)
 190     {
 191       this->handle_wait_for_completion ();
 192     }
 193   else if (this->command_ == SHUTDOWN)
 194     {
 195       this->handle_shutdown ();
 196     }
 197   else if (this->command_ == DUMP_STATE)
 198     {
 199       this->handle_dump_stats ();
 200     }
 201   else if (this->command_ == SIGNAL_PEER)
 202     {
 203       this->handle_signal_peer ();
 204     }
 205   else if (this->command_ == WAIT_TO_START)
 206     {
 207       this->handle_wait_to_start ();
 208     }
 209 }