TAO/tests/POA/Forwarding/client.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file     client.cpp
   5  *
   6  *   This is a simple test client implementation.  Also looks out
   7  *   for forwarding exceptions
   8  *
   9  *  @author  Irfan Pyarali
  10  */
  11 //=============================================================================
  12
  13
  14 #include "ace/streams.h"
  15 #include "ace/Get_Opt.h"
  16 #include "ace/Read_Buffer.h"
  17 #include "testC.h"
  18
  19 static ACE_TCHAR *IOR = 0;
  20 static int servers = 2;
  21 static int iterations = 3;
  22
  23 static int
  24 parse_args (int argc, ACE_TCHAR **argv)
  25 {
  26   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("s:i:k:"));
  27   int c;
  28
  29   while ((c = get_opts ()) != -1)
  30     switch (c)
  31       {
  32       case 'k':
  33         IOR = get_opts.opt_arg ();
  34         break;
  35       case 's':
  36         servers = ACE_OS::atoi (get_opts.opt_arg ());
  37         break;
  38       case 'i':
  39         iterations = ACE_OS::atoi (get_opts.opt_arg ());
  40         break;
  41       case '?':
  42       default:
  43         ACE_ERROR_RETURN ((LM_ERROR,
  44                            "usage:  %s"
  45                            "-k IOR\n"
  46                            "-i iterations\n"
  47                            "\n",
  48                            argv [0]),
  49                           -1);
  50       }
  51
  52   if (IOR == 0)
  53     ACE_ERROR_RETURN ((LM_ERROR,
  54                        "Please specify the IOR for the servant\n"),
  55                       -1);
  56
  57   // Indicates successful parsing of command line.
  58   return 0;
  59 }
  60
  61 void
  62 do_calls (test_ptr test)
  63 {
  64   for (int j = 1; j <= servers; j++)
  65     {
  66       for (int i = 1; i <= iterations; i++)
  67         {
  68           // Invoke the doit() method of the test reference.
  69           CORBA::Long result = test->doit ();
  70
  71           // Print the result of doit () method of the test reference.
  72           ACE_DEBUG ((LM_DEBUG,
  73                       "doit() returned %d\n",
  74                       result));
  75         }
  76
  77       // Don't forward the last server
  78       if (j != servers)
  79         {
  80           ACE_DEBUG ((LM_DEBUG,
  81                       "Asking server to forward next time\n"));
  82           test->forward ();
  83         }
  84     }
  85 }
  86
  87
  88 int
  89 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  90 {
  91   try
  92     {
  93       // Initialize the ORB
  94       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
  95
  96       // Initialize options based on command-line arguments.
  97       int parse_args_result =
  98         parse_args (argc, argv);
  99       if (parse_args_result != 0)
 100         return parse_args_result;
 101
 102       // Get an object reference from the argument string.
 103       CORBA::Object_var object =
 104         orb->string_to_object (IOR);
 105
 106       // Try to narrow the object reference to a test reference.
 107       test_var test =
 108         test::_narrow (object.in ());
 109
 110       do_calls (test.in ());
 111
 112       test->shutdown ();
 113
 114       orb->destroy ();
 115     }
 116   catch (const CORBA::Exception& ex)
 117     {
 118       ex._tao_print_exception ("Exception caught in client");
 119       return -1;
 120     }
 121
 122   return 0;
 123 }