TAO/orbsvcs/tests/Notify/MC/Structured_Consumer.cpp

   1 // ******************************************************************
   2 // Include Section
   3 // ******************************************************************
   4
   5 #include "ace/Get_Opt.h"
   6 #include "ace/OS_NS_unistd.h"
   7 #include "ace/OS_NS_stdio.h"
   8
   9 #include "Notify_Structured_Push_Consumer.h"
  10 #include "MonitorTestInterfaceC.h"
  11
  12 #include "Notify_Test_Client.h"
  13
  14 // ******************************************************************
  15 // Data Section
  16 // ******************************************************************
  17
  18 static const ACE_TCHAR *ior = ACE_TEXT ("file://test_monitor.ior");
  19 static const ACE_TCHAR *ready_output_file = ACE_TEXT ("ready.txt");
  20 static unsigned int expected = 2000;
  21 static unsigned int delay_period = 5;
  22 static unsigned int delay_count = 0;
  23 static Notify_Structured_Push_Consumer* consumer_1 = 0;
  24
  25 class Consumer_Client : public Notify_Test_Client
  26 {
  27 public:
  28   virtual int parse_args (int argc, ACE_TCHAR *argv[]);
  29 };
  30
  31
  32 int
  33 Consumer_Client::parse_args (int argc, ACE_TCHAR *argv[])
  34 {
  35   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:e:p:d:o:"));
  36   int c;
  37
  38   while ((c = get_opts ()) != -1)
  39     switch (c)
  40       {
  41         case 'k':
  42           ior = get_opts.optarg;
  43           break;
  44
  45         case 'e':
  46           expected = ACE_OS::atoi (get_opts.optarg);
  47           break;
  48
  49         case 'p':
  50           delay_period = ACE_OS::atoi (get_opts.optarg);
  51           break;
  52
  53         case 'd':
  54           delay_count = ACE_OS::atoi (get_opts.optarg);
  55           break;
  56
  57         case 'o':
  58           ready_output_file = get_opts.optarg;
  59           break;
  60
  61         default:
  62           ACE_ERROR_RETURN ((LM_ERROR,
  63             "usage:  %s "
  64             "-k <ior> "
  65             "-e <expected events> "
  66             "-d <delay every 'n' seconds> "
  67             "-p <how many seconds to delay> "
  68             "-o <readyfile> -e <# of events> "
  69             "\n",
  70             argv [0]),
  71             -1);
  72       }
  73   // Indicates successful parsing of the command line
  74   return 0;
  75 }
  76
  77
  78 static CosNotifyChannelAdmin::ConsumerAdmin_ptr
  79 create_consumeradmin (CosNotifyChannelAdmin::EventChannel_ptr ec)
  80 {
  81   CosNotifyChannelAdmin::AdminID adminid = 0;
  82   CosNotifyChannelAdmin::ConsumerAdmin_var admin =
  83     ec->new_for_consumers (CosNotifyChannelAdmin::OR_OP, adminid);
  84
  85   return CosNotifyChannelAdmin::ConsumerAdmin::_duplicate (admin.in ());
  86 }
  87
  88
  89 static void
  90 create_consumers (CosNotifyChannelAdmin::ConsumerAdmin_ptr admin,
  91                   Notify_Test_Client* client)
  92 {
  93   // startup the consumer
  94   ACE_NEW_THROW_EX (consumer_1,
  95                     Notify_Structured_Push_Consumer ("consumer1",
  96                                                      expected,
  97                                                      *client),
  98                     CORBA::NO_MEMORY ());
  99   consumer_1->set_delay_parameters (delay_count, delay_period);
 100   consumer_1->init (client->root_poa ());
 101   consumer_1->_connect (admin);
 102 }
 103
 104 // ******************************************************************
 105 // Main Section
 106 // ******************************************************************
 107
 108 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 109 {
 110   int status = 0;
 111
 112   try
 113     {
 114       Consumer_Client client;
 115       status = client.init (argc, argv);
 116
 117       if (status != 0)
 118         {
 119           ACE_ERROR_RETURN ((LM_ERROR,
 120                              "Error: Client init failed.\n"),
 121                             1);
 122         }
 123
 124       CosNotifyChannelAdmin::EventChannel_var ec =
 125         client.create_event_channel ("MyEventChannel", 0);
 126
 127 #ifdef TEST_QOS_MAX_QUEUE_LENGTH
 128       ACE_OS::printf ("%s: setting max queue length to 1000\n", argv[0]);
 129       CosNotification::AdminProperties properties(1);
 130       properties.length(1);
 131       properties[0].name = CORBA::string_dup (CosNotification::MaxQueueLength);
 132       properties[0].value <<= 1000;
 133       ec->set_admin(properties);
 134 #endif //TEST_QOS_MAX_QUEUE_LENGTH
 135
 136       CORBA::ORB_ptr orb = client.orb ();
 137       CORBA::Object_var object =
 138         orb->string_to_object (ior);
 139
 140       MonitorTestInterface_var sig =
 141         MonitorTestInterface::_narrow (object.in ());
 142
 143       if (CORBA::is_nil (sig.in ()))
 144         {
 145           ACE_ERROR_RETURN ((LM_ERROR,
 146                              "Error: Narrow failed.\n"),
 147                             1);
 148         }
 149
 150       CosNotifyChannelAdmin::ConsumerAdmin_var admin =
 151         create_consumeradmin (ec.in ());
 152
 153       if (CORBA::is_nil (admin.in ()))
 154         {
 155           ACE_ERROR_RETURN ((LM_ERROR,
 156                              "Error: nil ConsumerAdmin.\n"),
 157                             1);
 158         }
 159
 160       create_consumers (admin.in (), &client);
 161
 162       sig->running (MonitorTestInterface::Consumer);
 163
 164       ACE_DEBUG ((LM_DEBUG, "\nConsumer waiting for events...\n"));
 165
 166       FILE* ready_file = ACE_OS::fopen (ACE_TEXT_ALWAYS_CHAR(ready_output_file), "w");
 167
 168       if (ready_file == 0)
 169         {
 170           ACE_ERROR_RETURN ((LM_ERROR,
 171                              "Cannot open ready file for writing\n"),
 172                             1);
 173         }
 174
 175       ACE_OS::fprintf (ready_file, "ready\n");
 176       ACE_OS::fclose (ready_file);
 177
 178       client.ORB_run ();
 179 #ifdef PAUSE_ON_EXIT
 180       _cputs( "All events received. Still connected.\n");
 181       _cputs( "Hit a key to exit consumer: " );
 182       _getch();
 183 #endif // PAUSE_ON_EXIT
 184       ACE_DEBUG ((LM_DEBUG, "Consumer done.\n"));
 185       consumer_1->disconnect ();
 186
 187       ec->destroy ();
 188
 189       sig->finished (MonitorTestInterface::Consumer);
 190     }
 191   catch (const CORBA::Exception& e)
 192     {
 193       e._tao_print_exception ("Consumer Error: ");
 194       status = 1;
 195     }
 196
 197   return status;
 198 }