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

   1 // ******************************************************************
   2 // Include Section
   3 // ******************************************************************
   4
   5 #include "ace/Get_Opt.h"
   6 #include "ace/OS_NS_unistd.h"
   7
   8 #include "orbsvcs/CosNotifyCommC.h"
   9 #include "orbsvcs/CosNamingC.h"
  10 #include "orbsvcs/TimeBaseC.h"
  11 #include "Notify_Structured_Push_Consumer.h"
  12 #include "goC.h"
  13
  14 #include "Notify_Test_Client.h"
  15
  16 // ******************************************************************
  17 // Data Section
  18 // ******************************************************************
  19
  20 static const ACE_TCHAR *ior = ACE_TEXT ("file://supplier.ior");
  21 static unsigned int expected = 10;
  22 static Notify_Structured_Push_Consumer* consumer_1 = 0;
  23
  24 class Consumer_Client : public Notify_Test_Client
  25 {
  26 public:
  27   virtual int parse_args (int argc, ACE_TCHAR *argv[]);
  28 };
  29
  30
  31 int
  32 Consumer_Client::parse_args (int argc, ACE_TCHAR *argv[])
  33 {
  34   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:e:"));
  35   int c;
  36
  37   while ((c = get_opts ()) != -1)
  38     switch (c)
  39       {
  40         case 'k':
  41           ior = get_opts.optarg;
  42           break;
  43
  44         case 'e':
  45           expected = ACE_OS::atoi (get_opts.optarg);
  46           break;
  47
  48         default:
  49           ACE_ERROR_RETURN ((LM_ERROR,
  50             "usage:  %s "
  51             "-k <ior> "
  52             "-e <expected events> "
  53             "\n",
  54             argv [0]),
  55             -1);
  56       }
  57   // Indicates successful parsing of the command line
  58   return 0;
  59 }
  60
  61
  62 static CosNotifyChannelAdmin::ConsumerAdmin_ptr
  63 create_consumeradmin (CosNotifyChannelAdmin::EventChannel_ptr ec)
  64 {
  65   CosNotifyChannelAdmin::AdminID adminid = 0;
  66   CosNotifyChannelAdmin::ConsumerAdmin_var admin =
  67     ec->new_for_consumers (CosNotifyChannelAdmin::OR_OP, adminid);
  68
  69   return CosNotifyChannelAdmin::ConsumerAdmin::_duplicate (admin.in ());
  70 }
  71
  72
  73 static void
  74 create_consumers (CosNotifyChannelAdmin::ConsumerAdmin_ptr admin,
  75                   Notify_Test_Client* client)
  76 {
  77   // startup the consumer
  78   ACE_NEW_THROW_EX (consumer_1,
  79                     Notify_Structured_Push_Consumer ("consumer1",
  80                                                      expected,
  81                                                      *client),
  82                     CORBA::NO_MEMORY ());
  83   consumer_1->init (client->root_poa ());
  84   consumer_1->_connect (admin);
  85 }
  86
  87 // ******************************************************************
  88 // Main Section
  89 // ******************************************************************
  90
  91 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  92 {
  93   int status = 0;
  94   try
  95     {
  96       Consumer_Client client;
  97
  98       status = client.init (argc, argv);
  99       if (status != 0)
 100         ACE_ERROR_RETURN ((LM_ERROR, "Error: Client init failed.\n"),1);
 101
 102       CosNotifyChannelAdmin::EventChannel_var ec =
 103         client.create_event_channel ("MyEventChannel", 1);
 104
 105       CORBA::ORB_ptr orb = client.orb ();
 106       CORBA::Object_var object =
 107         orb->string_to_object (ior);
 108
 109       sig_var sig = sig::_narrow (object.in ());
 110
 111       if (CORBA::is_nil (sig.in ()))
 112         ACE_ERROR_RETURN ((LM_ERROR, "Error: Narrow failed.\n"),1);
 113
 114       CosNotifyChannelAdmin::ConsumerAdmin_var admin =
 115         create_consumeradmin (ec.in ());
 116
 117       if (CORBA::is_nil (admin.in ()))
 118         ACE_ERROR_RETURN ((LM_ERROR, "Error: nil ConsumerAdmin.\n"),1);
 119
 120       create_consumers (admin.in (), &client);
 121
 122       sig->go ();
 123
 124       ACE_DEBUG ((LM_DEBUG, "\nConsumer waiting for events...\n"));
 125
 126       client.ORB_run ();
 127
 128       ACE_DEBUG ((LM_DEBUG, "Consumer done.\n"));
 129       consumer_1->disconnect ();
 130
 131       sig->done ();
 132     }
 133   catch (const CORBA::Exception& e)
 134     {
 135       e._tao_print_exception ("Error: ");
 136       status = 1;
 137     }
 138
 139   return status;
 140 }