1 files changed, 152 insertions, 0 deletions
diff --git a/trunk/TAO/tests/RTCORBA/Priority_Inversion_With_Bands/test_i.cpp b/trunk/TAO/tests/RTCORBA/Priority_Inversion_With_Bands/test_i.cpp
new file mode 100644
index 00000000000..e4ae87b051b
--- /dev/null
+++ b/trunk/TAO/tests/RTCORBA/Priority_Inversion_With_Bands/test_i.cpp
@@ -0,0 +1,152 @@
+// $Id$
+
+#include "test_i.h"
+#include "tao/ORB_Core.h"
+#include "tao/RTCORBA/Thread_Pool.h"
+#include "ace/Countdown_Time.h"
+
+ACE_RCSID(Priority_Inversion_With_Bands, test_i, "$Id$")
+
+test_i::test_i (CORBA::ORB_ptr orb,
+                PortableServer::POA_ptr poa,
+                RTCORBA::Priority low_priority,
+                RTCORBA::Priority high_priority)
+  : orb_ (CORBA::ORB::_duplicate (orb)),
+    poa_ (PortableServer::POA::_duplicate (poa)),
+    low_priority_ (low_priority),
+    high_priority_ (high_priority),
+    rt_current_ (),
+    work_iterations_in_one_sec_ (0),
+    prime_number_ (9619),
+    current_iteration_ (0),
+    total_iterations_ (0)
+{
+  ACE_DECLARE_NEW_CORBA_ENV;
+
+  ACE_TRY
+    {
+
+      CORBA::Object_var object =
+        this->orb_->resolve_initial_references ("RTCurrent"
+                                                ACE_ENV_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      this->rt_current_ =
+        RTCORBA::Current::_narrow (object.in ()
+                                   ACE_ENV_ARG_PARAMETER);
+
+      this->work_iterations_in_one_sec_ =
+        this->estimate_iterations ();
+    }
+  ACE_CATCHANY
+    {
+      ACE_RE_THROW;
+    }
+  ACE_ENDTRY;
+}
+
+void
+test_i::method (CORBA::ULong work,
+                const char * iteration
+                ACE_ENV_ARG_DECL)
+  ACE_THROW_SPEC ((CORBA::SystemException))
+{
+  // Get the ORB_Core's TSS resources.
+  TAO_ORB_Core_TSS_Resources *tss =
+    this->orb_->orb_core ()->get_tss_resources ();
+
+  /// Get the lane attribute in TSS.
+  TAO_Thread_Lane *lane =
+    (TAO_Thread_Lane *) tss->lane_;
+
+  RTCORBA::Priority current_priority =
+    this->rt_current_->the_priority (ACE_ENV_SINGLE_ARG_PARAMETER);
+  ACE_CHECK;
+
+  const char *priority_string = 0;
+  if (current_priority == this->low_priority_)
+    priority_string = "low";
+  else if (current_priority == this->high_priority_)
+    priority_string = "high";
+  else
+    {
+      ACE_ERROR ((LM_ERROR, "ERROR: %d != %d and %d != %d\n",
+                            current_priority, this->low_priority_,
+                            current_priority, this->high_priority_));
+      ACE_ASSERT (0);
+    }
+
+
+  ACE_DEBUG ((LM_DEBUG,
+              "test_i::method - %s started at %T (%P|%t|%d|%d) for %d secs at priority %s\n",
+              iteration,
+              lane->pool ().id (),
+              lane->id (),
+              work,
+              priority_string));
+
+  this->work (work);
+
+  ACE_DEBUG ((LM_DEBUG,
+              "test_i::method - %s ended   at %T (%P|%t|%d|%d)\n",
+              iteration,
+              lane->pool ().id (),
+              lane->id ()));
+
+  ++this->current_iteration_;
+
+  if (this->current_iteration_ == this->total_iterations_)
+    {
+      this->orb_->shutdown (0
+                            ACE_ENV_ARG_PARAMETER);
+      ACE_CHECK;
+    }
+}
+
+PortableServer::POA_ptr
+test_i::_default_POA (ACE_ENV_SINGLE_ARG_DECL_NOT_USED)
+{
+  return PortableServer::POA::_duplicate (this->poa_.in ());
+}
+
+void
+test_i::initialize (CORBA::ULong total_iterations
+                    ACE_ENV_ARG_DECL_NOT_USED)
+  ACE_THROW_SPEC ((CORBA::SystemException))
+{
+  this->total_iterations_ = total_iterations;
+}
+
+int
+test_i::estimate_iterations (void)
+{
+  // Estimate for 2 seconds.
+  ACE_Time_Value work_time (2);
+  ACE_Countdown_Time countdown (&work_time);
+
+  int i = 0;
+  for (; work_time > ACE_Time_Value::zero; ++i)
+    {
+      ACE::is_prime (this->prime_number_,
+                     2,
+                     this->prime_number_ / 2);
+
+      countdown.update ();
+    }
+
+  return i / 2;
+}
+
+void
+test_i::work (CORBA::ULong work)
+{
+  int i =
+    this->work_iterations_in_one_sec_ * work;
+
+  for (; i > 0; --i)
+    {
+      ACE::is_prime (this->prime_number_,
+                     2,
+                     this->prime_number_ / 2);
+    }
+}