TAO/tests/No_Server_MT_Connect_Test/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 int server_shutdown = 0;
  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 = 1;
  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
  58
  59   /// The thread entry point.
  60   virtual int svc ();
  61
  62 private:
  63   Simple_Server_var server_;
  64   // The server.
  65
  66   int niterations_;
  67   // The number of iterations on each client thread.
  68 };
  69
  70 int
  71 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  72 {
  73   try
  74     {
  75       CORBA::ORB_var orb =
  76         CORBA::ORB_init (argc, argv);
  77
  78       if (parse_args (argc, argv) != 0)
  79         return 1;
  80
  81       CORBA::Object_var object =
  82         orb->string_to_object (ior);
  83
  84       Simple_Server_var server =
  85         Simple_Server::_narrow (object.in ());
  86
  87       if (CORBA::is_nil (server.in ()))
  88         {
  89           ACE_ERROR_RETURN ((LM_ERROR,
  90                              "Object reference <%s> is nil.\n",
  91                              ior),
  92                             1);
  93         }
  94
  95       if (server_shutdown)
  96         {
  97           server->shutdown ();
  98         }
  99
 100       Client client (server.in (), niterations);
 101       if (client.activate (THR_NEW_LWP | THR_JOINABLE,
 102                            nthreads) != 0)
 103         ACE_ERROR_RETURN ((LM_ERROR,
 104                            "Cannot activate client threads\n"),
 105                           1);
 106
 107       client.thr_mgr ()->wait ();
 108
 109       ACE_DEBUG ((LM_DEBUG, "threads finished\n"));
 110
 111       orb->destroy ();
 112     }
 113   catch (const CORBA::Exception& ex)
 114     {
 115       ex._tao_print_exception ("Caught exception:");
 116       return 1;
 117     }
 118
 119   return 0;
 120 }
 121
 122 // ****************************************************************
 123
 124 Client::Client (Simple_Server_ptr server,
 125                 int niterations)
 126   :  server_ (Simple_Server::_duplicate (server)),
 127      niterations_ (niterations)
 128 {
 129 }
 130
 131 int
 132 Client::svc ()
 133 {
 134   try
 135     {
 136       for (int i = 0; i < this->niterations_; ++i)
 137         {
 138           this->server_->test_method ();
 139
 140           if (TAO_debug_level > 0 && i % 100 == 0)
 141             ACE_DEBUG ((LM_DEBUG, "(%P|%t) iteration = %d\n",
 142                         i));
 143         }
 144     }
 145   catch (const CORBA::Exception&)
 146     {
 147       ACE_DEBUG ((LM_DEBUG,
 148                   "(%P|%t) Expected exceptions\n"));
 149     }
 150   return 0;
 151 }