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

   1 // ******************************************************************
   2 // Include Section
   3 // ******************************************************************
   4
   5 #include "Notify_Push_Consumer.h"
   6 #include "goC.h"
   7 #include "Notify_Test_Client.h"
   8
   9 #include "ace/Get_Opt.h"
  10 #include "ace/Argv_Type_Converter.h"
  11
  12 // ******************************************************************
  13 // Data Section
  14 // ******************************************************************
  15
  16 static const ACE_TCHAR* ior = ACE_TEXT ("file://supplier.ior");
  17 static Notify_Sequence_Push_Consumer* consumer_1;
  18 static bool go = false;
  19
  20 // ******************************************************************
  21 // Subroutine Section
  22 // ******************************************************************
  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_Argv_Type_Converter argcon (argc, argv);
  35   ACE_Get_Opt get_opts (argcon.get_argc (),
  36                         argcon.get_TCHAR_argv (), ACE_TEXT ("gk:"));
  37   int c;
  38
  39   while ((c = get_opts ()) != -1)
  40     switch (c)
  41       {
  42       case 'g':
  43         go = true;
  44         break;
  45       case 'k':
  46         ior = get_opts.optarg;
  47         break;
  48
  49       default:
  50         ACE_ERROR_RETURN ((LM_ERROR,
  51                            "usage: %s "
  52                            "-g "
  53                            "-k <ior> "
  54                            "\n",
  55                            argv [0]),
  56                           -1);
  57       }
  58   // Indicates successful parsing of the command line
  59   return 0;
  60 }
  61
  62
  63 static CosNotifyChannelAdmin::ConsumerAdmin_ptr
  64 create_consumeradmin (CosNotifyChannelAdmin::EventChannel_ptr ec)
  65 {
  66   CosNotifyChannelAdmin::AdminID adminid = 0;
  67   CosNotifyChannelAdmin::ConsumerAdmin_var admin =
  68     ec->new_for_consumers (CosNotifyChannelAdmin::OR_OP,
  69                            adminid);
  70
  71
  72   return CosNotifyChannelAdmin::ConsumerAdmin::_duplicate (admin.in ());
  73 }
  74
  75
  76 static void
  77 create_consumers (CosNotifyChannelAdmin::ConsumerAdmin_ptr admin,
  78                   Notify_Test_Client* client)
  79 {
  80   // startup the consumer
  81   ACE_NEW_THROW_EX (consumer_1,
  82                     Notify_Sequence_Push_Consumer (*client),
  83                     CORBA::NO_MEMORY ());
  84   consumer_1->init (client->root_poa ());
  85
  86   consumer_1->_connect (admin);
  87 }
  88
  89 // ******************************************************************
  90 // Main Section
  91 // ******************************************************************
  92
  93 int
  94 ACE_TMAIN (int argc, ACE_TCHAR* argv[])
  95 {
  96   int status = 0;
  97   try
  98     {
  99       Consumer_Client client;
 100       ACE_Argv_Type_Converter argcon (argc, argv);
 101       status = client.init (argcon.get_argc (), argcon.get_TCHAR_argv ());
 102
 103       if (status == 0)
 104         {
 105           CosNotifyChannelAdmin::EventChannel_var ec =
 106             client.create_event_channel ("MyEventChannel", 1);
 107
 108           CORBA::ORB_ptr orb = client.orb ();
 109           CORBA::Object_var object =
 110                               orb->string_to_object (ior);
 111
 112           sig_var sig = sig::_narrow (object.in ());
 113
 114           if (CORBA::is_nil (sig.in ()))
 115             {
 116               ACE_ERROR_RETURN ((LM_ERROR,
 117                                  "Object reference <%s> is nil\n",
 118                                  ior),
 119                                 1);
 120             }
 121
 122           CosNotifyChannelAdmin::ConsumerAdmin_var admin =
 123             create_consumeradmin (ec.in ());
 124
 125           if (!CORBA::is_nil (admin.in ()))
 126             {
 127               create_consumers (admin.in (), &client);
 128               if (go)
 129                 {
 130                   // Tell the supplier to go
 131                   sig->go ();
 132                 }
 133
 134               ACE_DEBUG ((LM_DEBUG, "Consumer waiting for events...\n"));
 135
 136               client.ORB_run ();
 137
 138               ACE_DEBUG ((LM_DEBUG, "Consumer done.\n"));
 139               consumer_1->disconnect ();
 140
 141               if (go)
 142                 {
 143                   sig->done ();
 144                 }
 145             }
 146         }
 147     }
 148   catch (const CORBA::Exception& e)
 149     {
 150       e._tao_print_exception ("Error: ");
 151       status = 1;
 152     }
 153
 154   return status;
 155 }