TAO/orbsvcs/tests/Notify/Bug_2561_Regression/Supplier.cpp

   1 // ******************************************************************
   2 // Include Section
   3 // ******************************************************************
   4
   5 #include "ace/Get_Opt.h"
   6 #include "ace/Auto_Ptr.h"
   7 #include "ace/Argv_Type_Converter.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 #include "Notify_SequencePushSupplier.h"
  16 #include "goS.h"
  17
  18 #include "Notify_Test_Client.h"
  19
  20
  21 // ******************************************************************
  22 // Data Section
  23 // ******************************************************************
  24
  25 static TAO_Notify_Tests_SequencePushSupplier* supplier_1 = 0;
  26 static const ACE_TCHAR* ior_output_file = ACE_TEXT ("supplier.ior");
  27
  28 // ******************************************************************
  29 // Subroutine Section
  30 // ******************************************************************
  31
  32 // The supplier will not start sending events until the
  33 // go() operation is invoked.
  34 class sig_i : public POA_sig
  35 {
  36 public:
  37   sig_i (CORBA::ORB_ptr orb)
  38     : orb_ (orb)
  39     , started_ (false)
  40   {
  41   }
  42
  43   void go (void)
  44   {
  45     started_ = true;
  46   }
  47
  48   void done (void)
  49   {
  50     started_ = false;
  51   }
  52
  53   void wait_for_startup ()
  54   {
  55     while (! started_)
  56       {
  57         ACE_Time_Value tv (0, 100 * 1000); // 100ms
  58         orb_->run (tv);
  59       }
  60   }
  61
  62   void wait_for_completion ()
  63   {
  64     while (started_)
  65       {
  66         ACE_Time_Value tv (0, 100 * 1000); // 100ms
  67         orb_->run (tv);
  68       }
  69   }
  70
  71 private:
  72   CORBA::ORB_ptr orb_;
  73   bool started_;
  74 };
  75
  76 class Supplier_Client : public Notify_Test_Client
  77 {
  78 public:
  79   virtual int parse_args (int argc, ACE_TCHAR* argv[]);
  80 };
  81
  82
  83 int
  84 Supplier_Client::parse_args (int argc, ACE_TCHAR *argv[])
  85 {
  86   ACE_Argv_Type_Converter argcon (argc, argv);
  87   ACE_Get_Opt get_opts (argcon.get_argc (),
  88                         argcon.get_TCHAR_argv (), ACE_TEXT ("o:"));
  89   int c;
  90
  91   while ((c = get_opts ()) != -1)
  92     switch (c)
  93   {
  94     case 'o':
  95       ior_output_file = get_opts.optarg;
  96       break;
  97
  98     default:
  99       ACE_ERROR_RETURN ((LM_ERROR,
 100         "usage: %s "
 101         "-o <iorfile> "
 102         "\n",
 103         argv [0]),
 104         -1);
 105   }
 106
 107   // Indicates successful parsing of the command line
 108   return 0;
 109 }
 110
 111
 112 static CosNotifyChannelAdmin::SupplierAdmin_ptr
 113 create_supplieradmin (CosNotifyChannelAdmin::EventChannel_ptr ec)
 114 {
 115   CosNotifyChannelAdmin::AdminID adminid = 0;
 116   CosNotifyChannelAdmin::SupplierAdmin_var admin =
 117     ec->new_for_suppliers (CosNotifyChannelAdmin::AND_OP,
 118     adminid);
 119
 120
 121   return CosNotifyChannelAdmin::SupplierAdmin::_duplicate (admin.in ());
 122 }
 123
 124
 125 static void
 126 create_suppliers (CosNotifyChannelAdmin::SupplierAdmin_ptr admin,
 127                   PortableServer::POA_ptr poa)
 128 {
 129   ACE_NEW_THROW_EX (supplier_1,
 130     TAO_Notify_Tests_SequencePushSupplier (),
 131     CORBA::NO_MEMORY ());
 132
 133   supplier_1->init (poa);
 134
 135   supplier_1->connect (admin);
 136 }
 137
 138
 139 // ******************************************************************
 140 // Main Section
 141 // ******************************************************************
 142
 143 int
 144 ACE_TMAIN (int argc, ACE_TCHAR* argv[])
 145 {
 146   int status = 0;
 147   ACE_Auto_Ptr<sig_i> sig_impl;
 148   try
 149     {
 150       Supplier_Client client;
 151       ACE_Argv_Type_Converter argcon (argc, argv);
 152       status = client.init (argcon.get_argc (), argcon.get_TCHAR_argv ());
 153
 154       if (status == 0)
 155         {
 156           CosNotifyChannelAdmin::EventChannel_var ec =
 157             client.create_event_channel ("MyEventChannel", 0);
 158
 159           sig_impl.reset (new sig_i (client.orb ()));
 160           sig_var sig = sig_impl->_this ();
 161
 162           // If the ior_output_file exists, output the ior to it
 163           if (ior_output_file != 0)
 164             {
 165               CORBA::String_var ior =
 166                 client.orb ()->object_to_string (sig.in ());
 167
 168               FILE *output_file= ACE_OS::fopen (ior_output_file, ACE_TEXT ("w"));
 169               ACE_ASSERT (output_file != 0);
 170               ACE_OS::fprintf (output_file, "%s", ior.in ());
 171               ACE_OS::fclose (output_file);
 172             }
 173
 174           CosNotifyChannelAdmin::SupplierAdmin_var admin =
 175             create_supplieradmin (ec.in ());
 176           ACE_ASSERT (!CORBA::is_nil (admin.in ()));
 177
 178           create_suppliers (admin.in (), client.root_poa ());
 179
 180           sig_impl->wait_for_startup ();
 181
 182           // If this is changed, you should update max_events
 183           // found in Notify_Push_Consumer.cpp
 184           int batch_size = 4;
 185           int num_events = 80 / batch_size;
 186           ACE_DEBUG ((LM_DEBUG,
 187                       "1 supplier sending %d batches of %d events...\n",
 188                       num_events, batch_size));
 189           CosNotification::EventBatch events;
 190           events.length(batch_size);
 191           for (int i = 0; i < num_events; ++i)
 192             {
 193               supplier_1->send_events (events);
 194             }
 195           ACE_DEBUG ((LM_DEBUG, "\nSupplier sent %d events.\n", num_events));
 196
 197           sig_impl->wait_for_completion ();
 198
 199           ACE_OS::unlink (ior_output_file);
 200
 201           supplier_1->disconnect ();
 202
 203           ec->destroy ();
 204         }
 205     }
 206   catch (const CORBA::Exception& e)
 207   {
 208     e._tao_print_exception ("Error: ");
 209     status = 1;
 210   }
 211
 212   return status;
 213 }