TAO/tests/MT_Client/client.cpp

   1 #include "testC.h"
   2 #include "tao/debug.h"
   3 #include "ace/Get_Opt.h"
   4 #include "ace/Task.h"
   5
   6 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
   7 int nthreads = 5;
   8 int niterations = 5;
   9 bool server_shutdown = false;
  10
  11 int
  12 parse_args (int argc, ACE_TCHAR *argv[])
  13 {
  14   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:n:i:x"));
  15   int c;
  16
  17   while ((c = get_opts ()) != -1)
  18     switch (c)
  19       {
  20       case 'k':
  21         ior = get_opts.opt_arg ();
  22         break;
  23       case 'n':
  24         nthreads = ACE_OS::atoi (get_opts.opt_arg ());
  25         break;
  26       case 'i':
  27         niterations = ACE_OS::atoi (get_opts.opt_arg ());
  28         break;
  29       case 'x':
  30         server_shutdown = true;
  31         break;
  32       case '?':
  33       default:
  34         ACE_ERROR_RETURN ((LM_ERROR,
  35                            "usage:  %s "
  36                            "-k <ior> "
  37                            "-n <nthreads> "
  38                            "-i <niterations> "
  39                            "\n",
  40                            argv [0]),
  41                           -1);
  42       }
  43   // Indicates successful parsing of the command line
  44   return 0;
  45 }
  46
  47 class Client : public ACE_Task_Base
  48 {
  49   // = TITLE
  50   //   Run the client thread
  51   //
  52   // = DESCRIPTION
  53   //   Use the ACE_Task_Base class to run the client threads.
  54   //
  55 public:
  56   Client (Simple_Server_ptr server, int niterations);
  57   // ctor
  58
  59   virtual int svc ();
  60   // The thread entry point.
  61
  62 private:
  63   void validate_connection ();
  64   // Validate the connection
  65
  66 private:
  67   Simple_Server_var server_;
  68   // The server.
  69
  70   int niterations_;
  71   // The number of iterations on each client thread.
  72 };
  73
  74 int
  75 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  76 {
  77   try
  78     {
  79       CORBA::ORB_var orb =
  80         CORBA::ORB_init (argc, argv);
  81
  82       if (parse_args (argc, argv) != 0)
  83         return 1;
  84
  85       CORBA::Object_var object =
  86         orb->string_to_object (ior);
  87
  88       Simple_Server_var server =
  89         Simple_Server::_narrow (object.in ());
  90
  91       if (CORBA::is_nil (server.in ()))
  92         {
  93           ACE_ERROR_RETURN ((LM_ERROR,
  94                              "Object reference <%s> is nil.\n",
  95                              ior),
  96                             1);
  97         }
  98
  99       Client client (server.in (), niterations);
 100       if (client.activate (THR_NEW_LWP | THR_JOINABLE,
 101                            nthreads) != 0)
 102         ACE_ERROR_RETURN ((LM_ERROR,
 103                            "Cannot activate client threads\n"),
 104                           1);
 105
 106       client.thr_mgr ()->wait ();
 107
 108       ACE_DEBUG ((LM_DEBUG, "threads finished\n"));
 109
 110       if (server_shutdown)
 111         {
 112           server->shutdown ();
 113         }
 114
 115       orb->destroy ();
 116     }
 117   catch (const CORBA::Exception& ex)
 118     {
 119       ex._tao_print_exception ("Caught exception:");
 120       return 1;
 121     }
 122
 123   return 0;
 124 }
 125
 126 // ****************************************************************
 127
 128 Client::Client (Simple_Server_ptr server,
 129                 int niterations)
 130   :  server_ (Simple_Server::_duplicate (server)),
 131      niterations_ (niterations)
 132 {
 133 }
 134
 135 void
 136 Client::validate_connection ()
 137 {
 138   // Ping the object 100 times, ignoring all exceptions.
 139   // It would be better to use validate_connection() but the test must
 140   // run on minimum CORBA builds too!
 141   for (int j = 0; j != 100; ++j)
 142     {
 143       try
 144         {
 145           this->server_->test_method ();
 146         }
 147       catch (const CORBA::Exception&){}
 148     }
 149 }
 150
 151 int
 152 Client::svc ()
 153 {
 154   try
 155     {
 156       this->validate_connection ();
 157
 158       for (int i = 0; i < this->niterations_; ++i)
 159         {
 160           this->server_->test_method ();
 161
 162           if (TAO_debug_level > 0 && i % 100 == 0)
 163             ACE_DEBUG ((LM_DEBUG, "(%P|%t) iteration = %d\n",
 164                         i));
 165         }
 166     }
 167   catch (const CORBA::Exception& ex)
 168     {
 169       ex._tao_print_exception ("MT_Client: exception raised");
 170     }
 171   return 0;
 172 }