TAO/orbsvcs/tests/EC_Throughput/ECT_Supplier_Driver.cpp

   1 #include "ECT_Supplier_Driver.h"
   2
   3 #include "orbsvcs/CosNamingC.h"
   4 #include "orbsvcs/Event_Utilities.h"
   5 #include "orbsvcs/Event_Service_Constants.h"
   6 #include "orbsvcs/Time_Utilities.h"
   7
   8 #include "tao/Timeprobe.h"
   9 #include "tao/debug.h"
  10
  11 #include "ace/Get_Opt.h"
  12 #include <memory>
  13 #include "ace/Sched_Params.h"
  14 #include "ace/OS_NS_errno.h"
  15 #include "ace/OS_NS_unistd.h"
  16
  17 int
  18 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  19 {
  20   ECT_Supplier_Driver driver;
  21   return driver.run (argc, argv);
  22 }
  23
  24 // ****************************************************************
  25
  26 ECT_Supplier_Driver::ECT_Supplier_Driver ()
  27   : n_suppliers_ (1),
  28     burst_count_ (10),
  29     burst_size_ (100),
  30     event_size_ (128),
  31     burst_pause_ (100),
  32     type_start_ (ACE_ES_EVENT_UNDEFINED),
  33     type_count_ (1),
  34     pid_file_name_ (0)
  35 {
  36 }
  37
  38 ECT_Supplier_Driver::~ECT_Supplier_Driver ()
  39 {
  40 }
  41
  42 void
  43 ECT_Supplier_Driver::shutdown_consumer (void*)
  44 {
  45 }
  46
  47 int
  48 ECT_Supplier_Driver::run (int argc, ACE_TCHAR* argv[])
  49 {
  50   try
  51     {
  52       CORBA::ORB_var orb =
  53         CORBA::ORB_init (argc, argv);
  54
  55       CORBA::Object_var poa_object =
  56         orb->resolve_initial_references("RootPOA");
  57
  58       if (CORBA::is_nil (poa_object.in ()))
  59         ACE_ERROR_RETURN ((LM_ERROR,
  60                            " (%P|%t) Unable to initialize the POA.\n"),
  61                           1);
  62
  63       PortableServer::POA_var root_poa =
  64         PortableServer::POA::_narrow (poa_object.in ());
  65
  66       PortableServer::POAManager_var poa_manager =
  67         root_poa->the_POAManager ();
  68
  69       if (this->parse_args (argc, argv))
  70         return 1;
  71
  72       if (TAO_debug_level > 0)
  73         {
  74           ACE_DEBUG ((LM_DEBUG,
  75                       "Execution parameters:\n"
  76                       "  suppliers = <%d>\n"
  77                       "  burst count = <%d>\n"
  78                       "  burst size = <%d>\n"
  79                       "  event size = <%d>\n"
  80                       "  burst pause = <%d>\n"
  81                       "  type start = <%d>\n"
  82                       "  type count = <%d>\n"
  83                       "  pid file name = <%s>\n",
  84
  85                       this->n_suppliers_,
  86                       this->burst_count_,
  87                       this->burst_size_,
  88                       this->event_size_,
  89                       this->burst_pause_,
  90                       this->type_start_,
  91                       this->type_count_,
  92
  93                       this->pid_file_name_?this->pid_file_name_:ACE_TEXT("nil")));
  94         }
  95
  96       if (this->pid_file_name_ != 0)
  97         {
  98           FILE* pid = ACE_OS::fopen (this->pid_file_name_, "w");
  99           if (pid != 0)
 100             {
 101               ACE_OS::fprintf (pid, "%ld\n",
 102                                static_cast<long> (ACE_OS::getpid ()));
 103               ACE_OS::fclose (pid);
 104             }
 105         }
 106
 107       int min_priority = ACE_Sched_Params::priority_min (ACE_SCHED_FIFO);
 108
 109       // Enable FIFO scheduling
 110       if (ACE_OS::sched_params (ACE_Sched_Params (ACE_SCHED_FIFO,
 111                                                   min_priority,
 112                                                   ACE_SCOPE_PROCESS)) != 0)
 113         {
 114           if (ACE_OS::last_error () == EPERM)
 115             ACE_DEBUG ((LM_DEBUG,
 116                         "%s: user is not superuser, "
 117                         "so remain in time-sharing class\n", argv[0]));
 118           else
 119             ACE_ERROR ((LM_ERROR,
 120                         "%s: ACE_OS::sched_params failed\n", argv[0]));
 121         }
 122
 123       if (ACE_OS::thr_setprio (min_priority) == -1)
 124         {
 125           ACE_ERROR ((LM_ERROR, "(%P|%t) main thr_setprio failed,"
 126                       "no real-time features\n"));
 127         }
 128
 129       CORBA::Object_var naming_obj =
 130         orb->resolve_initial_references ("NameService");
 131
 132       if (CORBA::is_nil (naming_obj.in ()))
 133         ACE_ERROR_RETURN ((LM_ERROR,
 134                            " (%P|%t) Unable to get the Naming Service.\n"),
 135                           1);
 136
 137       CosNaming::NamingContext_var naming_context =
 138         CosNaming::NamingContext::_narrow (naming_obj.in ());
 139
 140       CosNaming::Name schedule_name (1);
 141       schedule_name.length (1);
 142       schedule_name[0].id = CORBA::string_dup ("ScheduleService");
 143
 144       CORBA::Object_var sched_obj =
 145         naming_context->resolve (schedule_name);
 146       if (CORBA::is_nil (sched_obj.in ()))
 147         return 1;
 148       RtecScheduler::Scheduler_var scheduler =
 149         RtecScheduler::Scheduler::_narrow (sched_obj.in ());
 150       CosNaming::Name name (1);
 151       name.length (1);
 152       name[0].id = CORBA::string_dup ("EventService");
 153
 154       CORBA::Object_var ec_obj =
 155         naming_context->resolve (name);
 156
 157       RtecEventChannelAdmin::EventChannel_var channel;
 158       if (CORBA::is_nil (ec_obj.in ()))
 159         channel = RtecEventChannelAdmin::EventChannel::_nil ();
 160       else
 161         channel = RtecEventChannelAdmin::EventChannel::_narrow (ec_obj.in ());
 162
 163       poa_manager->activate ();
 164
 165       this->connect_suppliers (scheduler.in (),
 166                                channel.in ());
 167
 168       ACE_DEBUG ((LM_DEBUG, "connected supplier(s)\n"));
 169
 170       this->activate_suppliers ();
 171
 172       ACE_DEBUG ((LM_DEBUG, "suppliers are active\n"));
 173
 174       // Wait for the supplier threads...
 175       if (ACE_Thread_Manager::instance ()->wait () == -1)
 176         {
 177           ACE_ERROR ((LM_ERROR, "Thread_Manager wait failed\n"));
 178           return 1;
 179         }
 180
 181       ACE_DEBUG ((LM_DEBUG, "suppliers finished\n"));
 182
 183       this->dump_results ();
 184
 185       this->disconnect_suppliers ();
 186
 187       ACE_DEBUG ((LM_DEBUG, "suppliers disconnected\n"));
 188
 189       // @@ Deactivate the suppliers (as CORBA Objects?)
 190
 191       root_poa->destroy (true, true);
 192
 193       orb->destroy ();
 194
 195       ACE_DEBUG ((LM_DEBUG, "orb and poa destroyed\n"));
 196     }
 197   catch (const CORBA::SystemException& sys_ex)
 198     {
 199       sys_ex._tao_print_exception ("SYS_EX");
 200     }
 201   catch (const CORBA::Exception& ex)
 202     {
 203       ex._tao_print_exception ("NON SYS EX");
 204     }
 205   return 0;
 206 }
 207
 208 void
 209 ECT_Supplier_Driver::connect_suppliers
 210     (RtecScheduler::Scheduler_ptr scheduler,
 211      RtecEventChannelAdmin::EventChannel_ptr channel)
 212 {
 213   for (int i = 0; i < this->n_suppliers_; ++i)
 214     {
 215       char buf[BUFSIZ];
 216       ACE_OS::sprintf (buf, "supplier_%02d", i);
 217
 218       ACE_NEW (this->suppliers_[i], Test_Supplier (this));
 219
 220       this->suppliers_[i]->connect (scheduler,
 221                                     buf,
 222                                     this->burst_count_,
 223                                     this->burst_size_,
 224                                     this->event_size_,
 225                                     this->burst_pause_,
 226                                     this->type_start_,
 227                                     this->type_count_,
 228                                     channel);
 229     }
 230 }
 231
 232 void
 233 ECT_Supplier_Driver::activate_suppliers ()
 234 {
 235   for (int i = 0; i < this->n_suppliers_; ++i)
 236     {
 237       this->suppliers_[i]->activate ();
 238     }
 239 }
 240
 241 void
 242 ECT_Supplier_Driver::disconnect_suppliers ()
 243 {
 244   for (int i = 0; i < this->n_suppliers_; ++i)
 245     {
 246       this->suppliers_[i]->disconnect ();
 247
 248       delete this->suppliers_[i];
 249       this->suppliers_[i] = 0;
 250     }
 251 }
 252
 253 void
 254 ECT_Supplier_Driver::dump_results ()
 255 {
 256   ACE_High_Res_Timer::global_scale_factor_type gsf =
 257     ACE_High_Res_Timer::global_scale_factor ();
 258
 259   ACE_Throughput_Stats throughput;
 260   for (int i = 0; i < this->n_suppliers_; ++i)
 261     {
 262       ACE_TCHAR buf[BUFSIZ];
 263       ACE_OS::sprintf (buf, ACE_TEXT("supplier_%02d"), i);
 264
 265       this->suppliers_[i]->dump_results (buf, gsf);
 266       this->suppliers_[i]->accumulate (throughput);
 267     }
 268   throughput.dump_results (ACE_TEXT("ECT_Supplier/totals"), gsf);
 269 }
 270
 271 int
 272 ECT_Supplier_Driver::parse_args (int argc, ACE_TCHAR *argv [])
 273 {
 274   ACE_Get_Opt get_opt (argc, argv, ACE_TEXT("ds:u:n:t:b:h:p:"));
 275   int opt;
 276
 277   while ((opt = get_opt ()) != EOF)
 278     {
 279       switch (opt)
 280         {
 281         case 's':
 282           this->n_suppliers_ = ACE_OS::atoi (get_opt.opt_arg ());
 283           break;
 284
 285         case 'u':
 286           this->burst_count_ = ACE_OS::atoi (get_opt.opt_arg ());
 287           break;
 288
 289         case 'n':
 290           this->burst_size_ = ACE_OS::atoi (get_opt.opt_arg ());
 291           break;
 292
 293         case 'b':
 294           this->event_size_ = ACE_OS::atoi (get_opt.opt_arg ());
 295           break;
 296
 297         case 't':
 298           this->burst_pause_ = ACE_OS::atoi (get_opt.opt_arg ());
 299           break;
 300
 301         case 'h':
 302           {
 303             char* aux;
 304                 char* arg = ACE_OS::strtok_r (ACE_TEXT_ALWAYS_CHAR(get_opt.opt_arg ()), ",", &aux);
 305
 306             this->type_start_ = ACE_ES_EVENT_UNDEFINED + ACE_OS::atoi (arg);
 307                 arg = ACE_OS::strtok_r (0, ",", &aux);
 308             this->type_count_ = ACE_OS::atoi (arg);
 309           }
 310           break;
 311
 312         case 'p':
 313           this->pid_file_name_ = get_opt.opt_arg ();
 314           break;
 315
 316         case '?':
 317         default:
 318           ACE_DEBUG ((LM_DEBUG,
 319                       "Usage: %s "
 320                       "[ORB options] "
 321                       "-s <nsuppliers> "
 322                       "-u <burst count> "
 323                       "-n <burst size> "
 324                       "-b <event payload size> "
 325                       "-t <burst pause (usecs)> "
 326                       "-h <type_start,type_count> "
 327                       "-p <pid file name> "
 328                       "\n",
 329                       argv[0]));
 330           return -1;
 331         }
 332     }
 333
 334   if (this->burst_count_ <= 0)
 335     {
 336       ACE_DEBUG ((LM_DEBUG,
 337                   "%s: burst count (%d) is out of range, "
 338                   "reset to default (%d)\n",
 339                   argv[0], this->burst_count_,
 340                   100));
 341       this->burst_count_ = 100;
 342     }
 343
 344   if (this->burst_size_ <= 0)
 345     {
 346       ACE_DEBUG ((LM_DEBUG,
 347                   "%s: burst size (%d) is out of range, "
 348                   "reset to default (%d)\n",
 349                   argv[0], this->burst_size_,
 350                   10));
 351       this->burst_size_ = 10;
 352     }
 353
 354   if (this->event_size_ < 0)
 355     {
 356       ACE_DEBUG ((LM_DEBUG,
 357                   "%s: event size (%d) is out of range, "
 358                   "reseting to default (%d)\n",
 359                   argv[0], this->event_size_,
 360                   128));
 361       this->event_size_ = 128;
 362     }
 363
 364   if (this->n_suppliers_ <= 0)
 365     {
 366       this->n_suppliers_ = 1;
 367       ACE_ERROR_RETURN ((LM_ERROR,
 368                          "%s: number of suppliers out of range, "
 369                          "reset to default (%d)\n",
 370                          argv[0], 1), -1);
 371     }
 372
 373   return 0;
 374 }