TAO/tests/Leader_Followers/server.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "ace/Task.h"
   3 #include "test_i.h"
   4
   5 #include "tao/Strategies/advanced_resource.h"
   6
   7 const ACE_TCHAR *ior_output_file = ACE_TEXT("ior");
   8
   9 int number_of_event_loop_threads = 1;
  10
  11 int
  12 parse_args (int argc, ACE_TCHAR *argv[])
  13 {
  14   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("e:o:"));
  15   int c;
  16
  17   while ((c = get_opts ()) != -1)
  18     switch (c)
  19       {
  20       case 'o':
  21         ior_output_file = get_opts.opt_arg ();
  22         break;
  23
  24       case 'e':
  25         number_of_event_loop_threads = ACE_OS::atoi (get_opts.opt_arg ());
  26         break;
  27
  28       case '?':
  29       default:
  30         ACE_ERROR_RETURN ((LM_ERROR,
  31                            "usage:  %s "
  32                            "-o <iorfile> "
  33                            "-e <number of event loop threads> "
  34                            "\n",
  35                            argv [0]),
  36                           -1);
  37       }
  38
  39   // Indicates successful parsing of the command line
  40   return 0;
  41 }
  42
  43 class Event_Loop_Task : public ACE_Task_Base
  44 {
  45 public:
  46   Event_Loop_Task (CORBA::ORB_ptr orb)
  47     :  orb_ (CORBA::ORB::_duplicate (orb))
  48     {
  49     }
  50
  51   int svc (void)
  52     {
  53
  54       try
  55         {
  56           this->orb_->run ();
  57         }
  58       catch (const CORBA::Exception& ex)
  59         {
  60           ex._tao_print_exception ("Exception caught in thread:");
  61           return -1;
  62         }
  63
  64
  65       return 0;
  66     }
  67
  68 private:
  69   CORBA::ORB_var orb_;
  70   // ORB reference.
  71 };
  72
  73 int
  74 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  75 {
  76   try
  77     {
  78       CORBA::ORB_var orb =
  79         CORBA::ORB_init (argc,
  80                          argv);
  81
  82       CORBA::Object_var poa_object =
  83         orb->resolve_initial_references ("RootPOA");
  84
  85       PortableServer::POA_var root_poa =
  86         PortableServer::POA::_narrow (poa_object.in ());
  87
  88       PortableServer::POAManager_var poa_manager =
  89         root_poa->the_POAManager ();
  90
  91       poa_manager->activate ();
  92
  93       if (parse_args (argc, argv) != 0)
  94         return -1;
  95
  96       test_i servant (orb.in ());
  97
  98       PortableServer::ObjectId_var id =
  99         root_poa->activate_object (&servant);
 100
 101       CORBA::Object_var object = root_poa->id_to_reference (id.in ());
 102
 103       test_var server =
 104         test::_narrow (object.in ());
 105
 106       CORBA::String_var ior =
 107         orb->object_to_string (server.in ());
 108
 109       ACE_DEBUG ((LM_DEBUG, "Activated as <%C>\n", ior.in ()));
 110
 111       FILE *output_file = ACE_OS::fopen (ior_output_file, "w");
 112       if (output_file == 0)
 113         ACE_ERROR_RETURN ((LM_ERROR,
 114                            "Cannot open output file for writing IOR: %s",
 115                            ior_output_file),
 116                           -1);
 117       ACE_OS::fprintf (output_file, "%s", ior.in ());
 118       ACE_OS::fclose (output_file);
 119
 120       Event_Loop_Task event_loop_task (orb.in ());
 121
 122       if (event_loop_task.activate (THR_NEW_LWP | THR_JOINABLE,
 123                                     number_of_event_loop_threads) != 0)
 124         ACE_ERROR_RETURN ((LM_ERROR,
 125                            "Cannot activate event_loop threads\n"),
 126                           -1);
 127
 128       event_loop_task.thr_mgr ()->wait ();
 129
 130       ACE_DEBUG ((LM_DEBUG, "Server: Event loop finished\n"));
 131
 132       root_poa->destroy (1,
 133                          1);
 134     }
 135   catch (const CORBA::Exception& ex)
 136     {
 137       ex._tao_print_exception ("Exception caught:");
 138       return -1;
 139     }
 140
 141   return 0;
 142 }