TAO/tests/Oneway_Buffering/Oneway_Buffering_Admin.cpp

   1 #include "Oneway_Buffering_Admin.h"
   2 #include "ace/OS_NS_unistd.h"
   3
   4 namespace
   5 {
   6   const unsigned int timeout_in_sec= 2u;
   7 }
   8
   9 Oneway_Buffering_Admin::Oneway_Buffering_Admin (CORBA::ORB_ptr orb)
  10   : orb_ (CORBA::ORB::_duplicate (orb))
  11   , request_count_ (0)
  12   , bytes_received_count_ (0)
  13 {
  14 }
  15
  16 CORBA::ULong
  17 Oneway_Buffering_Admin::request_count (CORBA::ULong expected_request_count)
  18 {
  19   ACE_Time_Value timeout (timeout_in_sec);
  20   while ((this->request_count_ < expected_request_count)
  21         && (ACE_Time_Value::zero < timeout) )
  22     {
  23       orb_->perform_work (timeout); // Do some work, decreases timeout for amount done.
  24     }
  25   return this->request_count_;
  26 }
  27
  28 CORBA::ULong
  29 Oneway_Buffering_Admin::bytes_received_count (CORBA::ULong expected_bytes_received_count)
  30 {
  31   ACE_Time_Value timeout (timeout_in_sec);
  32   while ((this->bytes_received_count_ < expected_bytes_received_count)
  33         && (ACE_Time_Value::zero < timeout) )
  34     {
  35       orb_->perform_work (timeout); // Do some work, decreases timeout for amount done.
  36     }
  37   return this->bytes_received_count_;
  38 }
  39
  40 void
  41 Oneway_Buffering_Admin::request_received (CORBA::ULong payload_length)
  42 {
  43   this->request_count_++;
  44   this->bytes_received_count_ += payload_length;
  45 }
  46
  47 void
  48 Oneway_Buffering_Admin::flush ()
  49 {
  50 }
  51
  52 void
  53 Oneway_Buffering_Admin::shutdown ()
  54 {
  55   this->orb_->shutdown (false);
  56 }