summaryrefslogtreecommitdiff
path: root/TAO/tests/CSD_Strategy_Tests/TP_Test_4/ServerApp.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'TAO/tests/CSD_Strategy_Tests/TP_Test_4/ServerApp.cpp')
-rw-r--r--TAO/tests/CSD_Strategy_Tests/TP_Test_4/ServerApp.cpp368
1 files changed, 368 insertions, 0 deletions
diff --git a/TAO/tests/CSD_Strategy_Tests/TP_Test_4/ServerApp.cpp b/TAO/tests/CSD_Strategy_Tests/TP_Test_4/ServerApp.cpp
new file mode 100644
index 00000000000..00e668c62bc
--- /dev/null
+++ b/TAO/tests/CSD_Strategy_Tests/TP_Test_4/ServerApp.cpp
@@ -0,0 +1,368 @@
+// $Id$
+#include "ServerApp.h"
+#include "AppHelper.h"
+#include "OrbRunner.h"
+#include "AppShutdown.h"
+#include "TestAppExceptionC.h"
+#include "Foo_C_Custom_ClientEngine.h"
+#include "Foo_C_ClientEngine.h"
+#include "Foo_C_Statistics.h"
+#include "ace/OS.h"
+#include "ace/Get_Opt.h"
+// To force static load the service.
+#include "tao/PI/PI.h"
+#include "tao/CSD_ThreadPool/CSD_ThreadPool.h"
+
+ServerApp::ServerApp()
+ : TestAppBase("TP_Test_4_Server"),
+ ior_filename_prefix_("foo"),
+ num_servants_(1),
+ num_csd_threads_(1),
+ num_orb_threads_(1),
+ num_remote_clients_(1),
+ num_collocated_clients_(0),
+ collocated_client_kind_(0)
+{
+}
+
+
+ServerApp::~ServerApp()
+{
+}
+
+
+int
+ServerApp::run_i(int argc, char* argv[])
+{
+ int result = this->init(argc, argv);
+ if (result != 0)
+ {
+ return result;
+ }
+
+ this->poa_setup();
+ this->csd_setup();
+ this->servant_setup();
+ this->collocated_setup();
+ this->poa_activate();
+ this->run_collocated_clients();
+ this->run_orb_event_loop();
+
+ // Calling wait on ACE_Thread_Manager singleton to avoid the problem
+ // that the main thread might exit before all CSD Threads exit.
+
+ // Wait for all CSD task threads exit.
+ ACE_Thread_Manager::instance ()->wait ();
+
+ this->cleanup();
+ return this->check_validity () ? 0 : -1;
+}
+
+
+int
+ServerApp::init(int argc, char* argv[])
+{
+ this->orb_ = CORBA::ORB_init(argc, argv, "");
+
+ // 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;
+ }
+
+ unsigned num_clients = this->num_remote_clients_ +
+ this->num_collocated_clients_;
+
+ TheAppShutdown->init(this->orb_.in(), num_clients);
+
+ return 0;
+}
+
+
+void
+ServerApp::poa_setup(void)
+{
+ this->poa_ = this->create_poa(this->orb_.in(),
+ "ChildPoa");
+}
+
+
+void
+ServerApp::csd_setup(void)
+{
+ this->tp_strategy_ = new TAO::CSD::TP_Strategy(this->num_csd_threads_);
+
+ if (!this->tp_strategy_->apply_to(this->poa_.in()))
+ {
+ ACE_ERROR((LM_ERROR,
+ "Failed to apply CSD strategy to poa.\n"));
+ throw TestAppException();
+ }
+}
+
+
+void
+ServerApp::servant_setup(void)
+{
+ this->servants_.create_and_activate(this->num_servants_,
+ this->orb_.in (),
+ this->poa_.in (),
+ this->ior_filename_prefix_.c_str());
+}
+
+
+void
+ServerApp::collocated_setup()
+{
+ int custom_client_id_start = this->num_remote_clients_;
+
+ unsigned servant_index = 0;
+
+ for (unsigned i = 0; i < this->num_collocated_clients_; i++)
+ {
+ if (i > 0)
+ {
+ // Dole out the servant object references in a round-robin fashion.
+ servant_index = (servant_index + 1) % this->num_servants_;
+ }
+
+ ServantListType::T_stub_var obj = this->servants_.objref(servant_index);
+
+ ClientEngine_Handle engine =
+ new Foo_C_Custom_ClientEngine(this->servants_.servant(servant_index),
+ obj.in(),
+ this->tp_strategy_.in(),
+ ++ custom_client_id_start);
+ this->collocated_client_task_.add_engine(engine.in());
+ }
+}
+
+
+void
+ServerApp::poa_activate(void)
+{
+ PortableServer::POAManager_var poa_manager
+ = this->poa_->the_POAManager();
+
+ poa_manager->activate();
+}
+
+
+void
+ServerApp::run_collocated_clients(void)
+{
+ if (this->num_collocated_clients_ > 0)
+ {
+ if (this->collocated_client_task_.open() == -1)
+ {
+ throw TestAppException ();
+ }
+ }
+}
+
+
+void
+ServerApp::run_orb_event_loop(void)
+{
+ OrbRunner orb_runner(this->orb_.in(), this->num_orb_threads_);
+ orb_runner.run();
+ TheAppShutdown->wait ();
+}
+
+
+void
+ServerApp::cleanup()
+{
+ for (unsigned i = 0; i < this->num_servants_; i++)
+ {
+ this->servants_.servant(i)->dump();
+ }
+
+ // Wait for all of the collocated client task threads to finish.
+ if (this->num_collocated_clients_ > 0)
+ {
+ this->collocated_client_task_.wait();
+ }
+}
+
+
+int
+ServerApp::parse_args(int argc, char* argv[])
+{
+ this->exe_name_ = argv[0];
+
+ ACE_Get_Opt get_opts(argc, argv, "p:s:n:t:r:c:k:");
+
+ int c;
+
+ while ((c = get_opts()) != -1)
+ {
+ int result = 0;
+ switch (c)
+ {
+ case 'p':
+ this->ior_filename_prefix_ = get_opts.opt_arg();
+ break;
+
+ case 's':
+ result = this->set_arg(this->num_servants_,
+ get_opts.opt_arg(),
+ c,
+ "num_servants",
+ 1);
+ break;
+
+ case 'n':
+ result = this->set_arg(this->num_csd_threads_,
+ get_opts.opt_arg(),
+ c,
+ "num_servants",
+ 1);
+ break;
+
+ case 't':
+ result = this->set_arg(this->num_orb_threads_,
+ get_opts.opt_arg(),
+ c,
+ "num_orb_threads",
+ 1);
+ break;
+
+ case 'r':
+ result = this->set_arg(this->num_remote_clients_,
+ get_opts.opt_arg(),
+ c,
+ "num_remote_clients");
+ break;
+
+ case 'c':
+ result = this->set_arg(this->num_collocated_clients_,
+ get_opts.opt_arg(),
+ c,
+ "num_collocated_clients");
+ break;
+
+ case 'k':
+ result = this->set_arg(this->collocated_client_kind_,
+ get_opts.opt_arg(),
+ c,
+ "collocated_client_kind");
+ break;
+
+ case '?':
+ this->usage_statement();
+ return 1;
+
+ default:
+ this->usage_statement();
+ return -1;
+ }
+
+ if (result != 0)
+ {
+ return result;
+ }
+ }
+
+ return this->arg_dependency_checks();
+}
+
+void
+ServerApp::usage_statement()
+{
+ ACE_ERROR((LM_ERROR,
+ "Usage: %s [options]\n\n"
+ "OPTIONS:\n\n"
+ "\t[-p <ior_filename_prefix>]\n"
+ "\t[-s <num_servants>]\n"
+ "\t[-n <num_csd_threads>]\n"
+ "\t[-t <num_orb_threads>]\n"
+ "\t[-r <num_remote_clients>]\n"
+ "\t[-c <num_collocated_clients>]\n"
+ "\t[-k <collocated_client_kind>]\n"
+ "\t[-?]\n\n",
+ this->exe_name_.c_str()));
+}
+
+
+int
+ServerApp::arg_dependency_checks()
+{
+ return (this->num_remote_clients_
+ + this->num_collocated_clients_) > 0 ? 0 : -1;
+}
+
+
+int
+ServerApp::set_arg(unsigned& value,
+ const char* arg,
+ char opt,
+ const char* name,
+ int min)
+{
+ int tmp = ACE_OS::atoi(arg);
+
+ if (tmp < min)
+ {
+ ACE_ERROR((LM_ERROR,
+ "Error: -%c <%s> must be integer type with a value of, "
+ "at least, %d.\n", opt, name, min));
+ this->usage_statement();
+ return -1;
+ }
+
+ value = tmp;
+
+ return 0;
+}
+
+
+PortableServer::POA_ptr
+ServerApp::create_poa(CORBA::ORB_ptr orb,
+ const char* poa_name)
+{
+ // Get the Root POA.
+ PortableServer::POA_var root_poa
+ = RefHelper<PortableServer::POA>::resolve_initial_ref(orb,
+ "RootPOA");
+
+ // Get the POAManager from the Root POA.
+ PortableServer::POAManager_var poa_manager
+ = root_poa->the_POAManager();
+
+ // Create the child POA Policies.
+ CORBA::PolicyList policies(0);
+ policies.length(0);
+
+ // Create the child POA
+ PortableServer::POA_var poa
+ = AppHelper::create_poa(poa_name,
+ root_poa.in(),
+ poa_manager.in(),
+ policies);
+
+ // Give away the child POA_ptr from the POA_var variable.
+ return poa._retn();
+}
+
+
+bool
+ServerApp::check_validity ()
+{
+ Foo_C_Statistics stats (this->num_remote_clients_,
+ this->num_collocated_clients_);
+
+ Foo_C_ClientEngine::expected_results (stats);
+ Foo_C_Custom_ClientEngine::expected_results (stats);
+
+ for (unsigned i = 0; i < this->num_servants_; i++)
+ {
+ this->servants_.servant(i)->gather_stats(stats);
+ }
+
+ return stats.actual_vs_expected ();
+}
View Lorry Controller Status