TAO/tao/Wait_On_Reactor.cpp

   1 #include "tao/Wait_On_Reactor.h"
   2 #include "tao/ORB_Core.h"
   3 #include "tao/Transport.h"
   4 #include "tao/Synch_Reply_Dispatcher.h"
   5 #include "tao/ORB_Time_Policy.h"
   6
   7 #include "ace/Reactor.h"
   8
   9 TAO_BEGIN_VERSIONED_NAMESPACE_DECL
  10
  11 TAO_Wait_On_Reactor::TAO_Wait_On_Reactor (TAO_Transport *transport)
  12   : TAO_Wait_Strategy (transport)
  13 {
  14 }
  15
  16 int
  17 TAO_Wait_On_Reactor::wait (ACE_Time_Value *max_wait_time,
  18                            TAO_Synch_Reply_Dispatcher &rd)
  19 {
  20   // Start the count down timer to account for the time spent in this
  21   // method.
  22   TAO::ORB_Countdown_Time countdown (max_wait_time);
  23
  24   // Reactor does not change inside the loop.
  25   ACE_Reactor *const reactor =
  26     this->transport_->orb_core ()->reactor ();
  27
  28   TAO_Leader_Follower &leader_follower =
  29     this->transport_->orb_core ()->leader_follower ();
  30
  31   // Do the event loop, till we fully receive a reply.
  32   int result = 0;
  33
  34   while (1)
  35     {
  36       // Run the event loop.
  37       result = reactor->handle_events (max_wait_time);
  38
  39       // If we got our reply, no need to run the event loop any
  40       // further.
  41       if (!rd.keep_waiting (leader_follower))
  42         {
  43           break;
  44         }
  45
  46       // Did we timeout? If so, stop running the loop.
  47       if (result == 0
  48           && max_wait_time != nullptr
  49           && *max_wait_time == ACE_Time_Value::zero)
  50         {
  51           break;
  52         }
  53
  54       // Other errors? If so, stop running the loop.
  55       if (result == -1)
  56         {
  57           break;
  58         }
  59
  60       // Otherwise, keep going...
  61     }
  62
  63   if (result == -1 || rd.error_detected (leader_follower))
  64     {
  65       return -1;
  66     }
  67
  68   // Return an error if there was a problem receiving the reply.
  69   if (max_wait_time != nullptr)
  70     {
  71       if (rd.successful (leader_follower) && *max_wait_time == ACE_Time_Value::zero)
  72         {
  73           result = -1;
  74           errno = ETIME;
  75         }
  76     }
  77   else
  78     {
  79       result = 0;
  80
  81       if (rd.error_detected (leader_follower))
  82         {
  83           result = -1;
  84         }
  85     }
  86
  87   return result;
  88 }
  89
  90 // Register the handler with the Reactor.
  91 int
  92 TAO_Wait_On_Reactor::register_handler ()
  93 {
  94   if (!this->is_registered_)
  95     {
  96       return this->transport_->register_handler ();
  97     }
  98
  99   return 1;
 100 }
 101
 102 bool
 103 TAO_Wait_On_Reactor::non_blocking () const
 104 {
 105   return true;
 106 }
 107
 108 bool
 109 TAO_Wait_On_Reactor::can_process_upcalls () const
 110 {
 111   return true;
 112 }
 113
 114 TAO_END_VERSIONED_NAMESPACE_DECL