diff options
Diffstat (limited to 'TAO/tests/MT_Client/client.cpp')
-rw-r--r-- | TAO/tests/MT_Client/client.cpp | 188 |
1 files changed, 188 insertions, 0 deletions
diff --git a/TAO/tests/MT_Client/client.cpp b/TAO/tests/MT_Client/client.cpp new file mode 100644 index 00000000000..aaab6575632 --- /dev/null +++ b/TAO/tests/MT_Client/client.cpp @@ -0,0 +1,188 @@ +// $Id$ + +#include "testC.h" +#include "tao/debug.h" +#include "ace/Get_Opt.h" +#include "ace/Task.h" + +ACE_RCSID(MT_Client, client, "$Id$") + +const char *ior = "file://test.ior"; +int nthreads = 5; +int niterations = 5; +int server_shutdown = 0; + +int +parse_args (int argc, char *argv[]) +{ + ACE_Get_Opt get_opts (argc, argv, "k:n:i:x"); + int c; + + while ((c = get_opts ()) != -1) + switch (c) + { + case 'k': + ior = get_opts.opt_arg (); + break; + case 'n': + nthreads = ACE_OS::atoi (get_opts.opt_arg ()); + break; + case 'i': + niterations = ACE_OS::atoi (get_opts.opt_arg ()); + break; + case 'x': + server_shutdown = 1; + break; + case '?': + default: + ACE_ERROR_RETURN ((LM_ERROR, + "usage: %s " + "-k <ior> " + "-n <nthreads> " + "-i <niterations> " + "\n", + argv [0]), + -1); + } + // Indicates sucessful parsing of the command line + return 0; +} + +class Client : public ACE_Task_Base +{ + // = TITLE + // Run the client thread + // + // = DESCRIPTION + // Use the ACE_Task_Base class to run the client threads. + // +public: + Client (Simple_Server_ptr server, int niterations); + // ctor + + virtual int svc (void); + // The thread entry point. + +private: + void validate_connection (ACE_ENV_SINGLE_ARG_DECL_NOT_USED); + // Validate the connection + +private: + Simple_Server_var server_; + // The server. + + int niterations_; + // The number of iterations on each client thread. +}; + +int +main (int argc, char *argv[]) +{ + ACE_TRY_NEW_ENV + { + CORBA::ORB_var orb = + CORBA::ORB_init (argc, argv, "" ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + if (parse_args (argc, argv) != 0) + return 1; + + CORBA::Object_var object = + orb->string_to_object (ior ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + Simple_Server_var server = + Simple_Server::_narrow (object.in () ACE_ENV_ARG_PARAMETER); + ACE_TRY_CHECK; + + if (CORBA::is_nil (server.in ())) + { + ACE_ERROR_RETURN ((LM_ERROR, + "Object reference <%s> is nil\n", + ior), + 1); + } + + Client client (server.in (), niterations); + if (client.activate (THR_NEW_LWP | THR_JOINABLE, + nthreads) != 0) + ACE_ERROR_RETURN ((LM_ERROR, + "Cannot activate client threads\n"), + 1); + + client.thr_mgr ()->wait (); + + ACE_DEBUG ((LM_DEBUG, "threads finished\n")); + + if (server_shutdown) + { + server->shutdown (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + } + + orb->destroy (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, + "Caught exception:"); + return 1; + } + ACE_ENDTRY; + + return 0; +} + +// **************************************************************** + +Client::Client (Simple_Server_ptr server, + int niterations) + : server_ (Simple_Server::_duplicate (server)), + niterations_ (niterations) +{ +} + +void +Client::validate_connection (ACE_ENV_SINGLE_ARG_DECL) +{ + // Ping the object 100 times, ignoring all exceptions. + // It would be better to use validate_connection() but the test must + // run on minimum CORBA builds too! + for (int j = 0; j != 100; ++j) + { + ACE_TRY + { + this->server_->test_method (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + } + ACE_CATCHANY {} ACE_ENDTRY; + } +} + +int +Client::svc (void) +{ + ACE_TRY_NEW_ENV + { + this->validate_connection (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + + for (int i = 0; i < this->niterations_; ++i) + { + this->server_->test_method (ACE_ENV_SINGLE_ARG_PARAMETER); + ACE_TRY_CHECK; + + if (TAO_debug_level > 0 && i % 100 == 0) + ACE_DEBUG ((LM_DEBUG, "(%P|%t) iteration = %d\n", + i)); + } + } + ACE_CATCHANY + { + ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION, + "MT_Client: exception raised"); + } + ACE_ENDTRY; + return 0; +} |