TAO/tao/DiffServPolicy/DiffServPolicy_ORBInitializer.cpp

   1 // -*- C++ -*-
   2 #include "tao/DiffServPolicy/DiffServPolicy_ORBInitializer.h"
   3 #include "tao/DiffServPolicy/DiffServPolicy.h"
   4 #include "tao/DiffServPolicy/DiffServPolicy_Factory.h"
   5 #include "tao/DiffServPolicy/DiffServ_Service_Context_Handler.h"
   6 #include "tao/DiffServPolicy/DiffServ_Protocols_Hooks.h"
   7 #include "tao/ORB_Core.h"
   8 #include "tao/PI/ORBInitInfo.h"
   9
  10 TAO_BEGIN_VERSIONED_NAMESPACE_DECL
  11
  12 void
  13 TAO_DiffServPolicy_ORBInitializer::pre_init (
  14     PortableInterceptor::ORBInitInfo_ptr info)
  15 {
  16   // Narrow to a TAO_ORBInitInfo object to get access to the
  17   // orb_core() TAO extension.
  18   TAO_ORBInitInfo_var tao_info = TAO_ORBInitInfo::_narrow (info);
  19
  20   if (CORBA::is_nil (tao_info.in ()))
  21     {
  22       if (TAO_debug_level > 0)
  23         TAOLIB_ERROR ((LM_ERROR,
  24                     "(%P|%t) TAO_RT_ORBInitializer::pre_init:\n"
  25                     "(%P|%t)    Unable to narrow "
  26                     "\"PortableInterceptor::ORBInitInfo_ptr\" to\n"
  27                     "(%P|%t)   \"TAO_ORBInitInfo *.\"\n"));
  28
  29       throw ::CORBA::INTERNAL ();
  30     }
  31
  32   // Bind the service context handler for Diffserv
  33   TAO_DiffServ_Service_Context_Handler* h = 0;
  34   ACE_NEW (h,
  35            TAO_DiffServ_Service_Context_Handler());
  36   tao_info->orb_core ()->service_context_registry ().bind (IOP::REP_NWPRIORITY, h);
  37 }
  38
  39 void
  40 TAO_DiffServPolicy_ORBInitializer::post_init (
  41     PortableInterceptor::ORBInitInfo_ptr info)
  42 {
  43   this->register_policy_factories (info);
  44 }
  45
  46 void
  47 TAO_DiffServPolicy_ORBInitializer::register_policy_factories (
  48   PortableInterceptor::ORBInitInfo_ptr info)
  49 {
  50   if (CORBA::is_nil (this->policy_factory_.in ()))
  51     {
  52       PortableInterceptor::PolicyFactory_ptr policy_factory;
  53       ACE_NEW_THROW_EX (policy_factory,
  54                         TAO_DiffServ_PolicyFactory,
  55                           CORBA::NO_MEMORY (
  56                             CORBA::SystemException::_tao_minor_code (
  57                               TAO::VMCID,
  58                               ENOMEM),
  59                             CORBA::COMPLETED_NO));
  60
  61       this->policy_factory_ = policy_factory;
  62     }
  63
  64   // Bind the same policy factory to all DiffServ related policy
  65   // types since a single policy factory is used to create each of
  66   // the different types of DiffServ policies.
  67   static CORBA::PolicyType const type[] = {
  68     TAO::CLIENT_NETWORK_PRIORITY_TYPE,
  69     TAO::NETWORK_PRIORITY_TYPE
  70   };
  71
  72   const CORBA::PolicyType *end =
  73     type + sizeof (type) / sizeof (type[0]);
  74
  75   for (CORBA::PolicyType const * i = type;
  76        i != end;
  77        ++i)
  78     {
  79       try
  80         {
  81           info->register_policy_factory (*i, this->policy_factory_.in ());
  82         }
  83       catch (const ::CORBA::BAD_INV_ORDER& ex)
  84         {
  85           if (ex.minor () == (CORBA::OMGVMCID | 16))
  86             {
  87               // The factory is already there, it happens because the
  88               // magic initializer in PortableServer.cpp registers
  89               // with the ORB multiple times.  This is an indication
  90               // that we should do no more work in this
  91               // ORBInitializer.
  92               return;
  93             }
  94           throw;
  95         }
  96       catch ( ::CORBA::Exception&)
  97         {
  98           // Rethrow any other exceptions...
  99           throw;
 100         }
 101     }
 102 }
 103
 104 TAO_END_VERSIONED_NAMESPACE_DECL