1 files changed, 89 insertions, 16 deletions

diff --git a/TAO/examples/RTCORBA/Activity/Thread_Task.cpp b/TAO/examples/RTCORBA/Activity/Thread_Task.cpp
index 49beaae62f5..167d1bca360 100644
--- a/TAO/examples/RTCORBA/Activity/Thread_Task.cpp
+++ b/TAO/examples/RTCORBA/Activity/Thread_Task.cpp
@@ -3,6 +3,7 @@
 
 #include "ace/High_Res_Timer.h"
 #include "tao/debug.h"
+#include "tao/ORB_Core.h"
 
 #include "Activity.h"
 #include "Task_Stats.h"
@@ -42,9 +43,71 @@ Thread_Task::activate_task (ACE_Barrier* barrier)
   return 0;
 }
 
+void
+Thread_Task::change_network_priority (void)
+{
+  ACE_DEBUG ((LM_DEBUG,
+	      "Change Network Priority\n"));
+
+  ACE_TRY_NEW_ENV
+    {
+
+      RTCORBA::RTORB_var rt_orb = ACTIVITY::instance ()->rt_orb ();
+
+      //  if (check_for_nil (rt_orb.in (), "RTORB") == -1)
+      //          ACE_ERROR ((LM_ERROR,
+      //                      "RTORB is nil\n"));
+      
+      //Set the tcp protocol protperties
+      RTCORBA::TCPProtocolProperties_var tcp_properties =
+        rt_orb->create_tcp_protocol_properties (ACE_DEFAULT_MAX_SOCKET_BUFSIZ,
+                                                ACE_DEFAULT_MAX_SOCKET_BUFSIZ,
+						1,
+                                                0,
+                                                1,
+						1
+                                                ACE_ENV_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      RTCORBA::ProtocolList protocols;
+      protocols.length (1);
+      protocols[0].protocol_type = 0;
+      protocols[0].transport_protocol_properties =
+        RTCORBA::ProtocolProperties::_duplicate (tcp_properties.in ());
+      protocols[0].orb_protocol_properties =
+        RTCORBA::ProtocolProperties::_nil ();
+
+      CORBA::PolicyList policy_list;
+      policy_list.length (1);
+      policy_list[0] =
+        rt_orb->create_client_protocol_policy (protocols
+                                               ACE_ENV_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+
+      ACTIVITY::instance ()->orb ()->orb_core ()->policy_current ().set_policy_overrides (policy_list,
+											  CORBA::SET_OVERRIDE
+											  ACE_ENV_ARG_PARAMETER);
+      
+      ACE_TRY_CHECK;
+    }
+  ACE_CATCHANY
+    {
+      ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION,
+                           "Caught exception: change_network_priority\n");
+    }
+  ACE_ENDTRY;
+}
+
 int
 Thread_Task::svc (void)
 {
+  ACE_DEBUG ((LM_DEBUG,
+	      "Thread_Task::svc\n"));
+
+  if (this->enable_network_priority_)
+    this->change_network_priority ();
+
   // if debugging, dump the priority that we're actually at.
   if (TAO_debug_level > 0)
     {
@@ -81,25 +144,35 @@ Thread_Task::svc (void)
 
   ACE_UINT32 gsf = ACE_High_Res_Timer::global_scale_factor ();
 
-  ACE_hrtime_t before, after;
+  ACE_hrtime_t before, after = 0, total_time = 0;
+
+  Job::data payload;
+  payload.length (1500);
+  
+  total_time = ACE_OS::gethrtime () + ACE_UINT64 (exec_time_ / 1000000 * gsf * ACE_HR_SCALE_CONVERSION);
 
-  for (int i = 0; i < iter_ ; ++i)
+  int i = 0;
+  while ( i < iter_ )
     {
+      i++;
+      
       before = ACE_OS::gethrtime ();
-
-      job_->work (load_);
-
+      
+      job_->work (load_,
+		  payload);
+      
       after = ACE_OS::gethrtime ();
 
-      task_stats_->sample (before, after);
-
+      if (after <= total_time || iter_ > 0)
+	task_stats_->sample (before, after);
+      
       if (period_ != 0) // blast mode, no sleep.
-        {
+	{
           // convert to microseconds
 #if !defined ACE_LACKS_LONGLONG_T
-
+	  
           ACE_UINT32 elapsed_microseconds = ACE_UINT32((after - before) / gsf);
-
+	  
 #else  /* ! ACE_LACKS_LONGLONG_T */
 
           ACE_UINT32 elapsed_microseconds = (after - before) / gsf;
@@ -110,22 +183,22 @@ Thread_Task::svc (void)
           elapsed_microseconds*=1000; // convert to uSec on Win32
 #endif /* ACE_WIN32 */
 
-      // did we miss any deadlines?
-
+	  // did we miss any deadlines?
+	  
           int missed =
             elapsed_microseconds > period_ ? elapsed_microseconds/period_ : 0;
-
+	  
           long sleep_time = (missed + 1)*period_ ;
           sleep_time -= elapsed_microseconds;
-
+	  
           if (TAO_debug_level > 0)
             ACE_DEBUG ((LM_DEBUG, "(%t) sleep time = %d\n", sleep_time));
-
+	  
           ACE_Time_Value t_sleep (0, sleep_time);
           ACE_OS::sleep (t_sleep);
         } /* period != 0 */
     } /* for */
-
+  
   task_stats_->end_time (ACE_OS::gethrtime ());
 
   job_->shutdown (); // tell the job that we're done.