TAO/examples/AMH/Sink_Server/Base_Server.cpp

   1 #include "ace/Sched_Params.h"
   2 #include "ace/Get_Opt.h"
   3
   4 #include "AMH_Servant.h"
   5 #include "Base_Server.h"
   6 #include "tao/Strategies/advanced_resource.h"
   7
   8 #include "TestC.h"
   9
  10 #if !defined (__ACE_INLINE__)
  11 # include "Base_Server.inl"
  12 #endif /* __ACE_INLINE__ */
  13
  14 Base_Server::Base_Server (int &argc, ACE_TCHAR **argv)
  15   : argc_ (argc)
  16   , argv_ (argv)
  17   , ior_output_file_ (ACE_TEXT("test.ior"))
  18 {
  19 }
  20
  21 Base_Server::~Base_Server (void)
  22 {
  23 }
  24
  25 int
  26 Base_Server::parse_args (void)
  27 {
  28   // *** To get correct behaviour, set ** POSIXLY_CORECT=1 ** on Linux
  29   // systems!!! ***
  30   ACE_Get_Opt get_opts (this->argc_, this->argv_, ACE_TEXT("o:"));
  31   int c;
  32   int count_argv = 0;
  33
  34   while ((c = get_opts ()) != -1)
  35     {
  36       ++count_argv;
  37       switch (c)
  38         {
  39         case 'o':
  40           {
  41             this->ior_output_file_ = get_opts.opt_arg ();
  42
  43             // Remove the option '-o' from argv []
  44             // to avoid any confusion that might result.
  45             {
  46               for (int i = count_argv; i <= this->argc_; ++i)
  47                 this->argv_ [i] = this->argv_ [i+2];
  48             }
  49
  50             // Decrement the value of this->argc_ to reflect the removal
  51             // of '-o' option.
  52             this->argc_ = this->argc_ - 2;
  53
  54             return 1;
  55           }
  56         case '?':
  57         default:
  58           // Don't do anything.
  59           break;
  60         }
  61     }
  62
  63   return 0;
  64 }
  65
  66 void
  67 Base_Server::try_RT_scheduling (void)
  68 {
  69   int priority =
  70     (ACE_Sched_Params::priority_min (ACE_SCHED_FIFO)
  71      + ACE_Sched_Params::priority_max (ACE_SCHED_FIFO)) / 2;
  72   priority = ACE_Sched_Params::next_priority (ACE_SCHED_FIFO,
  73                                               priority);
  74
  75   // Enable FIFO scheduling, e.g., RT scheduling class on Solaris.
  76   if (ACE_OS::sched_params (ACE_Sched_Params (ACE_SCHED_FIFO,
  77                                               priority,
  78                                               ACE_SCOPE_PROCESS)) != 0)
  79     {
  80       if (errno == EPERM)
  81         {
  82           ACE_DEBUG ((LM_DEBUG,
  83                       "server (%P|%t): user is not superuser, "
  84                       "test runs in time-shared class\n"));
  85         }
  86       else
  87         ACE_ERROR ((LM_ERROR,
  88                     "server (%P|%t): sched_params failed\n"));
  89     }
  90 }
  91
  92 int
  93 Base_Server::shutdown_orb_and_poa (void)
  94 {
  95   try
  96     {
  97       this->root_poa_->destroy (1, 1);
  98       this->root_poa_ = PortableServer::POA::_nil ();
  99
 100       this->orb_->destroy ();
 101       this->orb_ = CORBA::ORB::_nil ();
 102     }
 103   catch (const CORBA::Exception& ex)
 104     {
 105       ex._tao_print_exception ("Exception raised shutting down ORB or POA");
 106       return -1;
 107     }
 108
 109   // If we have got to this point, everything has gone well.  return
 110   // normally
 111   return 1;
 112 }
 113
 114 int
 115 Base_Server::start_orb_and_poa (void)
 116 {
 117   try
 118     {
 119       this->orb_ = CORBA::ORB_init (this->argc_, this->argv_);
 120
 121       CORBA::Object_var poa_object =
 122         this->orb_->resolve_initial_references("RootPOA");
 123
 124       if (CORBA::is_nil (poa_object.in ()))
 125         ACE_ERROR_RETURN ((LM_ERROR,
 126                            " (%P|%t) Unable to initialize the POA.\n"),
 127                           1);
 128
 129       this->root_poa_ = PortableServer::POA::_narrow (poa_object.in ());
 130
 131       PortableServer::POAManager_var poa_manager =
 132         this->root_poa_->the_POAManager ();
 133
 134       poa_manager->activate ();
 135     }
 136   catch (const CORBA::Exception& ex)
 137     {
 138       ex._tao_print_exception ("Exception raised initialising ORB or POA");
 139       return -1;
 140     }
 141
 142   // If we have got to this point, everything has gone well.  return
 143   // normally
 144   return 1;
 145 }
 146
 147 void
 148 Base_Server::register_servant (AMH_Servant *servant)
 149 {
 150   try
 151     {
 152       Test::Roundtrip_var roundtrip =
 153         servant->_this ();
 154
 155       CORBA::String_var ior =
 156         this->orb_->object_to_string (roundtrip.in ());
 157
 158       (void) this->write_ior_to_file (ior.in ());
 159     }
 160   catch (const CORBA::Exception& ex)
 161     {
 162       ex._tao_print_exception ("Exception raised while registering servant");
 163     }
 164 }
 165
 166
 167 void
 168 Base_Server::run_event_loop (void)
 169 {
 170   try
 171     {
 172       ACE_Time_Value period (0, 11000);
 173       while (1)
 174         {
 175           // @@ Mayur, where's the work_pending() call?
 176           //
 177           // Mayur: Nope. There is purposely no work_pending call. The
 178           // reactor doesn't acknowledge expired timers as work. The
 179           // reactor only thinks I/O events as work.  Thus, timers
 180           // that have expired after all the client requests are made
 181           // are never handled. This results in the server not sending
 182           // (some) replies to the client and the client just ends up
 183           // waiting (forever) for them.
 184           this->orb_->perform_work (period);
 185         }
 186     }
 187   catch (const CORBA::Exception& ex)
 188     {
 189       ex._tao_print_exception (
 190         "Caught exceptionin Base_Server::run_event_loop");
 191     }
 192 }
 193
 194 int
 195 Base_Server::write_ior_to_file (const char * ior)
 196 {
 197   // If the ior_output_file exists, output the ior to it
 198   FILE *output_file =
 199     ACE_OS::fopen (this->ior_output_file_, "w");
 200
 201   if (output_file == 0)
 202     {
 203       ACE_ERROR ((LM_ERROR,
 204                   "Cannot open output file for writing IOR: %s",
 205                   this->ior_output_file_));
 206       return -1;
 207     }
 208
 209   ACE_OS::fprintf (output_file, "%s", ior);
 210   ACE_OS::fclose (output_file);
 211   return 0;
 212 }