1 files changed, 215 insertions, 0 deletions
diff --git a/TAO/tests/RTCORBA/Dynamic_Thread_Pool/client.cpp b/TAO/tests/RTCORBA/Dynamic_Thread_Pool/client.cpp
new file mode 100644
index 00000000000..bfe535ecc1e
--- /dev/null
+++ b/TAO/tests/RTCORBA/Dynamic_Thread_Pool/client.cpp
@@ -0,0 +1,215 @@
+// $Id$
+
+#include "ace/Get_Opt.h"
+#include "testC.h"
+#include "tao/RTCORBA/RTCORBA.h"
+#include "tao/ORB_Core.h"
+#include "ace/Task.h"
+#include "ace/OS_NS_unistd.h"
+ #include "../check_supported_priorities.cpp"
+
+ACE_RCSID(Thread_Pool, client, "$Id$")
+
+const char *ior = "file://ior_1";
+int iterations = 6;
+int shutdown_server = 0;
+bool decreased = false;
+
+int
+parse_args (int argc, char *argv[])
+{
+  ACE_Get_Opt get_opts (argc, argv, "xk:i:");
+  int c;
+
+  while ((c = get_opts ()) != -1)
+    switch (c)
+      {
+      case 'x':
+        shutdown_server = 1;
+        break;
+
+      case 'k':
+        ior = get_opts.opt_arg ();
+        break;
+
+      case 'i':
+        iterations = ACE_OS::atoi (get_opts.opt_arg ());
+        break;
+
+      case '?':
+      default:
+        ACE_ERROR_RETURN ((LM_ERROR,
+                           "usage:  %s "
+                           "-k <ior> "
+                           "-i <iterations> "
+                           "-x [shutdown server] "
+                           "\n",
+                           argv [0]),
+                          -1);
+      }
+
+  // Indicates sucessful parsing of the command line
+  return 0;
+}
+
+class Task : public ACE_Task_Base
+{
+public:
+
+  Task (ACE_Thread_Manager &thread_manager,
+        CORBA::ORB_ptr orb);
+
+  int svc (void);
+
+  CORBA::ORB_var orb_;
+
+};
+
+Task::Task (ACE_Thread_Manager &thread_manager,
+            CORBA::ORB_ptr orb)
+  : ACE_Task_Base (&thread_manager),
+    orb_ (CORBA::ORB::_duplicate (orb))
+{
+}
+
+int
+Task::svc (void)
+{
+  try
+    {
+      CORBA::Object_var object =
+        this->orb_->string_to_object (ior);
+
+      test_var test =
+        test::_narrow (object.in ());
+
+      pid_t pid =
+        ACE_OS::getpid ();
+
+      CORBA::Long tc = 0;
+
+      for (int i = 0; i != iterations; ++i)
+        {
+          CORBA::Long mtc = 0;
+          CORBA::Long r =
+            test->method (pid,
+                          i,
+                          mtc);
+
+          // Each 2 iterations sleep 5 seconds
+          if (i % 2 == 0)
+            ACE_OS::sleep (5);
+
+          ACE_ASSERT (r == i);
+          // Assert disappears on with optimizations on.
+          ACE_UNUSED_ARG (r);
+
+          if (mtc > tc)
+            {
+              // Number of threads increased, so store this.
+              ACE_DEBUG ((LM_DEBUG, "Thread count increased to %d\n", mtc));
+              tc = mtc;
+            }
+          else if (mtc < tc)
+            {
+              // Number of threads decreased!
+              ACE_DEBUG ((LM_DEBUG, "Thread count decreased to %d\n", mtc));
+              decreased = true;
+              tc = mtc;
+            }
+        }
+
+      ACE_OS::sleep (10);
+
+      CORBA::Long end = 0;
+      CORBA::Long re =
+        test->method (pid,
+                      0,
+                      end);
+
+      ACE_ASSERT (re == 0);
+      // Assert disappears on with optimizations on.
+      ACE_UNUSED_ARG (re);
+
+      if (end != 0)
+        {
+          ACE_ERROR ((LM_ERROR, "Dynamic thread count should be 0, not %d\n", end));
+        }
+
+      if (shutdown_server)
+        {
+          test->shutdown ();
+        }
+    }
+  catch (const CORBA::Exception& ex)
+    {
+      ex._tao_print_exception ("Exception caught:");
+      return -1;
+    }
+
+  return 0;
+}
+
+int
+main (int argc, char *argv[])
+{
+  try
+    {
+      CORBA::ORB_var orb =
+        CORBA::ORB_init (argc, argv);
+
+      int result =
+        parse_args (argc, argv);
+      if (result != 0)
+        return result;
+
+      // Thread Manager for managing task.
+      ACE_Thread_Manager thread_manager;
+
+      // Create task.
+      Task task (thread_manager,
+                 orb.in ());
+
+      // Task activation flags.
+      long flags =
+        THR_NEW_LWP |
+        THR_JOINABLE |
+        orb->orb_core ()->orb_params ()->thread_creation_flags ();
+
+      // Activate task.
+      result =
+        task.activate (flags);
+      if (result == -1)
+        {
+          if (errno == EPERM)
+            {
+              ACE_ERROR_RETURN ((LM_ERROR,
+                                 "Cannot create thread with scheduling policy %s\n"
+                                 "because the user does not have the appropriate privileges, terminating program....\n"
+                                 "Check svc.conf options and/or run as root\n",
+                                 sched_policy_name (orb->orb_core ()->orb_params ()->ace_sched_policy ())),
+                                2);
+            }
+          else
+            // Unexpected error.
+            ACE_ASSERT (0);
+        }
+
+      // Wait for task to exit.
+      result =
+        thread_manager.wait ();
+      ACE_ASSERT (result != -1);
+
+      if (decreased == false)
+        {
+          ACE_ERROR_RETURN ((LM_ERROR, "ERROR: Number of threads didn't decrease\n"), -1);
+        }
+    }
+  catch (const CORBA::Exception& ex)
+    {
+      ex._tao_print_exception ("Exception caught");
+      return -1;
+    }
+
+  return 0;
+}