TAO/tests/Bug_3108_Regression/client.cpp

   1 #include "testC.h"
   2 #include "ace/Task.h"
   3 #include "ace/Get_Opt.h"
   4
   5 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
   6
   7 test_var global_client;
   8 int iterations = 100;
   9 bool server_shutdown = true;
  10 int n_tasks = 45;
  11
  12 int
  13 parse_args (int argc, ACE_TCHAR *argv[])
  14 {
  15   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:c:x"));
  16   int c;
  17
  18   while ((c = get_opts ()) != -1)
  19     switch (c)
  20       {
  21       case 'k':
  22         ior = get_opts.opt_arg ();
  23         break;
  24       case 'c':
  25         n_tasks = ACE_OS::atoi (get_opts.opt_arg ());
  26         break;
  27       case 'x':
  28         server_shutdown = false;
  29         break;
  30       case '?':
  31       default:
  32         ACE_ERROR_RETURN ((LM_ERROR,
  33                            "usage:  %s "
  34                            "-k <ior> "
  35                            "[-x] [-c concurrent_tasks]"
  36                            "\n",
  37                            argv [0]),
  38                           -1);
  39       }
  40   // Indicates successful parsing of the command line
  41   return 0;
  42 }
  43
  44
  45 class Client : public ACE_Task_Base
  46 {
  47 public:
  48   // ctor
  49   Client (test_ptr server);
  50
  51   virtual int svc ();
  52
  53   // The server.
  54   test_var server_;
  55 };
  56
  57 Client::Client (test_ptr server)
  58   :  server_ (test::_duplicate (server))
  59 {
  60 }
  61
  62 int
  63 Client::svc ()
  64 {
  65   try
  66     {
  67       this->server_->test_method();
  68     }
  69   catch (const CORBA::Exception& ex)
  70     {
  71       ex._tao_print_exception ("Client: exception raised");
  72     }
  73   return 0;
  74 }
  75
  76 int
  77 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  78 {
  79   try
  80     {
  81       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
  82
  83       if (parse_args (argc, argv) != 0)
  84         return 1;
  85
  86       CORBA::Object_var obj = orb->string_to_object(ior);
  87       test_var server = test::_narrow(obj.in ());
  88       if (CORBA::is_nil(server.in ()))
  89           {
  90             ACE_ERROR_RETURN ((LM_DEBUG,
  91                                "Nil reference <%s>\n",
  92                                ior),
  93                               1);
  94           }
  95
  96       Client client (server.in ());
  97       if (client.activate (THR_NEW_LWP | THR_JOINABLE,
  98                            n_tasks) != 0)
  99         ACE_ERROR_RETURN ((LM_ERROR,
 100                             "Cannot activate client threads\n"),
 101                           1);
 102
 103       client.thr_mgr ()->wait ();
 104
 105       if (server_shutdown)
 106         {
 107           server->shutdown ();
 108         }
 109
 110       orb->destroy ();
 111     }
 112   catch (const CORBA::Exception& ex)
 113     {
 114       ex._tao_print_exception ("Exception caught");
 115       return -1;
 116     }
 117
 118   return 0;
 119 }