TAO/tests/Client_Leaks/client.cpp

   1 #include "Client_Task.h"
   2 #include "ace/Get_Opt.h"
   3
   4 const ACE_TCHAR *ior = ACE_TEXT ("file://test.ior");
   5 int iterations = 100;
   6 int threads = 12;
   7
   8 int
   9 parse_args (int argc, ACE_TCHAR *argv[])
  10 {
  11   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:i:n:"));
  12   int c;
  13
  14   while ((c = get_opts ()) != -1)
  15     switch (c)
  16       {
  17       case 'k':
  18         ior = get_opts.opt_arg ();
  19         break;
  20
  21       case 'i':
  22         iterations = ACE_OS::atoi (get_opts.opt_arg ());
  23         break;
  24
  25       case 'n':
  26         threads = ACE_OS::atoi (get_opts.opt_arg ());
  27         break;
  28
  29       case '?':
  30       default:
  31         ACE_ERROR_RETURN ((LM_ERROR,
  32                            "usage:  %s "
  33                            "-k <ior> "
  34                            "-i <iterations> "
  35                            "-i <threads> "
  36                            "\n",
  37                            argv [0]),
  38                           -1);
  39       }
  40   // Indicates successful parsing of the command line
  41   return 0;
  42 }
  43
  44 int
  45 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  46 {
  47   try
  48     {
  49       CORBA::ORB_var orb =
  50         CORBA::ORB_init (argc, argv);
  51
  52       if (parse_args (argc, argv) != 0)
  53         return 1;
  54
  55       CORBA::Object_var tmp =
  56         orb->string_to_object (ior);
  57
  58       Test::Process_Factory_var process_factory =
  59         Test::Process_Factory::_narrow(tmp.in ());
  60
  61       if (CORBA::is_nil (process_factory.in ()))
  62         {
  63           ACE_ERROR_RETURN ((LM_DEBUG,
  64                              "Nil process reference <%s>\n",
  65                              ior),
  66                             1);
  67         }
  68
  69       Client_Task client_task (process_factory.in (),
  70                                iterations);
  71
  72       if (client_task.activate (THR_NEW_LWP | THR_JOINABLE,
  73                                 threads, 1) == -1)
  74         {
  75           ACE_ERROR ((LM_ERROR, "Error activating client task\n"));
  76         }
  77       ACE_Thread_Manager::instance ()->wait ();
  78
  79       process_factory->shutdown ();
  80
  81       orb->destroy ();
  82
  83       // Only pass the test if 90% of the calls worked
  84       if (client_task.successful_calls () < 0.9 * iterations * threads)
  85         {
  86           ACE_ERROR ((LM_ERROR,
  87                       "ERROR: no calls were successful\n"));
  88         }
  89     }
  90   catch (const CORBA::Exception& ex)
  91     {
  92       ex._tao_print_exception ("Exception caught:");
  93       return 1;
  94     }
  95
  96   return 0;
  97 }