TAO/orbsvcs/tests/Notify/Basic/LifeCycle.cpp

   1 #include "ace/Arg_Shifter.h"
   2 #include "ace/Get_Opt.h"
   3 #include "tao/debug.h"
   4 #include <orbsvcs/CosNamingC.h>
   5 #include <orbsvcs/CosNotifyCommC.h>
   6 #include <orbsvcs/CosNotifyChannelAdminC.h>
   7
   8 #include "LifeCycle.h"
   9
  10 LifeCycle::LifeCycle ()
  11   : count_ (10)
  12 {
  13 }
  14
  15 LifeCycle::~LifeCycle ()
  16 {
  17 }
  18
  19 int
  20 LifeCycle::parse_args (int argc,
  21                            ACE_TCHAR *argv[])
  22 {
  23     ACE_Arg_Shifter arg_shifter (argc,
  24                                  argv);
  25
  26     const ACE_TCHAR *current_arg = 0;
  27
  28     while (arg_shifter.is_anything_left ())
  29     {
  30       if (0 != (current_arg = arg_shifter.get_the_parameter (ACE_TEXT("-count"))))
  31         {
  32           this->count_ = ACE_OS::atoi (current_arg);
  33           // The number of times to create and destroy.
  34           arg_shifter.consume_arg ();
  35         }
  36       else if (arg_shifter.cur_arg_strncasecmp (ACE_TEXT("-?")) == 0)
  37         {
  38           ACE_DEBUG ((LM_DEBUG,
  39                       "usage: %s "
  40                       "-count testcount\n",
  41                       argv[0],
  42                       argv[0]));
  43
  44           arg_shifter.consume_arg ();
  45
  46           return -1;
  47         }
  48       else
  49         {
  50           arg_shifter.ignore_arg ();
  51         }
  52     }
  53
  54   return 0;
  55 }
  56
  57 void
  58 LifeCycle::init (int argc, ACE_TCHAR *argv[])
  59 {
  60   CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
  61
  62   CORBA::Object_var rootObj =
  63     orb->resolve_initial_references ("NameService");
  64
  65   if (CORBA::is_nil (rootObj.in ()))
  66     {
  67       ACE_ERROR ((LM_ERROR,
  68                   " (%P|%t) Unable to resolve naming service !\n"));
  69       return;
  70     }
  71
  72   CosNaming::NamingContext_var rootNC =
  73     CosNaming::NamingContext::_narrow (rootObj.in ());
  74
  75   CosNaming::Name name (1);
  76   name.length (1);
  77   name[0].id = CORBA::string_dup ("NotifyEventChannelFactory");
  78
  79   CORBA::Object_var obj = rootNC->resolve (name);
  80
  81   notify_factory_ =
  82     CosNotifyChannelAdmin::EventChannelFactory::_narrow (obj.in());
  83
  84   if (CORBA::is_nil (notify_factory_.in ()))
  85     {
  86       ACE_ERROR ((LM_ERROR,
  87                   " (%P|%t) Unable to locate Notify_Service\n"));
  88
  89       return;
  90     }
  91 }
  92
  93 void
  94 LifeCycle::run_test()
  95 {
  96   for (int i = 0; i < this->count_; ++i)
  97     {
  98      this->create_ec ();
  99
 100      this->create_supplier_admin ();
 101
 102      this->create_consumer_admin ();
 103
 104      this->destroy_consumer_admin ();
 105
 106      this->destroy_supplier_admin ();
 107
 108      this->destroy_ec ();
 109   }
 110 }
 111
 112 void
 113 LifeCycle::create_ec ()
 114 {
 115   CosNotifyChannelAdmin::ChannelID id;
 116   CosNotification::QoSProperties initial_qos;
 117   CosNotification::AdminProperties initial_admin;
 118
 119   this->ec_ = notify_factory_->create_channel (initial_qos,
 120                                                initial_admin,
 121                                                id);
 122
 123   if (CORBA::is_nil (ec_.in ())) {
 124     ACE_ERROR ((LM_ERROR,
 125                 " (%P|%t) Unable to create event channel\n"));
 126     return;
 127   }
 128
 129   if (TAO_debug_level)
 130     ACE_DEBUG ((LM_DEBUG,
 131                 "created event channel\n"));
 132 }
 133
 134 void
 135 LifeCycle::create_supplier_admin ()
 136 {
 137   CosNotifyChannelAdmin::AdminID adminid;
 138   CosNotifyChannelAdmin::InterFilterGroupOperator ifgop =
 139           CosNotifyChannelAdmin::OR_OP;
 140
 141   supplier_admin_ = this->ec_->new_for_suppliers (ifgop,
 142                                                   adminid);
 143
 144   if (CORBA::is_nil (supplier_admin_.in ()))
 145     {
 146       ACE_ERROR ((LM_ERROR,
 147                   " (%P|%t) Unable to create supplier admin\n"));
 148       return;
 149     }
 150
 151   if (TAO_debug_level)
 152     ACE_DEBUG ((LM_DEBUG,
 153                 "created supplier admin\n"));
 154 }
 155
 156 void
 157 LifeCycle::create_consumer_admin ()
 158 {
 159   CosNotifyChannelAdmin::AdminID adminid;
 160   CosNotifyChannelAdmin::InterFilterGroupOperator ifgop =
 161     CosNotifyChannelAdmin::OR_OP;
 162
 163   consumer_admin_ = ec_->new_for_consumers (ifgop, adminid);
 164
 165   if (CORBA::is_nil (consumer_admin_.in()))
 166     {
 167       ACE_ERROR ((LM_ERROR,
 168                          " (%P|%t) Unable to find supplier admin\n"));
 169       return;
 170     }
 171
 172   if (TAO_debug_level)
 173    ACE_DEBUG ((LM_DEBUG,
 174                "created consumer admin\n"));
 175 }
 176
 177 void
 178 LifeCycle::destroy_supplier_admin ()
 179 {
 180   this->supplier_admin_->destroy ();
 181
 182   if (TAO_debug_level)
 183     ACE_DEBUG ((LM_DEBUG,
 184                 "destroyed supplier admin\n"));
 185 }
 186
 187 void
 188 LifeCycle::destroy_consumer_admin ()
 189 {
 190   this->consumer_admin_->destroy ();
 191
 192   if (TAO_debug_level)
 193     ACE_DEBUG ((LM_DEBUG,
 194                 "destroyed consumer admin\n"));
 195 }
 196
 197 void
 198 LifeCycle::destroy_ec ()
 199 {
 200   this->ec_->destroy ();
 201
 202   if (TAO_debug_level)
 203     ACE_DEBUG ((LM_DEBUG,
 204                 "destroyed event channel\n"));
 205 }
 206
 207
 208 int
 209 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 210 {
 211   try
 212     {
 213       LifeCycle test;
 214
 215       test.parse_args (argc,
 216                        argv);
 217
 218       test.init (argc,
 219                  argv);
 220
 221       test.run_test ();
 222     }
 223   catch (const CORBA::Exception& ex)
 224     {
 225       ex._tao_print_exception ("Error: ");
 226       return 1;
 227     }
 228
 229   return 0;
 230 }