TAO/performance-tests/Sequence_Latency/Thread_Pool/client.cpp

   1 #include "Client_Task.h"
   2 #include "ace/Get_Opt.h"
   3 #include "ace/High_Res_Timer.h"
   4 #include "ace/Sched_Params.h"
   5 #include "ace/Stats.h"
   6 #include "ace/Throughput_Stats.h"
   7 #include "ace/OS_NS_errno.h"
   8
   9 #include "tao/Strategies/advanced_resource.h"
  10
  11 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
  12 int niterations = 1000;
  13 int do_shutdown = 1;
  14 int sz = 512;
  15
  16 const ACE_TCHAR *data_type = ACE_TEXT("octet");
  17
  18 int
  19 parse_args (int argc, ACE_TCHAR *argv[])
  20 {
  21   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("t:s:xk:i:"));
  22   int c;
  23
  24   while ((c = get_opts ()) != -1)
  25     switch (c)
  26       {
  27       case 't':
  28         data_type = get_opts.opt_arg ();
  29
  30         if (ACE_OS::strcmp (data_type, ACE_TEXT("octet")) != 0 &&
  31             ACE_OS::strcmp (data_type, ACE_TEXT("char")) != 0 &&
  32             ACE_OS::strcmp (data_type, ACE_TEXT("long")) != 0 &&
  33             ACE_OS::strcmp (data_type, ACE_TEXT("short")) != 0 &&
  34             ACE_OS::strcmp (data_type, ACE_TEXT("double")) != 0 &&
  35             ACE_OS::strcmp (data_type, ACE_TEXT("longlong")) != 0)
  36           return -1;
  37         break;
  38
  39           case 's':
  40         sz = ACE_OS::atoi (get_opts.opt_arg ());
  41         break;
  42
  43       case 'x':
  44         do_shutdown = 0;
  45         break;
  46
  47       case 'k':
  48         ior = get_opts.opt_arg ();
  49         break;
  50
  51       case 'i':
  52         niterations = ACE_OS::atoi (get_opts.opt_arg ());
  53         break;
  54
  55       case '?':
  56       default:
  57         ACE_ERROR_RETURN ((LM_ERROR,
  58                            "usage:  %s "
  59                            "-t <datatype> "
  60                                                    "-s <size> "
  61                            "-k <ior> "
  62                            "-i <niterations> "
  63                            "-x (disable shutdown) "
  64                            "\n",
  65                            argv [0]),
  66                           -1);
  67       }
  68   // Indicates successful parsing of the command line
  69   return 0;
  70 }
  71
  72 int
  73 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  74 {
  75   int priority =
  76     (ACE_Sched_Params::priority_min (ACE_SCHED_FIFO)
  77      + ACE_Sched_Params::priority_max (ACE_SCHED_FIFO)) / 2;
  78   // Enable FIFO scheduling
  79
  80   if (ACE_OS::sched_params (ACE_Sched_Params (ACE_SCHED_FIFO,
  81                                               priority,
  82                                               ACE_SCOPE_PROCESS)) != 0)
  83     {
  84       if (ACE_OS::last_error () == EPERM)
  85         {
  86           ACE_DEBUG ((LM_DEBUG,
  87                       "client (%P|%t): user is not superuser, "
  88                       "test runs in time-shared class\n"));
  89         }
  90       else
  91         ACE_ERROR ((LM_ERROR,
  92                     "client (%P|%t): sched_params failed\n"));
  93     }
  94
  95
  96   try
  97     {
  98       CORBA::ORB_var orb =
  99         CORBA::ORB_init (argc, argv);
 100
 101       if (parse_args (argc, argv) != 0)
 102         return 1;
 103
 104       CORBA::Object_var object =
 105         orb->string_to_object (ior);
 106
 107       Test::Roundtrip_var roundtrip =
 108         Test::Roundtrip::_narrow (object.in ());
 109
 110       if (CORBA::is_nil (roundtrip.in ()))
 111         {
 112           ACE_ERROR_RETURN ((LM_ERROR,
 113                              "Nil Test::Roundtrip reference <%s>\n",
 114                              ior),
 115                             1);
 116         }
 117
 118           /// Begin the test
 119
 120       ACE_DEBUG ((LM_DEBUG, "Starting threads\n"));
 121
 122       Client_Task task0 (data_type, sz, roundtrip.in (), niterations);
 123       Client_Task task1 (data_type, sz, roundtrip.in (), niterations);
 124       Client_Task task2 (data_type, sz, roundtrip.in (), niterations);
 125       Client_Task task3 (data_type, sz, roundtrip.in (), niterations);
 126
 127       ACE_hrtime_t test_start = ACE_OS::gethrtime ();
 128       task0.activate (THR_NEW_LWP | THR_JOINABLE);
 129       task1.activate (THR_NEW_LWP | THR_JOINABLE);
 130       task2.activate (THR_NEW_LWP | THR_JOINABLE);
 131       task3.activate (THR_NEW_LWP | THR_JOINABLE);
 132
 133       task0.thr_mgr()->wait ();
 134       ACE_hrtime_t test_end = ACE_OS::gethrtime ();
 135
 136       ACE_DEBUG ((LM_DEBUG, "Threads finished\n"));
 137
 138       ACE_DEBUG ((LM_DEBUG, "High resolution timer calibration...."));
 139       ACE_High_Res_Timer::global_scale_factor_type gsf =
 140         ACE_High_Res_Timer::global_scale_factor ();
 141       ACE_DEBUG ((LM_DEBUG, "done\n"));
 142
 143       ACE_Basic_Stats totals;
 144       task0.accumulate_and_dump (totals, ACE_TEXT("Task[0]"), gsf);
 145       task1.accumulate_and_dump (totals, ACE_TEXT("Task[1]"), gsf);
 146       task2.accumulate_and_dump (totals, ACE_TEXT("Task[2]"), gsf);
 147       task3.accumulate_and_dump (totals, ACE_TEXT("Task[3]"), gsf);
 148
 149       totals.dump_results (ACE_TEXT("Total"), gsf);
 150
 151       ACE_Throughput_Stats::dump_throughput (ACE_TEXT("Total"), gsf,
 152                                              test_end - test_start,
 153                                              totals.samples_count ());
 154
 155       if (do_shutdown)
 156         {
 157           roundtrip->shutdown ();
 158         }
 159     }
 160   catch (const CORBA::Exception& ex)
 161     {
 162       ex._tao_print_exception ("Exception caught:");
 163       return 1;
 164     }
 165
 166   return 0;
 167 }