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

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