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 ()
  52     {
  53       try
  54         {
  55           this->orb_->run ();
  56         }
  57       catch (const CORBA::Exception& ex)
  58         {
  59           ex._tao_print_exception ("Exception caught in thread:");
  60           return -1;
  61         }
  62
  63
  64       return 0;
  65     }
  66
  67 private:
  68   CORBA::ORB_var orb_;
  69   // ORB reference.
  70 };
  71
  72 int
  73 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  74 {
  75   try
  76     {
  77       CORBA::ORB_var orb =
  78         CORBA::ORB_init (argc,
  79                          argv);
  80
  81       CORBA::Object_var poa_object =
  82         orb->resolve_initial_references ("RootPOA");
  83
  84       PortableServer::POA_var root_poa =
  85         PortableServer::POA::_narrow (poa_object.in ());
  86
  87       PortableServer::POAManager_var poa_manager =
  88         root_poa->the_POAManager ();
  89
  90       poa_manager->activate ();
  91
  92       if (parse_args (argc, argv) != 0)
  93         return -1;
  94
  95       test_i servant (orb.in ());
  96
  97       PortableServer::ObjectId_var id =
  98         root_poa->activate_object (&servant);
  99
 100       CORBA::Object_var object = root_poa->id_to_reference (id.in ());
 101
 102       test_var server =
 103         test::_narrow (object.in ());
 104
 105       CORBA::String_var ior =
 106         orb->object_to_string (server.in ());
 107
 108       ACE_DEBUG ((LM_DEBUG, "Activated as <%C>\n", ior.in ()));
 109
 110       FILE *output_file = ACE_OS::fopen (ior_output_file, "w");
 111       if (output_file == 0)
 112         ACE_ERROR_RETURN ((LM_ERROR,
 113                            "Cannot open output file for writing IOR: %s",
 114                            ior_output_file),
 115                           -1);
 116       ACE_OS::fprintf (output_file, "%s", ior.in ());
 117       ACE_OS::fclose (output_file);
 118
 119       Event_Loop_Task event_loop_task (orb.in ());
 120
 121       if (event_loop_task.activate (THR_NEW_LWP | THR_JOINABLE,
 122                                     number_of_event_loop_threads) != 0)
 123         ACE_ERROR_RETURN ((LM_ERROR,
 124                            "Cannot activate event_loop threads\n"),
 125                           -1);
 126
 127       event_loop_task.thr_mgr ()->wait ();
 128
 129       ACE_DEBUG ((LM_DEBUG, "Server: Event loop finished\n"));
 130
 131       root_poa->destroy (true, true);
 132     }
 133   catch (const CORBA::Exception& ex)
 134     {
 135       ex._tao_print_exception ("Exception caught:");
 136       return -1;
 137     }
 138
 139   return 0;
 140 }