TAO/orbsvcs/tests/Notify/lib/Notify_Test_Client.cpp

   1 #include "Notify_Test_Client.h"
   2 #define NOTIFY_FACTORY_NAME "NotifyEventChannelFactory"
   3 #define NAMING_SERVICE_NAME "NameService"
   4
   5 Notify_Test_Client::Notify_Test_Client ()
   6 : num_clients_( 0 )
   7 , done_( false )
   8 {
   9   // @@ Later: accept the inter filter operator as a param.
  10   ifgop_ = CosNotifyChannelAdmin::OR_OP;
  11 }
  12
  13 Notify_Test_Client::~Notify_Test_Client ()
  14 {
  15   if (!CORBA::is_nil (root_poa_.in ()) &&
  16       !CORBA::is_nil (orb_.in ()))
  17     {
  18       try
  19       {
  20         root_poa_->destroy(true, true);
  21         orb_->destroy();
  22       }
  23       catch (const CORBA::Exception& e)
  24       {
  25         e._tao_print_exception ("\nError: ");
  26       }
  27     }
  28 }
  29
  30 int
  31 Notify_Test_Client::init (int argc, ACE_TCHAR *argv[])
  32 {
  33   int status = this->init_ORB (argc, argv);
  34   if (status == 0)
  35     {
  36       this->resolve_naming_service ();
  37       this->resolve_Notify_factory ();
  38     }
  39   return status;
  40 }
  41
  42 int
  43 Notify_Test_Client::parse_args (int /*argc*/, ACE_TCHAR *[] /*argv*/)
  44 {
  45    return 0;
  46 }
  47
  48
  49 int
  50 Notify_Test_Client::init_ORB (int argc, ACE_TCHAR *argv[])
  51 {
  52   this->orb_ = CORBA::ORB_init (argc, argv);
  53
  54   if (this->parse_args (argc, argv) != 0)
  55     {
  56       return -1;
  57     }
  58
  59   CORBA::Object_var poa_object =
  60     this->orb_->resolve_initial_references("RootPOA");
  61
  62   if (CORBA::is_nil (poa_object.in ()))
  63     {
  64       ACE_ERROR ((LM_ERROR,
  65                   " (%P|%t) Unable to initialize the POA.\n"));
  66       return -1;
  67     }
  68   this->root_poa_ =
  69     PortableServer::POA::_narrow (poa_object.in ());
  70
  71   PortableServer::POAManager_var poa_manager =
  72     root_poa_->the_POAManager ();
  73
  74   poa_manager->activate ();
  75
  76   return 0;
  77 }
  78
  79 void
  80 Notify_Test_Client::resolve_naming_service ()
  81 {
  82   CORBA::Object_var naming_obj =
  83     this->orb_->resolve_initial_references (NAMING_SERVICE_NAME);
  84
  85   // Need to check return value for errors.
  86   if (CORBA::is_nil (naming_obj.in ()))
  87     throw CORBA::UNKNOWN ();
  88
  89   this->naming_context_ =
  90     CosNaming::NamingContext::_narrow (naming_obj.in ());
  91 }
  92
  93 void
  94 Notify_Test_Client::resolve_Notify_factory ()
  95 {
  96   CosNaming::Name name (1);
  97   name.length (1);
  98   name[0].id = CORBA::string_dup (NOTIFY_FACTORY_NAME);
  99
 100   CORBA::Object_var obj =
 101     this->naming_context_->resolve (name);
 102
 103   this->notify_factory_ =
 104     CosNotifyChannelAdmin::EventChannelFactory::_narrow (obj.in ());
 105 }
 106
 107 int
 108 Notify_Test_Client::ORB_run ()
 109 {
 110   while (! is_done())
 111   {
 112     ACE_Time_Value tv(0, 10 * 1000);
 113     orb_->run(tv);
 114   }
 115
 116   ACE_DEBUG((LM_DEBUG, "\nWaiting for stray events...\n"));
 117
 118   ACE_Time_Value tv(2);
 119   orb_->run(tv);
 120
 121   return 0;
 122 }
 123
 124 int
 125 Notify_Test_Client::ORB_run (ACE_Time_Value& tv)
 126 {
 127   orb_->run(tv);
 128
 129   ACE_DEBUG((LM_DEBUG, "\nWaiting for stray events...\n"));
 130
 131   ACE_Time_Value tv2(2,0);
 132   orb_->run(tv2);
 133
 134   return 0;
 135 }
 136
 137 void Notify_Test_Client::consumer_start (TAO_Notify_Tests_Peer*)
 138 {
 139   num_clients_++;
 140 }
 141
 142 void
 143 Notify_Test_Client::consumer_done (TAO_Notify_Tests_Peer*)
 144 {
 145   int value = --num_clients_;
 146   if ( value == 0 )
 147   {
 148     this->done_ = true;
 149   }
 150 }
 151
 152 bool
 153 Notify_Test_Client::is_done () const
 154 {
 155   return this->done_;
 156 }
 157
 158 CORBA::ORB_ptr
 159 Notify_Test_Client::orb ()
 160 {
 161   return this->orb_.in ();
 162 }
 163
 164
 165 PortableServer::POA_ptr
 166 Notify_Test_Client::root_poa ()
 167 {
 168   return this->root_poa_.in ();
 169 }
 170
 171 CosNaming::NamingContext_ptr
 172 Notify_Test_Client::naming_context ()
 173 {
 174   return this->naming_context_.in ();
 175 }
 176
 177 CosNotifyChannelAdmin::EventChannelFactory_ptr
 178 Notify_Test_Client::notify_factory ()
 179 {
 180   return this->notify_factory_.in ();
 181 }
 182
 183 CosNotifyChannelAdmin::EventChannel_ptr
 184 Notify_Test_Client::create_event_channel (const char* cname, bool resolve)
 185 {
 186   CosNotifyChannelAdmin::EventChannel_var ec;
 187   CosNaming::Name name (1);
 188   name.length (1);
 189   name[0].id = CORBA::string_dup (cname);
 190
 191   if (resolve)
 192     {
 193       CORBA::Object_var obj = naming_context_->resolve (name);
 194       ec = CosNotifyChannelAdmin::EventChannel::_narrow (obj.in ());
 195     }
 196   else
 197     {
 198       if (CORBA::is_nil (this->notify_factory_.in ()))
 199         {
 200           return CosNotifyChannelAdmin::EventChannel::_nil ();
 201         }
 202
 203       CosNotifyChannelAdmin::ChannelID id;
 204       CosNotification::QoSProperties initial_qos;
 205       CosNotification::AdminProperties initial_admin;
 206
 207       ec = notify_factory_->create_channel (initial_qos,
 208                                             initial_admin,
 209                                             id);
 210
 211       naming_context_->rebind(name, ec.in());
 212     }
 213
 214   return ec._retn ();
 215 }