TAO/orbsvcs/tests/Notify/Bug_1385_Regression/Structured_Supplier.cpp

   1 // ******************************************************************
   2 // Include Section
   3 // ******************************************************************
   4
   5 #include "ace/Auto_Ptr.h"
   6 #include "ace/Get_Opt.h"
   7 #include "ace/OS_NS_unistd.h"
   8
   9 #include "tao/ORB_Core.h"
  10
  11 #include "orbsvcs/CosNotifyChannelAdminS.h"
  12 #include "orbsvcs/CosNotifyCommC.h"
  13 #include "orbsvcs/CosNamingC.h"
  14 #include "orbsvcs/TimeBaseC.h"
  15
  16 #include "goS.h"
  17 #include "Notify_StructuredPushSupplier.h"
  18 #include "Notify_Test_Client.h"
  19
  20
  21 // ******************************************************************
  22 // Data Section
  23 // ******************************************************************
  24
  25 static TAO_Notify_Tests_StructuredPushSupplier* supplier_1 = 0;
  26 static int max_events = 10;
  27 static const ACE_TCHAR *ior_output_file = ACE_TEXT ("supplier.ior");
  28
  29 // ******************************************************************
  30 // Subroutine Section
  31 // ******************************************************************
  32 class sig_i : public POA_sig
  33 {
  34 public:
  35   sig_i (CORBA::ORB_ptr orb)
  36     : orb_ (orb),
  37       started_ (false)
  38   {
  39   }
  40
  41   void go (void)
  42   {
  43     started_ = true;
  44   }
  45
  46   void done (void)
  47   {
  48     started_ = false;
  49   }
  50
  51   void wait_for_startup ()
  52   {
  53     while (!started_)
  54       {
  55         ACE_Time_Value tv (0, 100 * 1000); // 100ms
  56         orb_->run (tv);
  57       }
  58   }
  59
  60   void wait_for_completion ()
  61   {
  62     while (started_)
  63       {
  64         ACE_Time_Value tv (0, 100 * 1000); // 100ms
  65         orb_->run (tv);
  66       }
  67   }
  68
  69 private:
  70   CORBA::ORB_ptr orb_;
  71   bool started_;
  72 };
  73
  74 class Supplier_Client : public Notify_Test_Client
  75 {
  76 public:
  77   virtual int parse_args (int argc, ACE_TCHAR *argv[]);
  78 };
  79
  80 int
  81 Supplier_Client::parse_args (int argc, ACE_TCHAR *argv[])
  82 {
  83   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:e:"));
  84   int c;
  85
  86   while ((c = get_opts ()) != -1)
  87     switch (c)
  88   {
  89     case 'e':
  90       max_events = ACE_OS::atoi (get_opts.optarg);
  91       break;
  92
  93     case 'o':
  94       ior_output_file = get_opts.optarg;
  95       break;
  96
  97     default:
  98       ACE_ERROR_RETURN ((LM_ERROR,
  99         "usage:  %s "
 100         "-o <iorfile> -e <# of events> -d"
 101         "\n",
 102         argv [0]),
 103         -1);
 104   }
 105
 106   // Indicates successful parsing of the command line
 107   return 0;
 108 }
 109
 110
 111 static CosNotifyChannelAdmin::SupplierAdmin_ptr
 112 create_supplieradmin (CosNotifyChannelAdmin::EventChannel_ptr ec)
 113 {
 114   CosNotifyChannelAdmin::AdminID adminid = 0;
 115   CosNotifyChannelAdmin::SupplierAdmin_var admin =
 116     ec->new_for_suppliers (CosNotifyChannelAdmin::AND_OP, adminid);
 117
 118   return CosNotifyChannelAdmin::SupplierAdmin::_duplicate (admin.in ());
 119 }
 120
 121
 122 static void
 123 SendEvent (int id)
 124 {
 125   CosNotification::StructuredEvent event;
 126
 127   event.header.fixed_header.event_type.domain_name = CORBA::string_dup ("DOC_TAO");
 128   event.header.fixed_header.event_type.type_name = CORBA::string_dup ("examples");
 129
 130   event.filterable_data.length (1);
 131   event.filterable_data[0].name = CORBA::string_dup ("id");
 132   event.filterable_data[0].value <<= id;
 133
 134   try
 135   {
 136     supplier_1->send_event (event);
 137   }
 138   catch (const CORBA::Exception& e)
 139   {
 140     e._tao_print_exception ("Error: ");
 141   }
 142 }
 143
 144 static void
 145 create_suppliers (CosNotifyChannelAdmin::SupplierAdmin_ptr admin,
 146                   PortableServer::POA_ptr poa)
 147 {
 148   // startup the supplier
 149   ACE_NEW_THROW_EX (supplier_1,
 150     TAO_Notify_Tests_StructuredPushSupplier (),
 151     CORBA::NO_MEMORY ());
 152
 153   supplier_1->init (poa);
 154   supplier_1->connect (admin);
 155
 156   CosNotification::EventTypeSeq added (1);
 157   CosNotification::EventTypeSeq removed (1);
 158   added.length (1);
 159   removed.length (1);
 160   added[0].domain_name = CORBA::string_dup ("DOC_TAO");
 161   added[0].type_name = CORBA::string_dup ("examples");
 162
 163   removed[0].domain_name = CORBA::string_dup ("*");
 164   removed[0].type_name = CORBA::string_dup ("*");
 165
 166   supplier_1->offer_change (added, removed);
 167 }
 168
 169
 170 // ******************************************************************
 171 // Main Section
 172 // ******************************************************************
 173
 174 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 175 {
 176   int status = 0;
 177   ACE_Auto_Ptr< sig_i > sig_impl;
 178   try
 179     {
 180       Supplier_Client client;
 181       status = client.init (argc, argv);
 182
 183       if (status == 0)
 184         {
 185           CosNotifyChannelAdmin::EventChannel_var ec =
 186             client.create_event_channel ("MyEventChannel", 0);
 187
 188           CORBA::ORB_ptr orb = client.orb ();
 189
 190           // Activate the signaler with the POA
 191           sig_impl.reset ( new sig_i ( orb ) );
 192           sig_var sig = sig_impl->_this ();
 193
 194           CORBA::String_var ior =
 195             orb->object_to_string (sig.in ());
 196
 197           // If the ior_output_file exists, output the ior to it
 198           if (ior_output_file != 0)
 199             {
 200               FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
 201               if (output_file == 0)
 202                 ACE_ERROR_RETURN ((LM_ERROR,
 203                                    "Cannot open output file %s for "
 204                                    "writing IOR: %C",
 205                                    ior_output_file,
 206                                    ior.in ()),
 207                                   1);
 208               ACE_OS::fprintf (output_file, "%s", ior.in ());
 209               ACE_OS::fclose (output_file);
 210             }
 211
 212           CosNotifyChannelAdmin::SupplierAdmin_var admin =
 213             create_supplieradmin (ec.in ());
 214           if (!CORBA::is_nil (admin.in ()))
 215             {
 216               create_suppliers (admin.in (), client.root_poa ());
 217
 218               sig_impl->wait_for_startup ();
 219
 220               ACE_DEBUG ((LM_DEBUG,
 221                           "1 supplier sending %d events...\n", max_events));
 222               for (int i = 0; i < max_events; ++i)
 223               {
 224                 ACE_DEBUG ((LM_DEBUG, "+"));
 225                 SendEvent (i);
 226               }
 227               ACE_DEBUG ((LM_DEBUG,
 228                           "\nSupplier sent %d events.\n", max_events));
 229
 230               sig_impl->wait_for_completion ();
 231
 232               ACE_OS::unlink (ior_output_file);
 233
 234               supplier_1->disconnect ();
 235
 236               ec->destroy ();
 237             }
 238         }
 239     }
 240   catch (const CORBA::Exception& e)
 241     {
 242       e._tao_print_exception ("Error: ");
 243       status = 1;
 244     }
 245
 246   return status;
 247 }