TAO/tests/Multiple_Retry_Tests/Retry_On_Reply_Failure/client.cpp

   1 #include "testC.h"
   2 #include "ace/Get_Opt.h"
   3 #include "ace/Task.h"
   4 #include "ace/streams.h"
   5 #include "tao/Invocation_Utils.h"
   6 #include "ace/OS_NS_unistd.h"
   7
   8 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
   9 int nthreads = 1;
  10 static const ACE_TCHAR corbaloc_prefix[] = ACE_TEXT("corbaloc:");
  11 int expect_ex_kind = TAO::FOE_NON;
  12 int num_requests = 1;
  13
  14 int
  15 parse_args (int argc, ACE_TCHAR *argv[])
  16 {
  17   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:e:"));
  18   int c;
  19
  20   while ((c = get_opts ()) != -1)
  21     switch (c)
  22       {
  23       case 'k':
  24         ior = get_opts.opt_arg ();
  25         break;
  26
  27       case 'e':
  28         expect_ex_kind = 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                            "-e <expected exception kind> "
  37                            "\n",
  38                            argv [0]),
  39                           -1);
  40       }
  41
  42   if (ACE_OS::strncmp (ior,
  43                        corbaloc_prefix,
  44                        ACE_OS::strlen(corbaloc_prefix)) != 0)
  45     return 1;
  46
  47   // Indicates successful parsing of the command line
  48   return 0;
  49 }
  50
  51 class Worker : public ACE_Task_Base
  52 {
  53 public:
  54   Worker (CORBA::ORB_ptr orb);
  55   // Constructor
  56
  57   // = The Task_Base methods
  58   virtual int svc (void);
  59
  60   // Caught any exception ?
  61   int received_ex_kind () const;
  62
  63   // Return number of received exceptions.
  64   int num_received_ex () const;
  65
  66   // Indicate if the invocation completed.
  67   bool invocation_completed () const;
  68
  69   // Is test done ?
  70   void done ();
  71
  72 private:
  73
  74   // The ORB reference
  75   CORBA::ORB_var orb_;
  76   // The exceptions caught.
  77   int received_ex_kind_;
  78   // The number of received exceptions.
  79   int num_received_ex_;
  80   // Flag indicating that the invocation was completed.
  81   bool invocation_completed_;
  82   // Flag for test completion. The result is
  83   // collected before done.
  84   bool done_;
  85 };
  86
  87 int
  88 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  89 {
  90   try
  91     {
  92       CORBA::ORB_var orb =
  93         CORBA::ORB_init (argc, argv);
  94
  95       if (parse_args (argc, argv) != 0)
  96         return 1;
  97
  98       Worker worker (orb.in ());
  99
 100       if (worker.activate (THR_NEW_LWP | THR_JOINABLE,
 101                            nthreads) != 0)
 102         ACE_ERROR_RETURN ((LM_ERROR,
 103                            "(%P|%t) Cannot activate worker threads\n"),
 104                           1);
 105
 106       int timeout = 30;
 107       int now = 0;
 108       while (now < timeout &&
 109              ((expect_ex_kind == 0 && !worker.invocation_completed ()) ||
 110               (expect_ex_kind != 0 && expect_ex_kind != worker.received_ex_kind ()))
 111              )
 112         {
 113           std::cout << "." << std::flush;
 114           now += 1;
 115           ACE_Time_Value tv (1, 0);
 116           orb->run (tv);
 117         }
 118       ACE_ASSERT (now != 0);
 119
 120
 121       std::cout << std::endl;
 122
 123       worker.done ();
 124
 125       CORBA::Object_var object =
 126         orb->string_to_object (ior);
 127
 128       Simple_Server_var server =
 129         Simple_Server::_narrow (object.in ());
 130
 131       server->shutdown ();
 132
 133       ACE_OS::sleep (1);
 134
 135       orb->destroy ();
 136
 137       worker.thr_mgr ()->wait ();
 138
 139
 140       bool expect_no_ex =
 141         expect_ex_kind == TAO::FOE_NON && worker.num_received_ex () == 0 && worker.invocation_completed ();
 142       bool expect_ex_received =
 143         expect_ex_kind == worker.received_ex_kind () && worker.num_received_ex () > 0 && !worker.invocation_completed ();
 144       if (expect_no_ex || expect_ex_received)
 145         {
 146           ACE_DEBUG ((LM_DEBUG, "(%P|%t)client: test passed.\n"));
 147           return 0;
 148         }
 149       else
 150         {
 151           ACE_DEBUG ((LM_ERROR, "(%P|%t)client: test failed.\n"));
 152           return 1;
 153         }
 154     }
 155
 156   catch (const CORBA::Exception& ex)
 157     {
 158       ex._tao_print_exception ("Exception caught in main:");
 159       return 1;
 160     }
 161
 162   return 0;
 163 }
 164
 165 // ****************************************************************
 166
 167 Worker::Worker (CORBA::ORB_ptr orb)
 168      :    orb_ (CORBA::ORB::_duplicate (orb)),
 169           received_ex_kind_ (TAO::FOE_NON),
 170           num_received_ex_ (0),
 171           invocation_completed_ (false),
 172           done_ (false)
 173 {
 174 }
 175
 176 int
 177 Worker::svc (void)
 178 {
 179   try
 180     {
 181       CORBA::Object_var object =
 182         this->orb_->string_to_object (ior);
 183
 184       Simple_Server_var server =
 185         Simple_Server::_narrow (object.in ());
 186
 187       if (CORBA::is_nil (server.in ()))
 188         {
 189           ACE_ERROR ((LM_ERROR,
 190                       "Object reference <%s> is nil.\n",
 191                       ior));
 192           return 0;
 193         }
 194
 195       try {
 196         CORBA::Boolean r =
 197           server->test_is_a ("IDL:Foo:1.0");
 198
 199         this->invocation_completed_ = true;
 200
 201         if (r != 0)
 202           ACE_DEBUG ((LM_DEBUG,
 203                       "(%P|%t) unexpected result = %d\n",
 204                       r));
 205
 206       }
 207       catch (const CORBA::OBJECT_NOT_EXIST &)
 208         {
 209           ACE_DEBUG ((LM_DEBUG, "(%P|%t)received OBJECT_NOT_EXIST \n"));
 210           if (!this->done_)
 211             {
 212               ++ this->num_received_ex_;
 213               received_ex_kind_ |= TAO::FOE_OBJECT_NOT_EXIST;
 214             }
 215         }
 216       catch (const CORBA::COMM_FAILURE &)
 217         {
 218           ACE_DEBUG ((LM_DEBUG, "(%P|%t)received COMM_FAILURE \n"));
 219           if (!this->done_)
 220             {
 221               ++ this->num_received_ex_;
 222               received_ex_kind_ |= TAO::FOE_COMM_FAILURE;
 223             }
 224         }
 225       catch (const CORBA::TRANSIENT &)
 226         {
 227           ACE_DEBUG ((LM_DEBUG, "(%P|%t)received TRANSIENT \n"));
 228           if (!this->done_)
 229             {
 230               ++ this->num_received_ex_;
 231               received_ex_kind_ |= TAO::FOE_TRANSIENT;
 232             }
 233         }
 234       catch (const CORBA::INV_OBJREF &)
 235         {
 236           ACE_DEBUG ((LM_DEBUG, "(%P|%t)received INV_OBJREF \n"));
 237           if (!this->done_)
 238             {
 239               ++ this->num_received_ex_;
 240               received_ex_kind_ |= TAO::FOE_INV_OBJREF;
 241             }
 242         }
 243     }
 244   catch (const CORBA::Exception& ex)
 245     {
 246       ex._tao_print_exception ("Unexpected exception caught");
 247     }
 248
 249   return 0;
 250 }
 251
 252
 253 int
 254 Worker::received_ex_kind () const
 255 {
 256   return received_ex_kind_;
 257 }
 258
 259 int
 260 Worker::num_received_ex () const
 261 {
 262   return num_received_ex_;
 263 }
 264
 265 bool
 266 Worker::invocation_completed () const
 267 {
 268   return invocation_completed_;
 269 }
 270
 271 void
 272 Worker::done ()
 273 {
 274   done_ = true;
 275 }
 276