TAO/tests/Big_Reply/client.cpp

   1 #include "Client_Task.h"
   2 #include "ace/Get_Opt.h"
   3
   4
   5 const ACE_TCHAR *ior = ACE_TEXT ("file://test.ior");
   6 bool server_shutdown = false;
   7
   8 int
   9 parse_args (int argc, ACE_TCHAR *argv[])
  10 {
  11   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:x"));
  12   int c;
  13
  14   while ((c = get_opts ()) != -1)
  15     switch (c)
  16       {
  17       case 'k':
  18         ior = get_opts.opt_arg ();
  19         break;
  20       case 'x':
  21         server_shutdown = true;
  22         break;
  23       case '?':
  24       default:
  25         ACE_ERROR_RETURN ((LM_ERROR,
  26                            "usage:  %s "
  27                            "-k <ior>"
  28                            "-x shutdown"
  29                            "\n",
  30                            argv [0]),
  31                           -1);
  32       }
  33   // Indicates successful parsing of the command line
  34   return 0;
  35 }
  36
  37 int
  38 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  39 {
  40   try
  41     {
  42       CORBA::ORB_var orb =
  43         CORBA::ORB_init (argc, argv);
  44
  45       if (parse_args (argc, argv) != 0)
  46         return 1;
  47
  48       CORBA::Object_var tmp =
  49         orb->string_to_object (ior);
  50
  51       Test::Big_Reply_var reply_gen =
  52         Test::Big_Reply::_narrow (tmp.in ());
  53
  54       if (CORBA::is_nil (reply_gen.in ()))
  55         {
  56           ACE_ERROR_RETURN ((LM_DEBUG,
  57                              "Nil coordinator reference <%s>\n",
  58                              ior),
  59                             1);
  60         }
  61
  62       Client_Task client_task (reply_gen.in (),
  63                                10,
  64                                ACE_Thread_Manager::instance ());
  65
  66       if (client_task.activate (THR_NEW_LWP | THR_JOINABLE, 4, 1) == -1)
  67         {
  68           ACE_ERROR ((LM_ERROR, "Error activating client task\n"));
  69         }
  70       ACE_Thread_Manager::instance ()->wait ();
  71
  72       ACE_DEBUG ((LM_DEBUG, "Got all the replies...\n"));
  73
  74       if (server_shutdown)
  75         {
  76           // Request the server to shutdown
  77           reply_gen->shutdown ();
  78         }
  79
  80       orb->destroy ();
  81     }
  82   catch (const CORBA::Exception& ex)
  83     {
  84       ex._tao_print_exception ("Exception caught:");
  85       return 1;
  86     }
  87
  88   return 0;
  89 }