TAO/orbsvcs/tests/Notify/Discarding/Sequence_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 "Notify_Sequence_Push_Consumer.h"
  11 #include "goC.h"
  12
  13 #include "Notify_Test_Client.h"
  14
  15 // ******************************************************************
  16 // Data Section
  17 // ******************************************************************
  18
  19 static const ACE_TCHAR *ior = ACE_TEXT ("file://supplier.ior");
  20 static CORBA::Short discard_policy = CosNotification::AnyOrder;
  21 static int sent = 40;
  22 static Notify_Sequence_Push_Consumer* consumer_1 = 0;
  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:l:h:d:"));
  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 'd':
  49       {
  50         const ACE_TCHAR* discard = get_opts.optarg;
  51         if (ACE_OS::strcmp (discard, ACE_TEXT ("fifo")) == 0)
  52         {
  53           discard_policy = CosNotification::FifoOrder;
  54         }
  55         else if (ACE_OS::strcmp (discard, ACE_TEXT ("priority")) == 0)
  56         {
  57           discard_policy = CosNotification::PriorityOrder;
  58         }
  59         else if (ACE_OS::strcmp (discard, ACE_TEXT ("lifo")) == 0)
  60         {
  61           discard_policy = CosNotification::LifoOrder;
  62         }
  63         else if (ACE_OS::strcmp (discard, ACE_TEXT ("deadline")) == 0)
  64         {
  65           discard_policy = CosNotification::DeadlineOrder;
  66 #if !defined (ACE_HAS_TIMED_MESSAGE_BLOCKS)
  67           ACE_ERROR_RETURN ((LM_ERROR,
  68             "This order policy requires timed message "
  69             "blocks.\nPlease #define "
  70             "ACE_HAS_TIMED_MESSAGE_BLOCKS in your "
  71             "config.h\n"), -1);
  72 #endif
  73         }
  74         else
  75         {
  76           ACE_ERROR_RETURN ((LM_ERROR,
  77             "Unknown discard policy: %s\n",
  78             discard_policy),
  79             -1);
  80         }
  81         break;
  82       }
  83     default:
  84       ACE_ERROR_RETURN ((LM_ERROR,
  85         "usage:  %s "
  86         "-k <ior> "
  87         "-d <any|fifo|lifo|priority|deadline> "
  88         "\n",
  89         argv [0]),
  90         -1);
  91   }
  92   // Indicates successful parsing of the command line
  93   return 0;
  94 }
  95
  96
  97 static CosNotifyChannelAdmin::ConsumerAdmin_ptr
  98 create_consumeradmin (CosNotifyChannelAdmin::EventChannel_ptr ec)
  99 {
 100   CosNotifyChannelAdmin::AdminID adminid = 0;
 101   CosNotifyChannelAdmin::ConsumerAdmin_var admin =
 102     ec->new_for_consumers (CosNotifyChannelAdmin::OR_OP,
 103     adminid);
 104
 105
 106   return CosNotifyChannelAdmin::ConsumerAdmin::_duplicate (admin.in ());
 107 }
 108
 109
 110 static void
 111 create_consumers (CosNotifyChannelAdmin::ConsumerAdmin_ptr admin,
 112                   Notify_Test_Client* client)
 113 {
 114   // startup the consumer
 115   ACE_NEW_THROW_EX (consumer_1,
 116     Notify_Sequence_Push_Consumer ("consumer1", discard_policy, *client, sent),
 117     CORBA::NO_MEMORY ());
 118
 119   consumer_1->init (client->root_poa ());
 120
 121   consumer_1->_connect (admin);
 122 }
 123
 124 // ******************************************************************
 125 // Main Section
 126 // ******************************************************************
 127
 128 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 129 {
 130   int status = 0;
 131   try
 132   {
 133     Consumer_Client client;
 134
 135     status = client.init (argc, argv);
 136     ACE_UNUSED_ARG(status);
 137     ACE_ASSERT(status == 0);
 138
 139     CosNotifyChannelAdmin::EventChannel_var ec =
 140       client.create_event_channel ("MyEventChannel", 1);
 141
 142     CORBA::ORB_ptr orb = client.orb ();
 143     CORBA::Object_var object =
 144       orb->string_to_object (ior);
 145
 146     sig_var sig = sig::_narrow (object.in ());
 147     ACE_ASSERT(! CORBA::is_nil (sig.in ()));
 148
 149     CosNotifyChannelAdmin::ConsumerAdmin_var admin =
 150       create_consumeradmin (ec.in ());
 151     ACE_ASSERT(!CORBA::is_nil (admin.in ()));
 152
 153     create_consumers (admin.in (), &client);
 154
 155     // Tell the supplier to go
 156     sig->go ();
 157
 158     ACE_DEBUG((LM_DEBUG, "Consumer waiting for events...\n"));
 159
 160     client.ORB_run( );
 161
 162     ACE_DEBUG((LM_DEBUG, "Consumer done.\n"));
 163     consumer_1->disconnect();
 164
 165     sig->done ();
 166   }
 167   catch (const CORBA::Exception& e)
 168   {
 169     e._tao_print_exception ("Error: ");
 170     status = 1;
 171   }
 172
 173   return status;
 174 }