TAO/tests/Forwarding/client.cpp

   1 #include "testC.h"
   2 #include "ace/Get_Opt.h"
   3 #include "ace/Task.h"
   4
   5 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
   6 int niterations = 10;
   7 int nthreads = 1;
   8 int do_shutdown = 0;
   9
  10 int
  11 parse_args (int argc, ACE_TCHAR *argv[])
  12 {
  13   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("xk:i:"));
  14   int c;
  15
  16   while ((c = get_opts ()) != -1)
  17     switch (c)
  18       {
  19       case 'x':
  20         do_shutdown = 1;
  21         break;
  22
  23       case 'k':
  24         ior = get_opts.opt_arg ();
  25         break;
  26
  27       case 'i':
  28         niterations = ACE_OS::atoi (get_opts.opt_arg ());
  29         break;
  30
  31       case '?':
  32       default:
  33         ACE_ERROR_RETURN ((LM_ERROR,
  34                            "usage:  %s "
  35                            "-k <ior> "
  36                            "-i <niterations> "
  37                            "\n",
  38                            argv [0]),
  39                           -1);
  40       }
  41   // Indicates successful parsing of the command line
  42   return 0;
  43 }
  44
  45 class Worker : public ACE_Task_Base
  46 {
  47 public:
  48   Worker (CORBA::ORB_ptr orb);
  49   // Constructor
  50
  51   virtual void run_test ();
  52   // The actual implementation of the test
  53
  54   // = The Task_Base methods
  55   virtual int svc ();
  56
  57 private:
  58   CORBA::ORB_var orb_;
  59   // The ORB reference
  60 };
  61
  62 int
  63 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  64 {
  65   try
  66     {
  67       CORBA::ORB_var orb =
  68         CORBA::ORB_init (argc, argv);
  69
  70       if (parse_args (argc, argv) != 0)
  71         return 1;
  72
  73       Worker worker (orb.in ());
  74
  75       if (worker.activate (THR_NEW_LWP | THR_JOINABLE,
  76                            nthreads) != 0)
  77         ACE_ERROR_RETURN ((LM_ERROR,
  78                            "(%P|%t) Cannot activate worker threads\n"),
  79                           1);
  80
  81       ACE_Time_Value tv (5, 0);
  82
  83       orb->run (tv);
  84
  85       // Now run a test in the main thread, just to confuse matters a
  86       // little more.
  87       worker.run_test ();
  88
  89       worker.thr_mgr ()->wait ();
  90
  91       if (do_shutdown)
  92         {
  93           CORBA::Object_var object =
  94             orb->string_to_object (ior);
  95
  96           Simple_Server_var server =
  97             Simple_Server::_narrow (object.in ());
  98
  99           server->shutdown ();
 100         }
 101
 102       orb->destroy ();
 103     }
 104   catch (const CORBA::Exception& ex)
 105     {
 106       ex._tao_print_exception ("Exception caught in main:");
 107       return 1;
 108     }
 109
 110   return 0;
 111 }
 112
 113 // ****************************************************************
 114
 115 Worker::Worker (CORBA::ORB_ptr orb)
 116   :  orb_ (CORBA::ORB::_duplicate (orb))
 117 {
 118 }
 119
 120 int
 121 Worker::svc ()
 122 {
 123   try
 124     {
 125       this->run_test ();
 126     }
 127   catch (const CORBA::Exception& ex)
 128     {
 129       ex._tao_print_exception ("Exception caught in thread (%t)\n");
 130     }
 131
 132   return 0;
 133 }
 134
 135 void
 136 Worker::run_test ()
 137 {
 138   for (int j = 0; j != niterations; ++j)
 139     {
 140       CORBA::Object_var object =
 141         this->orb_->string_to_object (ior);
 142
 143       CORBA::Boolean is_simple_server =
 144         object->_is_a ("IDL:Simple_Server:1.0");
 145       if (!is_simple_server)
 146         ACE_DEBUG ((LM_DEBUG,
 147                     "(%P|%t) unexpected result from _is_a()\n"));
 148     }
 149
 150   CORBA::Object_var object =
 151     this->orb_->string_to_object (ior);
 152
 153   Simple_Server_var server =
 154     Simple_Server::_narrow (object.in ());
 155
 156   if (CORBA::is_nil (server.in ()))
 157     {
 158       ACE_ERROR ((LM_ERROR,
 159                   "Object reference <%s> is nil.\n",
 160                   ior));
 161       return;
 162     }
 163
 164   for (int i = 0; i != niterations; ++i)
 165     {
 166       CORBA::Boolean r =
 167         server->test_is_a ("IDL:Foo:1.0");
 168
 169       if (r != 0)
 170         ACE_DEBUG ((LM_DEBUG,
 171                     "(%P|%t) unexpected result = %d\n",
 172                     r));
 173     }
 174 }