1 files changed, 288 insertions, 0 deletions
diff --git a/TAO/orbsvcs/tests/Notify/Discarding/Sequence_Supplier.cpp b/TAO/orbsvcs/tests/Notify/Discarding/Sequence_Supplier.cpp
new file mode 100644
index 00000000000..95d08daca21
--- /dev/null
+++ b/TAO/orbsvcs/tests/Notify/Discarding/Sequence_Supplier.cpp
@@ -0,0 +1,288 @@
+// $Id$
+
+// ******************************************************************
+// Include Section
+// ******************************************************************
+
+#include "ace/Get_Opt.h"
+#include "ace/Auto_Ptr.h"
+
+#include "tao/ORB_Core.h"
+
+#include "orbsvcs/CosNotifyChannelAdminS.h"
+#include "orbsvcs/CosNotifyCommC.h"
+#include "orbsvcs/CosNamingC.h"
+#include "orbsvcs/TimeBaseC.h"
+#include "Notify_SequencePushSupplier.h"
+#include "goS.h"
+
+#include "Notify_Test_Client.h"
+
+#include "ace/OS_NS_unistd.h"
+
+// ******************************************************************
+// Data Section
+// ******************************************************************
+
+static TAO_Notify_Tests_SequencePushSupplier* supplier_1 = 0;
+
+static CORBA::Boolean use_deadline_ordering = 0;
+
+// Must be a multiple of the consumers batch size.
+static int num_events = 40;
+
+// Must match the consumers batch size so that we send
+// the correct number of each type of event per batch, which
+// allows us to validate the discard policy.
+static const CORBA::Long BATCH_SIZE = 4;
+
+static const char* ior_output_file = "supplier.ior";
+
+// ******************************************************************
+// Subroutine Section
+// ******************************************************************
+
+// The supplier will not start sending events until the
+// go() operation is invoked.
+class sig_i : public POA_sig
+{
+public:
+  sig_i(CORBA::ORB_ptr orb)
+    : orb_(orb)
+    , started_(false)
+  {
+  }
+
+  void go (ACE_ENV_SINGLE_ARG_DECL_NOT_USED)
+    ACE_THROW_SPEC ((CORBA::SystemException))
+  {
+    started_ = true;
+  }
+
+  void done (ACE_ENV_SINGLE_ARG_DECL_NOT_USED)
+    ACE_THROW_SPEC ((CORBA::SystemException))
+  {
+    started_ = false;
+  }
+
+  void wait_for_startup()
+  {
+    while (! started_) {
+      ACE_Time_Value tv(0, 100 * 1000); // 100ms
+      orb_->run(tv);
+    }
+  }
+
+  void wait_for_completion()
+  {
+    while (started_) {
+      ACE_Time_Value tv(0, 100 * 1000); // 100ms
+      orb_->run(tv);
+    }
+  }
+
+private:
+  CORBA::ORB_ptr orb_;
+  bool started_;
+};
+
+class Supplier_Client : public Notify_Test_Client
+{
+public:
+  virtual int parse_args (int argc, char* argv[]);
+};
+
+
+int
+Supplier_Client::parse_args (int argc, char *argv[])
+{
+  ACE_Get_Opt get_opts (argc, argv, "o:e:d");
+  int c;
+
+  while ((c = get_opts ()) != -1)
+    switch (c)
+  {
+    case 'd':
+      use_deadline_ordering = 1;
+#if !defined (ACE_HAS_TIMED_MESSAGE_BLOCKS)
+      ACE_ERROR_RETURN ((LM_ERROR,
+        "This order policy requires timed message "
+        "blocks.\nPlease #define "
+        "ACE_HAS_TIMED_MESSAGE_BLOCKS in your "
+        "config.h\n"), -1);
+#endif /* ACE_HAS_TIMED_MESSAGE_BLOCKS */
+      break;
+    case 'e':
+      num_events = ACE_OS::atoi (get_opts.optarg);
+      num_events -= num_events % BATCH_SIZE; // round down
+      break;
+
+    case 'o':
+      ior_output_file = get_opts.optarg;
+      break;
+
+    default:
+      ACE_ERROR_RETURN ((LM_ERROR,
+        "usage:  %s "
+        "-o <iorfile> -e <# of events> -d"
+        "\n",
+        argv [0]),
+        -1);
+  }
+
+  // Indicates sucessful parsing of the command line
+  return 0;
+}
+
+
+static CosNotifyChannelAdmin::SupplierAdmin_ptr
+create_supplieradmin (CosNotifyChannelAdmin::EventChannel_ptr ec
+                      ACE_ENV_ARG_DECL)
+{
+  CosNotifyChannelAdmin::AdminID adminid = 0;
+  CosNotifyChannelAdmin::SupplierAdmin_var admin =
+    ec->new_for_suppliers (CosNotifyChannelAdmin::AND_OP,
+    adminid
+    ACE_ENV_ARG_PARAMETER);
+
+  ACE_CHECK_RETURN (0);
+
+  return CosNotifyChannelAdmin::SupplierAdmin::_duplicate (admin.in ());
+}
+
+
+static void
+SendBatch (int batch_id ACE_ENV_ARG_DECL)
+{
+  CosNotification::EventBatch events;
+  events.length(BATCH_SIZE);
+  for (CORBA::Long i = 0; i < BATCH_SIZE; ++i)
+  {
+    int id = batch_id * BATCH_SIZE + i + 1;
+
+    CosNotification::StructuredEvent event;
+
+    event.header.fixed_header.event_type.domain_name = CORBA::string_dup ("a");
+    event.header.fixed_header.event_type.type_name = CORBA::string_dup ("b");
+    event.header.fixed_header.event_name = CORBA::string_dup ("c");
+
+    event.header.variable_header.length (3);
+    event.header.variable_header[0].name =
+      CORBA::string_dup ("Id");
+    event.header.variable_header[0].value <<= (CORBA::Long)id;
+
+    event.header.variable_header[1].name =
+      CORBA::string_dup (CosNotification::Priority);
+    event.header.variable_header[1].value <<= (CORBA::Short)(id);
+
+    event.header.variable_header[2].name =
+      CORBA::string_dup (CosNotification::Timeout);
+    event.header.variable_header[2].value <<= (TimeBase::TimeT) (id * 10000);
+
+    events[i] = event;
+  }
+  supplier_1->send_events (events ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK;
+}
+
+static void
+create_suppliers (CosNotifyChannelAdmin::SupplierAdmin_ptr admin,
+                  PortableServer::POA_ptr poa
+                  ACE_ENV_ARG_DECL)
+{
+  // start up the supplier
+  ACE_NEW_THROW_EX (supplier_1,
+    TAO_Notify_Tests_SequencePushSupplier (),
+    CORBA::NO_MEMORY ());
+
+  supplier_1->init (poa ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK;
+
+  supplier_1->connect (admin ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK;
+}
+
+
+// ******************************************************************
+// Main Section
+// ******************************************************************
+
+int main (int argc, char* argv[])
+{
+  int status = 0;
+  ACE_Auto_Ptr< sig_i > sig_impl;
+  ACE_TRY_NEW_ENV
+  {
+    Supplier_Client client;
+    status = client.init (argc, argv ACE_ENV_ARG_PARAMETER);
+    ACE_TRY_CHECK;
+
+    if (status == 0)
+    {
+      CosNotifyChannelAdmin::EventChannel_var ec =
+        client.create_event_channel ("MyEventChannel", 0 ACE_ENV_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      if (use_deadline_ordering)
+      {
+        CosNotification::QoSProperties qos (1);
+        qos.length (1);
+        qos[0].name = CORBA::string_dup (CosNotification::OrderPolicy);
+        qos[0].value <<= (CORBA::Short)CosNotification::DeadlineOrder;
+        ec->set_qos (qos);
+      }
+
+      sig_impl.reset( new sig_i( client.orb() ) );
+      sig_var sig = sig_impl->_this (ACE_ENV_SINGLE_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      CORBA::String_var ior =
+        client.orb()->object_to_string (sig.in () ACE_ENV_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      // If the ior_output_file exists, output the ior to it
+      if (ior_output_file != 0)
+      {
+        FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
+        ACE_ASSERT(output_file != 0);
+        ACE_OS::fprintf (output_file, "%s", ior.in ());
+        ACE_OS::fclose (output_file);
+      }
+
+      CosNotifyChannelAdmin::SupplierAdmin_var admin =
+        create_supplieradmin (ec.in () ACE_ENV_ARG_PARAMETER);
+      ACE_ASSERT(!CORBA::is_nil (admin.in ()));
+      create_suppliers (admin.in (), client.root_poa () ACE_ENV_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      sig_impl->wait_for_startup();
+
+      ACE_DEBUG((LM_DEBUG, "1 supplier sending %d events...\n", num_events));
+      for (int i = 0; i < num_events / BATCH_SIZE; ++i)
+      {
+        ACE_DEBUG((LM_DEBUG, "+"));
+        SendBatch (i ACE_ENV_ARG_PARAMETER);
+        ACE_TRY_CHECK;
+      }
+      ACE_DEBUG((LM_DEBUG, "\nSupplier sent %d events.\n", num_events));
+
+      sig_impl->wait_for_completion();
+
+      ACE_OS::unlink (ior_output_file);
+
+      supplier_1->disconnect(ACE_ENV_SINGLE_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      ec->destroy(ACE_ENV_SINGLE_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+    }
+  }
+  ACE_CATCH (CORBA::Exception, e)
+  {
+    ACE_PRINT_EXCEPTION (e, "Error: ");
+    status = 1;
+  }
+  ACE_ENDTRY;
+
+  return status;
+}