1 files changed, 243 insertions, 0 deletions
diff --git a/TAO/examples/CSD_Strategy/ThreadPool6/ServerApp.cpp b/TAO/examples/CSD_Strategy/ThreadPool6/ServerApp.cpp
new file mode 100644
index 00000000000..62de607a4fb
--- /dev/null
+++ b/TAO/examples/CSD_Strategy/ThreadPool6/ServerApp.cpp
@@ -0,0 +1,243 @@
+// $Id$
+#include "ServerApp.h"
+#include "Foo_i.h"
+#include "FooS_T.h"
+#include "OrbShutdownTask.h"
+#include "ace/Get_Opt.h"
+#include "tao/CSD_ThreadPool/CSD_TP_Strategy.h"
+#include "tao/Intrusive_Ref_Count_Handle_T.h"
+// To force static load the service.
+#include "tao/PI/PI.h"
+#include "tao/CSD_ThreadPool/CSD_ThreadPool.h"
+
+
+ServerApp::ServerApp()
+: ior_filename_ ("server.ior")
+{
+}
+
+
+ServerApp::~ServerApp()
+{
+}
+
+
+int
+ServerApp::run(int argc, char* argv[] ACE_ENV_ARG_DECL)
+{
+  CORBA::ORB_var orb = CORBA::ORB_init(argc, argv, "" ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  // Parse the command-line args for this application.
+  // * Raises -1 if problems are encountered.
+  // * Returns 1 if the usage statement was explicitly requested.
+  // * Returns 0 otherwise.
+  int result = this->parse_args(argc, argv); 
+  if (result != 0)
+    {
+      return result;
+    }
+
+  TheOrbShutdownTask::instance()->orb (orb.in ());
+
+  CORBA::Object_var obj 
+    = orb->resolve_initial_references("RootPOA" ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  if (CORBA::is_nil(obj.in()))
+    {
+      ACE_ERROR((LM_ERROR,
+                 "(%P|%t) Failed to resolve initial ref for 'RootPOA'.\n"));
+      ACE_THROW_RETURN (TestException(), -1);;
+    }
+
+  PortableServer::POA_var root_poa 
+    = PortableServer::POA::_narrow(obj.in() ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  if (CORBA::is_nil(root_poa.in()))
+    {
+      ACE_ERROR((LM_ERROR,
+                 "(%P|%t) Failed to narrow obj ref to POA interface.\n"));
+      ACE_THROW_RETURN (TestException(), -1);;
+    }
+
+  PortableServer::POAManager_var poa_manager 
+    = root_poa->the_POAManager(ACE_ENV_SINGLE_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  // Create the child POA.
+  CORBA::PolicyList policies(1);
+  policies.length(1);
+
+  policies[0] =
+    root_poa->create_implicit_activation_policy (PortableServer::IMPLICIT_ACTIVATION
+    ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  PortableServer::POA_var child_poa 
+    = root_poa->create_POA("ChildPoa",
+                           poa_manager.in(),
+                           policies 
+                           ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  if (CORBA::is_nil(child_poa.in()))
+    {
+      ACE_ERROR((LM_ERROR, "(%P|%t) ERROR [ServerApp::run()]: "
+                 "Failed to create the child POA.\n"));
+      ACE_THROW_RETURN (TestException(), -1);;
+    }
+
+  policies[0]->destroy (ACE_ENV_SINGLE_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  // Create the thread pool servant dispatching strategy object, and
+  // hold it in a (local) smart pointer variable.
+  TAO_Intrusive_Ref_Count_Handle<TAO::CSD::TP_Strategy> csd_tp_strategy =
+                                                 new TAO::CSD::TP_Strategy();
+
+  // Tell the strategy to apply itself to the child poa.
+  if (csd_tp_strategy->apply_to(child_poa.in() ACE_ENV_ARG_PARAMETER) == false)
+    {
+      ACE_ERROR((LM_ERROR, "(%P|%t) ERROR [ServerApp::run()]: "
+                 "Failed to apply custom dispatching strategy to child poa.\n"));
+      ACE_THROW_RETURN (TestException(), -1);;
+    }
+  ACE_CHECK_RETURN (-1);
+
+  // Create the Foo_i object.
+  Foo_i foo_i (this->num_clients_);
+
+  // Create tie object with the Foo_i object.
+  POA_Foo_tie<Foo_i> foo_tie_i (foo_i, child_poa.in ()); 
+
+  // Get Object Reference for the foo_tie_i object.
+  Foo_var foo = foo_tie_i._this (ACE_ENV_SINGLE_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  if (CORBA::is_nil(foo.in()))
+    {
+      ACE_ERROR((LM_ERROR,
+                 "(%P|%t) Failed to activate servant foo_tie_i.\n"));
+      ACE_THROW_RETURN (TestException(), -1);;
+    }
+
+  // Stringify the object reference
+  CORBA::String_var ior 
+    = orb->object_to_string(foo.in() ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  // Write the stringified object reference to the ior file.
+  FILE* ior_file = ACE_OS::fopen(this->ior_filename_.c_str(), "w");
+
+  if (ior_file == 0)
+    {
+      ACE_ERROR((LM_ERROR,
+                 "(%P|%t) Cannot open output file for writing IOR: %s",
+                 this->ior_filename_.c_str()));
+      ACE_THROW_RETURN (TestException(), -1);;
+    }
+
+  ACE_OS::fprintf(ior_file, "%s", ior.in ());
+  ACE_OS::fclose(ior_file);
+
+  // Activate the POA Manager
+  poa_manager->activate(ACE_ENV_SINGLE_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  ACE_DEBUG((LM_DEBUG,
+             "(%P|%t) ServerApp is ready.  Running the ORB event loop.\n"));
+
+  // Run the ORB event loop.
+  orb->run(ACE_ENV_SINGLE_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  ACE_DEBUG((LM_DEBUG,
+             "(%P|%t) ServerApp ORB has stopped running. "
+             "Stop the CSD strategy.\n"));
+
+  ACE_DEBUG((LM_DEBUG,
+             "(%P|%t) ServerApp is waiting for OrbShutdownTask.\n"));
+  TheOrbShutdownTask::instance()->wait ();
+
+  ACE_DEBUG((LM_DEBUG,
+             "(%P|%t) ServerApp is destroying the Root POA.\n"));
+
+  // Sleep for 2 second to let the done() two-way call complete
+  // before cleanup.
+  ACE_OS::sleep (2);
+
+  // Tear-down the root poa and orb.
+  root_poa->destroy(1, 1 ACE_ENV_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  ACE_DEBUG((LM_DEBUG,
+             "(%P|%t) ServerApp is destroying the ORB.\n"));
+
+  orb->destroy(ACE_ENV_SINGLE_ARG_PARAMETER);
+  ACE_CHECK_RETURN (-1);
+
+  ACE_DEBUG((LM_DEBUG,
+             "(%P|%t) ServerApp has completed running successfully.\n"));
+
+  return 0;
+}
+
+
+int
+ServerApp::parse_args(int argc, char* argv[])
+{
+  this->exe_name_ = argv[0];
+
+  ACE_Get_Opt get_opts(argc, argv, "o:n:");
+
+  int c;
+
+  while ((c = get_opts()) != -1)
+    {
+      switch (c)
+      {
+        case 'o':
+          this->ior_filename_ = get_opts.opt_arg();
+          break;
+
+        case 'n':
+          {
+            int tmp = ACE_OS::atoi(get_opts.opt_arg());
+            if (tmp < 1)
+              {
+                this->usage_statement ();
+                return -1;
+              }
+
+            this->num_clients_ = tmp;
+          }
+          break;
+
+        case '?':
+          this->usage_statement();
+          return 1;
+
+        default:
+          this->usage_statement();
+          return -1;
+      }
+    }
+
+  return 0;
+}
+
+
+void
+ServerApp::usage_statement()
+{
+  ACE_ERROR((LM_ERROR,
+             "Usage:  %s [options]\n\n"
+             "OPTIONS:\n\n"
+             "\t[-o <ior_filename>]\n"
+             "\t[-n <num_clients>]\n"
+             "\t[-?]\n\n",
+             this->exe_name_.c_str()));
+}
+