1 files changed, 204 insertions, 0 deletions
diff --git a/TAO/orbsvcs/tests/Notify/performance-tests/Filter/Sequence_Consumer.cpp b/TAO/orbsvcs/tests/Notify/performance-tests/Filter/Sequence_Consumer.cpp
new file mode 100644
index 00000000000..6ed26aeaee9
--- /dev/null
+++ b/TAO/orbsvcs/tests/Notify/performance-tests/Filter/Sequence_Consumer.cpp
@@ -0,0 +1,204 @@
+// $Id$
+
+// ******************************************************************
+// Include Section
+// ******************************************************************
+
+#include "ace/Get_Opt.h"
+
+#include "orbsvcs/CosNotifyCommC.h"
+#include "orbsvcs/CosNamingC.h"
+#include "Notify_Sequence_Push_Consumer.h"
+#include "goC.h"
+#include "tao/debug.h"
+#include "ace/OS_NS_sys_time.h"
+
+#include "Notify_Test_Client.h"
+
+// ******************************************************************
+// Data Section
+// ******************************************************************
+
+static const char* ior = "file://supplier.ior";
+static CORBA::Boolean filter = 0;
+static unsigned int consumers = 2;
+static unsigned int batch_size = 10;
+static unsigned int expected = 100;  // 100 sets of 10
+
+#define GRAMMAR "TCL"
+
+// ******************************************************************
+// Subroutine Section
+// ******************************************************************
+
+class Consumer_Client : public Notify_Test_Client
+{
+public:
+  virtual int parse_args (int argc, char* argv[]);
+};
+
+
+int
+Consumer_Client::parse_args (int argc, char *argv[])
+{
+  ACE_Get_Opt get_opts (argc, argv, "k:c:e:f");
+  int c;
+
+  while ((c = get_opts ()) != -1)
+    switch (c)
+      {
+      case 'k':
+        ior = get_opts.optarg;
+        break;
+
+      case 'c':
+        consumers = ACE_OS::atoi (get_opts.optarg);
+        break;
+
+      case 'e':
+        expected = ACE_OS::atoi (get_opts.optarg);
+        break;
+
+      case 'f':
+        filter = 1;
+        break;
+
+      default:
+        ACE_ERROR_RETURN ((LM_ERROR,
+                           "usage:  %s "
+                           "-k <ior> "
+                           "-c <# of consumers> "
+                           "-e <expected events> "
+                           "-f "
+                           "\n",
+                           argv [0]),
+                          -1);
+      }
+  // Indicates sucessful parsing of the command line
+  return 0;
+}
+
+
+static CosNotifyChannelAdmin::ConsumerAdmin_ptr
+create_consumeradmin (CosNotifyChannelAdmin::EventChannel_ptr ec)
+{
+  CosNotifyChannelAdmin::AdminID adminid = 0;
+  CosNotifyChannelAdmin::ConsumerAdmin_var admin =
+    ec->new_for_consumers (CosNotifyChannelAdmin::AND_OP,
+                           adminid);
+
+
+  if (filter)
+    {
+      ACE_DEBUG((LM_DEBUG, "\nConsumer filter enabled.\n"));
+
+      CosNotifyFilter::FilterFactory_var ffact =
+          ec->default_filter_factory ();
+
+      CosNotifyFilter::Filter_var filter =
+          ffact->create_filter (GRAMMAR);
+
+      CosNotifyFilter::ConstraintExpSeq constraint_list (1);
+      constraint_list.length (1);
+
+      constraint_list[0].event_types.length (0);
+      constraint_list[0].constraint_expr =
+                                 CORBA::string_dup ("type == 'even'");
+
+      filter->add_constraints (constraint_list);
+
+      admin->add_filter (filter.in ());
+    }
+
+  return CosNotifyChannelAdmin::ConsumerAdmin::_duplicate (admin.in ());
+}
+
+
+static void
+create_consumers (CosNotifyChannelAdmin::ConsumerAdmin_ptr admin,
+                  Notify_Test_Client* client)
+{
+  for(unsigned int i = 0; i < consumers; i++)
+    {
+      // startup the first consumer
+      char name[64] = "";
+      ACE_OS::sprintf(name, "consumer%u", i);
+      Notify_Sequence_Push_Consumer* consumer;
+      ACE_NEW_THROW_EX (consumer,
+                        Notify_Sequence_Push_Consumer (
+                                         name,
+                                         expected * batch_size,
+                                         *client),
+                        CORBA::NO_MEMORY ());
+
+      consumer->init (client->root_poa ());
+
+      consumer->_connect (admin);
+    }
+}
+
+// ******************************************************************
+// Main Section
+// ******************************************************************
+
+int main (int argc, char* argv[])
+{
+  int status = 0;
+  try
+    {
+      Consumer_Client client;
+
+      status = client.init (argc, argv);
+
+      if (status == 0)
+        {
+          // If we are filtering, we will get half as many as originally
+          // expected
+          if (filter)
+            {
+              expected /= 2;
+            }
+
+          CosNotifyChannelAdmin::EventChannel_var ec =
+            client.create_event_channel ("MyEventChannel", 1);
+
+          CORBA::ORB_ptr orb = client.orb ();
+          CORBA::Object_var object =
+                              orb->string_to_object (ior);
+
+          sig_var sig = sig::_narrow (object.in ());
+
+          if (CORBA::is_nil (sig.in ()))
+            {
+              ACE_ERROR_RETURN ((LM_ERROR,
+                                 "Object reference <%s> is nil\n",
+                                 ior),
+                                1);
+            }
+
+          CosNotifyChannelAdmin::ConsumerAdmin_var admin =
+            create_consumeradmin (ec.in ());
+
+          if (!CORBA::is_nil (admin.in ()))
+            {
+              create_consumers(admin.in (), &client);
+
+              // Tell the supplier to go
+              sig->go ();
+
+              client.ORB_run();
+
+              ACE_DEBUG((LM_DEBUG, "Consumer done.\n"));
+
+              sig->done ();
+            }
+        }
+    }
+  catch (const CORBA::Exception& e)
+    {
+      e._tao_print_exception ("Error: Consumer exception: ");
+      status = 1;
+    }
+
+  return status;
+}