TAO/tests/POA/On_Demand_Act_Direct_Coll/Collocated_Test.cpp

   1 #include "Server_Task.h"
   2 #include "Client_Task.h"
   3 #include "ace/Get_Opt.h"
   4 #include "ace/Argv_Type_Converter.h"
   5 #include "ace/SString.h"
   6 #include "ace/Manual_Event.h"
   7
   8 const ACE_TCHAR *output = ACE_TEXT("test.ior");
   9 const ACE_TCHAR *input = ACE_TEXT("file://test.ior");
  10 CORBA::Boolean exception_occurred = false;
  11 ACE_CString server_orb;
  12 ACE_CString client_orb;
  13
  14 int
  15 parse_args (int argc, ACE_TCHAR *argv[])
  16 {
  17   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:o:td:"));
  18   int c;
  19
  20   while ((c = get_opts ()) != -1)
  21     switch (c)
  22       {
  23       case 'o':
  24         output = get_opts.opt_arg ();
  25         break;
  26       case 'k':
  27         input = get_opts.opt_arg ();
  28         break;
  29       case 't':
  30         // no exception expected
  31         exception_occurred = false;
  32         break;
  33       case 'd':
  34         // exception expected
  35         exception_occurred = true;
  36         break;
  37       case '?':
  38       default:
  39         // This is a hack but that is okay!
  40         return 0;
  41       }
  42   // Indicates successful parsing of the command line
  43   return 0;
  44 }
  45
  46 int
  47 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  48 {
  49   if (parse_args (argc,
  50                   argv) == -1)
  51     return -1;
  52
  53   try
  54     {
  55       ACE_Argv_Type_Converter satc (argc, argv);
  56       CORBA::ORB_var sorb =
  57         CORBA::ORB_init (satc.get_argc (),
  58                          satc.get_TCHAR_argv (),
  59                          server_orb.c_str());
  60
  61       ACE_Manual_Event me;
  62       Server_Task server_task (output,
  63                                sorb.in (),
  64                                me,
  65                                ACE_Thread_Manager::instance ());
  66
  67       if (server_task.activate (THR_NEW_LWP | THR_JOINABLE,
  68                                 1,
  69                                 1) == -1)
  70         {
  71           ACE_ERROR ((LM_ERROR, "Error activating server task\n"));
  72         }
  73
  74       // Wait for the server thread to do some processing
  75       me.wait ();
  76       ACE_Argv_Type_Converter catc (argc, argv);
  77       CORBA::ORB_var corb =
  78         CORBA::ORB_init (catc.get_argc (),
  79                          catc.get_TCHAR_argv (),
  80                          client_orb.c_str());
  81
  82       Client_Task client_task (input,
  83                                corb.in (),
  84                                ACE_Thread_Manager::instance (),
  85                                exception_occurred);
  86
  87       if (client_task.activate (THR_NEW_LWP | THR_JOINABLE,
  88                                 1,
  89                                 1) == -1)
  90         {
  91           ACE_ERROR ((LM_ERROR, "Error activating client task\n"));
  92         }
  93
  94       // Wait for the client and server to finish
  95       ACE_Thread_Manager::instance ()->wait ();
  96
  97       // Now that all threads have completed we can destroy the ORB
  98       sorb->destroy ();
  99       if (server_orb != client_orb)
 100         {
 101           corb->destroy();
 102         }
 103     }
 104   catch (const CORBA::Exception&)
 105     {
 106       // Ignore exceptions..
 107     }
 108   return 0;
 109 }