TAO/tests/AMI_Timeouts/timeout_client.cpp

   1 //=============================================================================
   2 /**
   3  *  @file    timeout_client.cpp
   4  *
   5  *  Tests for proper handling of timeouts with AMI
   6  *
   7  *  @author Michael Kircher <Michael.Kircher@mchp.siemens.de>
   8  */
   9 //=============================================================================
  10
  11 #include "timeout_client.h"
  12 #include "tao/Messaging/Messaging.h"
  13 #include "tao/PolicyC.h"
  14 #include "tao/AnyTypeCode/Any.h"
  15 #include "ace/OS_NS_unistd.h"
  16
  17 TimeoutClient::TimeoutClient (CORBA::ORB_ptr orb,
  18                               TimeoutObj_ptr timeoutObject,
  19                               AMI_TimeoutObjHandler_ptr replyHandlerObject,
  20                               TimeoutHandler_i *timeoutHandler_i,
  21                               unsigned long timeToWait)
  22 : orb_ (CORBA::ORB::_duplicate (orb))
  23   , timeoutObject_ (TimeoutObj::_duplicate (timeoutObject))
  24   , replyHandlerObject_ (AMI_TimeoutObjHandler::_duplicate (replyHandlerObject))
  25   , timeoutHandler_i_ (timeoutHandler_i)
  26   , local_reply_excep_counter_ (0)
  27   , INVOKE_SYNCH (0)
  28   , INVOKE_ASYNCH (1)
  29   , timeToWait_ (timeToWait)
  30 {
  31 }
  32
  33 TimeoutClient::~TimeoutClient ()
  34 {
  35 }
  36
  37 int
  38 TimeoutClient::svc ()
  39 {
  40   this->initialize ();
  41
  42   try
  43     {
  44       // Tests timeouts for synchronous
  45       this->synch_test ();
  46
  47       // Tests AMI timeouts for influences on non-timeout calls
  48       this->none_test ();
  49
  50       // Tests AMI timeouts for accuracy
  51       this->accuracy_test ();
  52
  53       // Tests AMI timeouts for influences on non-timeout calls
  54       this->none_test ();
  55
  56       // shut down remote ORB
  57       timeoutObject_->shutdown ();
  58
  59       ACE_Time_Value tv (0, 20); // wait for the ORB to deliver the shutdown
  60       ACE_OS::sleep (tv);
  61
  62       // shut down local ORB
  63       orb_->shutdown (false);
  64     }
  65   catch (const CORBA::Exception& ex)
  66     {
  67       ex._tao_print_exception ("Caught exception:");
  68       return 1;
  69     }
  70
  71   ACE_DEBUG ((LM_DEBUG,
  72               "TimeoutClient::svc: Done\n\n"));
  73
  74   return 0;
  75 }
  76
  77 int
  78 TimeoutClient::initialize ()
  79 {
  80   try
  81     {
  82       CORBA::Object_var object =
  83         orb_->resolve_initial_references ("ORBPolicyManager");
  84
  85       policy_manager_ =
  86         CORBA::PolicyManager::_narrow (object.in ());
  87     }
  88   catch (const CORBA::Exception& ex)
  89     {
  90       ex._tao_print_exception ("Caught exception:");
  91       return 1;
  92     }
  93
  94   return 0;
  95 }
  96
  97
  98 void
  99 TimeoutClient::send (CORBA::Boolean async,
 100                      unsigned long local_timeout,
 101                      unsigned long remote_sleep)
 102 {
 103   ACE_DEBUG ((LM_DEBUG,
 104               "(%P|%t) Invoking "));
 105   if (async)
 106     ACE_DEBUG ((LM_DEBUG,
 107                 "asynch "));
 108   else
 109     ACE_DEBUG ((LM_DEBUG,
 110                 "synch "));
 111
 112   ACE_DEBUG ((LM_DEBUG,
 113               "local: %dms remote: %dms  ... ",
 114               local_timeout,
 115               remote_sleep));
 116
 117   CORBA::PolicyList policy_list (1);
 118
 119   try
 120     {
 121       if (local_timeout != 0)
 122         {
 123           TimeBase::TimeT timeout = 10000 * local_timeout;
 124
 125           CORBA::Any any_orb;
 126           any_orb <<= timeout;
 127
 128           policy_list.length (1);
 129           policy_list[0] =
 130             orb_->create_policy (Messaging::RELATIVE_RT_TIMEOUT_POLICY_TYPE,
 131                                  any_orb);
 132
 133           policy_manager_->set_policy_overrides (policy_list,
 134                                                  CORBA::SET_OVERRIDE);
 135         }
 136       else
 137         {
 138           policy_list.length (0);
 139           policy_manager_->set_policy_overrides (policy_list,
 140                                                  CORBA::SET_OVERRIDE);
 141         }
 142
 143
 144       // Start time measurement at the reply handler
 145       timeoutHandler_i_->start ();
 146
 147       if (async)
 148         {
 149           timeoutObject_->sendc_sendTimeToWait (replyHandlerObject_.in (),
 150                                                 remote_sleep);
 151         }
 152       else // synch
 153         {
 154           timeoutObject_->sendTimeToWait (remote_sleep);
 155         }
 156     }
 157   catch (const CORBA::TIMEOUT& )
 158     {
 159       local_reply_excep_counter_++;
 160
 161       // Trap this exception and continue...
 162       ACE_DEBUG ((LM_DEBUG,
 163                   " timeout "));
 164     }
 165
 166   // get rid of the policy, you created before.
 167   try
 168     {
 169       if (local_timeout != 0)
 170         {
 171           policy_list[0]->destroy ();
 172         }
 173     }
 174   catch (const CORBA::Exception&)
 175     {
 176       ACE_DEBUG ((LM_DEBUG,
 177                   "Error: Unexpected exception\n\n"));
 178     }
 179
 180   // wait for responses
 181   ACE_Time_Value tv (0, (local_timeout + remote_sleep)*2000 + 4000);
 182   ACE_OS::sleep (tv);
 183
 184   ACE_Time_Value &elapsed_time = timeoutHandler_i_->elapsed_time ();
 185
 186   if (async)
 187     ACE_DEBUG ((LM_DEBUG,
 188                 " .. needed %dms\n",
 189                 elapsed_time.msec()));
 190   else
 191     ACE_DEBUG ((LM_DEBUG,
 192                 " .. done\n"));
 193 }
 194
 195
 196 int
 197 TimeoutClient::synch_test ()
 198 {
 199   ACE_DEBUG ((LM_DEBUG,
 200               "TimeoutClient::synch_test - Begin\n"));
 201   timeoutHandler_i_->reset_reply_counter ();
 202   timeoutHandler_i_->reset_reply_excep_counter ();
 203   local_reply_excep_counter_ = 0;
 204
 205   this->send (INVOKE_SYNCH,
 206               0,    // local
 207               0);   // remote
 208
 209   this->send (INVOKE_SYNCH,
 210               timeToWait_,    // local
 211               timeToWait_*2); // remote
 212
 213   this->send (INVOKE_SYNCH,
 214               0,    // local
 215               0);   // remote
 216
 217   if (timeoutHandler_i_->reply_counter () != 0
 218    || timeoutHandler_i_->reply_excep_counter () != 0
 219    || local_reply_excep_counter_ != 1)
 220     ACE_DEBUG ((LM_DEBUG,
 221                 "**** Error in replies %d %d %d.\n\n",
 222                 timeoutHandler_i_->reply_counter (),
 223                 timeoutHandler_i_->reply_excep_counter (),
 224                 local_reply_excep_counter_));
 225
 226   ACE_DEBUG ((LM_DEBUG,
 227               "TimeoutClient::synch_test - End\n\n"));
 228
 229   return 0;
 230 }
 231
 232 int
 233 TimeoutClient::accuracy_test ()
 234 {
 235   ACE_DEBUG ((LM_DEBUG,
 236               "TimeoutClient::accuracy_test - Begin\n\n"));
 237
 238   timeoutHandler_i_->reset_reply_counter ();
 239   timeoutHandler_i_->reset_reply_excep_counter ();
 240   local_reply_excep_counter_ = 0;
 241
 242   this->send (INVOKE_ASYNCH,
 243               timeToWait_,
 244               (unsigned long)(timeToWait_*1.5));
 245
 246   this->send (INVOKE_ASYNCH,
 247               timeToWait_,
 248               (unsigned long)(timeToWait_*0.5));
 249
 250   if (timeoutHandler_i_->reply_counter () != 1
 251    || timeoutHandler_i_->reply_excep_counter () != 1
 252    || local_reply_excep_counter_ != 0)
 253     ACE_DEBUG ((LM_DEBUG,
 254                 "**** Error in replies %d %d %d.\n\n",
 255                 timeoutHandler_i_->reply_counter (),
 256                 timeoutHandler_i_->reply_excep_counter (),
 257                 local_reply_excep_counter_));
 258
 259   ACE_DEBUG ((LM_DEBUG,
 260               "TimeoutClient::accuracy_test - End\n\n"));
 261
 262   return 0;
 263 }
 264
 265 int
 266 TimeoutClient::none_test ()
 267 {
 268   ACE_DEBUG ((LM_DEBUG,
 269               "TimeoutClient::none_test - Begin\n\n"));
 270
 271   timeoutHandler_i_->reset_reply_counter ();
 272   timeoutHandler_i_->reset_reply_excep_counter ();
 273   local_reply_excep_counter_ = 0;
 274
 275   this->send (INVOKE_ASYNCH,
 276               0,
 277               0);
 278
 279   this->send (INVOKE_ASYNCH,
 280               timeToWait_,
 281               timeToWait_+10);  // trigger a timeout
 282
 283   this->send (INVOKE_ASYNCH,
 284               timeToWait_,
 285               0);
 286
 287   if (timeoutHandler_i_->reply_counter () != 2
 288    || timeoutHandler_i_->reply_excep_counter () != 1
 289    || local_reply_excep_counter_ != 0)
 290     ACE_DEBUG ((LM_DEBUG,
 291                 "**** Error in replies %d %d %d.\n\n",
 292                 timeoutHandler_i_->reply_counter (),
 293                 timeoutHandler_i_->reply_excep_counter (),
 294                 local_reply_excep_counter_));
 295
 296   ACE_DEBUG ((LM_DEBUG,
 297               "TimeoutClient::none_test - End\n\n"));
 298
 299   return 0;
 300 }