diff options
Diffstat (limited to 'TAO/examples/AMH/Sink_Server/Base_Server.cpp')
| -rw-r--r-- | TAO/examples/AMH/Sink_Server/Base_Server.cpp | 226 |
1 files changed, 226 insertions, 0 deletions
diff --git a/TAO/examples/AMH/Sink_Server/Base_Server.cpp b/TAO/examples/AMH/Sink_Server/Base_Server.cpp new file mode 100644 index 00000000000..b1a722b18d8 --- /dev/null +++ b/TAO/examples/AMH/Sink_Server/Base_Server.cpp @@ -0,0 +1,226 @@ +// $Id$ + +#include "ace/Sched_Params.h" +#include "ace/Get_Opt.h" + +#include "AMH_Servant.h" +#include "Base_Server.h" +#include "tao/Strategies/advanced_resource.h" + +#include "TestC.h" + +#if !defined (__ACE_INLINE__) +# include "Base_Server.inl" +#endif /* __ACE_INLINE__ */ + +Base_Server::Base_Server (int& argc, char **argv) + : argc_ (argc) + , argv_ (argv) + , ior_output_file_ ("test.ior") +{ +} + +Base_Server::~Base_Server (void) +{ + ACE_TRY_NEW_ENV + { + this->root_poa_->destroy (1, 1 ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + this->orb_->destroy (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, + "Exception caught while destroying Base_Server \n"); + } + ACE_ENDTRY; +} + +int +Base_Server::parse_args (void) +{ + // *** To get correct behaviour, set ** POSIXLY_CORECT=1 ** on Linux + // systems!!! *** + ACE_Get_Opt get_opts (this->argc_, this->argv_, "o:"); + int c; + int count_argv = 0; + + while ((c = get_opts ()) != -1) + { + ++count_argv; + switch (c) + { + case 'o': + { + this->ior_output_file_ = get_opts.opt_arg (); + + // Remove the option '-o' from argv [] + // to avoid any confusion that might result. + { + for (int i = count_argv; i <= this->argc_; ++i) + this->argv_ [i] = this->argv_ [i+2]; + } + + // Decrement the value of this->argc_ to reflect the removal + // of '-o' option. + this->argc_ = this->argc_ - 2; + + return 1; + } + case '?': + default: + // Don't do anything. + break; + } + } + + return 0; +} + +void +Base_Server::try_RT_scheduling (void) +{ + int priority = + (ACE_Sched_Params::priority_min (ACE_SCHED_FIFO) + + ACE_Sched_Params::priority_max (ACE_SCHED_FIFO)) / 2; + priority = ACE_Sched_Params::next_priority (ACE_SCHED_FIFO, + priority); + + // Enable FIFO scheduling, e.g., RT scheduling class on Solaris. + if (ACE_OS::sched_params (ACE_Sched_Params (ACE_SCHED_FIFO, + priority, + ACE_SCOPE_PROCESS)) != 0) + { + if (errno == EPERM) + { + ACE_DEBUG ((LM_DEBUG, + "server (%P|%t): user is not superuser, " + "test runs in time-shared class\n")); + } + else + ACE_ERROR ((LM_ERROR, + "server (%P|%t): sched_params failed\n")); + } +} + +int +Base_Server::start_orb_and_poa (void) +{ + ACE_TRY_NEW_ENV + { + this->orb_ = CORBA::ORB_init (this->argc_, + this->argv_, + "" ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + CORBA::Object_var poa_object = + this->orb_->resolve_initial_references("RootPOA" + ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + if (CORBA::is_nil (poa_object.in ())) + ACE_ERROR_RETURN ((LM_ERROR, + " (%P|%t) Unable to initialize the POA.\n"), + 1); + + this->root_poa_ = PortableServer::POA::_narrow (poa_object.in () + ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + PortableServer::POAManager_var poa_manager = + this->root_poa_->the_POAManager (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + + poa_manager->activate (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, + "Exception raised initialising ORB or POA"); + return -1; + } + ACE_ENDTRY; + + // If we have got to this point, everything has gone well. return + // normally + return 1; +} + +void +Base_Server::register_servant (AMH_Servant *servant) +{ + ACE_TRY_NEW_ENV + { + Test::Roundtrip_var roundtrip = + servant->_this (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + + CORBA::String_var ior = + this->orb_->object_to_string (roundtrip.in () + ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + (void) this->write_ior_to_file (ior.in ()); + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, + "Exception raised while registering servant"); + } + ACE_ENDTRY; +} + + +void +Base_Server::run_event_loop (void) +{ + ACE_DECLARE_NEW_CORBA_ENV; + ACE_TRY + { + ACE_Time_Value period (0, 11000); + while (1) + { + // @@ Mayur, where's the work_pending() call? + // + // Mayur: Nope. There is purposely no work_pending call. The + // reactor doesn't acknowledge expired timers as work. The + // reactor only thinks I/O events as work. Thus, timers + // that have expired after all the client requests are made + // are never handled. This results in the server not sending + // (some) replies to the client and the client just ends up + // waiting (forever) for them. + this->orb_->perform_work (period ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + } + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, + "Caught exceptionin Base_Server::run_event_loop"); + } + ACE_ENDTRY; +} + +int +Base_Server::write_ior_to_file (const char * ior) +{ + // If the ior_output_file exists, output the ior to it + FILE *output_file = + ACE_OS::fopen (this->ior_output_file_, "w"); + + if (output_file == 0) + { + ACE_ERROR ((LM_ERROR, + "Cannot open output file for writing IOR: %s", + this->ior_output_file_)); + return -1; + } + + ACE_OS::fprintf (output_file, "%s", ior); + ACE_OS::fclose (output_file); + return 0; +} |