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