TAO/tests/TransportCurrent/Framework/Current_Test_Impl.cpp

   1 // -*- C++ -*-
   2 #include "tao/TransportCurrent/TCC.h"
   3
   4 #include "Current_Test_Impl.h"
   5
   6 Current_Test_Impl::Current_Test_Impl (CORBA::ORB_ptr orb,
   7                                       PortableServer::POA_ptr poa,
   8                                       int do_collocated_calls)
   9   : orb_ (CORBA::ORB::_duplicate (orb))
  10   , poa_ (PortableServer::POA::_duplicate (poa))
  11   , do_collocated_calls_ (do_collocated_calls)
  12 {
  13 }
  14
  15 Current_Test_Impl::~Current_Test_Impl ()
  16 {
  17 }
  18
  19 ::CORBA::Long
  20 Current_Test_Impl::self_test ()
  21 {
  22   return 0;
  23 }
  24
  25 void
  26 Current_Test_Impl::test_transport_current ()
  27 {
  28   if (TAO_debug_level > 2)
  29     {
  30       ACE_DEBUG ((LM_DEBUG,
  31                   ACE_TEXT ("(%P|%t) server - Testing access to Transport Current ...\n")));
  32     }
  33
  34   CORBA::Object_var tcobject =
  35     orb_->resolve_initial_references ("TAO::Transport::Current");
  36
  37   TAO::Transport::Current_var tc =
  38     TAO::Transport::Current::_narrow (tcobject.in ());
  39
  40   if (TAO_debug_level >= 1)
  41     {
  42       ::CORBA::Long id = tc->id ();
  43
  44       ::TAO::CounterT bs = tc->bytes_sent ();
  45
  46       ::TAO::CounterT br = tc->bytes_received ();
  47
  48       ::TAO::CounterT rs = tc->messages_sent ();
  49
  50       ::TAO::CounterT rr = tc->messages_received ();
  51
  52       ACE_DEBUG ((LM_DEBUG,
  53                   ACE_TEXT ("Current_Test_Impl (%P|%t) Transport [%q] - Sent/Received [bytes=%q/%q, messages=%q/%q]\n"),
  54                   (ACE_UINT64)id,
  55                   (ACE_UINT64)bs,
  56                   (ACE_UINT64)br,
  57                   (ACE_UINT64)rs,
  58                   (ACE_UINT64)rr));
  59     }
  60
  61   if (TAO_debug_level > 2)
  62     {
  63       ACE_DEBUG ((LM_DEBUG,
  64                 ACE_TEXT ("(%P|%t) server - Successfully accessed Transport Current\n")));
  65     }
  66 }
  67
  68 void
  69 Current_Test_Impl::invoked_by_client ()
  70 {
  71   try
  72     {
  73       this->invoked_by_client_i ();
  74     }
  75   catch (const TAO::Transport::NoContext&)
  76     {
  77       ACE_ERROR ((LM_ERROR,
  78                   ACE_TEXT ("(%P|%t) server - Caught a Transport::NoContext exception\n")));
  79     }
  80   catch (const CORBA::SystemException&)
  81     {
  82       ACE_ERROR ((LM_ERROR,
  83                   ACE_TEXT ("(%P|%t) server - Caught a CORBA::SystemException exception\n")));
  84     }
  85 }
  86
  87 void
  88 Current_Test_Impl::invoked_by_client_i ()
  89 {
  90   if (TAO_debug_level > 2) {
  91     ACE_DEBUG ((LM_DEBUG,
  92                 ACE_TEXT ("(%P|%t) server - Testing inside an upcall\n")));
  93   }
  94
  95   this->test_transport_current ();
  96
  97   if (this->do_collocated_calls_)
  98     {
  99       CORBA::Object_var selfobject =
 100         poa_->servant_to_reference (this);
 101
 102       Test::Transport::CurrentTest_var self =
 103         Test::Transport::CurrentTest::_narrow (selfobject.in ());
 104
 105       if (TAO_debug_level >= 1)
 106         {
 107           ACE_DEBUG ((LM_DEBUG,
 108                       ACE_TEXT ("(%P|%t) server - Testing a collocated call\n")));
 109         }
 110       self->invoked_during_upcall ();
 111     }
 112   else
 113     {
 114       if (TAO_debug_level >= 1)
 115         {
 116           ACE_DEBUG ((LM_DEBUG,
 117                       ACE_TEXT ("(%P|%t) server - Skipping the collocated call test\n")));
 118         }
 119     }
 120 }
 121
 122 void
 123 Current_Test_Impl::invoked_during_upcall ()
 124 {
 125   if (TAO_debug_level > 2) {
 126     ACE_DEBUG ((LM_DEBUG,
 127                 ACE_TEXT ("(%P|%t) server - Test method invoked ")
 128                 ACE_TEXT ("by the server during upcall.\n")));
 129   }
 130
 131 }
 132
 133 void
 134 Current_Test_Impl::shutdown ()
 135 {
 136   if (TAO_debug_level > 2) {
 137     ACE_DEBUG ((LM_DEBUG,
 138                 ACE_TEXT ("(%P|%t) server - shutting down.\n")));
 139   }
 140
 141   this->orb_->shutdown (false);
 142 }