1 files changed, 93 insertions, 0 deletions
diff --git a/TAO/tao/RTPortableServer/RT_Collocation_Resolver.cpp b/TAO/tao/RTPortableServer/RT_Collocation_Resolver.cpp
new file mode 100644
index 00000000000..f394b54e0db
--- /dev/null
+++ b/TAO/tao/RTPortableServer/RT_Collocation_Resolver.cpp
@@ -0,0 +1,93 @@
+// $Id$
+
+#include "tao/RTPortableServer/RT_Collocation_Resolver.h"
+
+ACE_RCSID(RTCORBA, RT_Collocation_Resolver, "$Id$")
+
+#include "tao/ORB_Core.h"
+#include "tao/Object.h"
+#include "tao/Stub.h"
+#include "tao/PortableServer/Object_Adapter.h"
+#include "tao/PortableServer/POA.h"
+#include "tao/RTCORBA/Thread_Pool.h"
+
+#if !defined (__ACE_INLINE__)
+# include "tao/RTPortableServer/RT_Collocation_Resolver.i"
+#endif /* ! __ACE_INLINE__ */
+
+CORBA::Boolean
+TAO_RT_Collocation_Resolver::is_collocated (CORBA::Object_ptr object) const
+{
+  if (!object->_is_collocated ())
+    return 0;
+
+  ACE_DECLARE_NEW_CORBA_ENV;
+
+  TAO_ORB_Core *orb_core =
+    object->_stubobj ()->servant_orb_var ()->orb_core ();
+
+  TAO_Object_Adapter::Servant_Upcall servant_upcall (orb_core);
+  TAO_POA *poa =
+    servant_upcall.lookup_POA (object->_object_key (),
+                               ACE_TRY_ENV);
+  ACE_CHECK_RETURN (0);
+
+  TAO_Thread_Pool *target_thread_pool =
+    (TAO_Thread_Pool *) poa->thread_pool ();
+
+  /// Get the ORB_Core's TSS resources.
+  TAO_ORB_Core_TSS_Resources &tss =
+    *orb_core->get_tss_resources ();
+
+  // Get the lane for this thread.
+  TAO_Thread_Lane *current_thread_lane =
+    (TAO_Thread_Lane *) tss.lane_;
+
+  TAO_Thread_Pool *current_thread_pool = 0;
+
+  // If we don't have a lane, we don't have a pool.
+  if (current_thread_lane)
+    current_thread_pool =
+      &current_thread_lane->pool ();
+
+  if (current_thread_pool != target_thread_pool)
+    return 0;
+
+  if (current_thread_pool == 0)
+    return 1;
+
+  if (!current_thread_pool->with_lanes ())
+    return 1;
+
+  TAO_POA_Cached_Policies &cached_policies =
+    poa->cached_policies ();
+
+  TAO_POA_Cached_Policies::PriorityModel priority_model =
+    cached_policies.priority_model ();
+
+  if (priority_model == TAO_POA_Cached_Policies::CLIENT_PROPAGATED)
+    return 1;
+
+  poa->locate_servant_i ("operation not used",
+                         servant_upcall.system_id_,
+                         servant_upcall,
+                         servant_upcall.current_context_,
+                         ACE_TRY_ENV);
+  ACE_CHECK_RETURN (0);
+
+  CORBA::Short target_priority =
+    servant_upcall.priority ();
+
+  if (target_priority == current_thread_lane->lane_priority ())
+    return 1;
+  else
+    return 0;
+}
+
+ACE_STATIC_SVC_DEFINE (TAO_RT_Collocation_Resolver,
+                       ACE_TEXT ("RT_Collocation_Resolver"),
+                       ACE_SVC_OBJ_T,
+                       &ACE_SVC_NAME (TAO_RT_Collocation_Resolver),
+                       ACE_Service_Type::DELETE_THIS | ACE_Service_Type::DELETE_OBJ,
+                       0)
+ACE_FACTORY_DEFINE (TAO, TAO_RT_Collocation_Resolver)