TAO/performance-tests/Latency/AMH_Single_Threaded/server.cpp

   1 #include "Roundtrip.h"
   2 #include "ace/Get_Opt.h"
   3 #include "ace/Sched_Params.h"
   4 #include "ace/OS_NS_errno.h"
   5
   6 #include "tao/Strategies/advanced_resource.h"
   7
   8 const ACE_TCHAR *ior_output_file = ACE_TEXT("test.ior");
   9
  10 int
  11 parse_args (int argc, ACE_TCHAR *argv[])
  12 {
  13   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:"));
  14   int c;
  15
  16   while ((c = get_opts ()) != -1)
  17     switch (c)
  18       {
  19       case 'o':
  20         ior_output_file = get_opts.opt_arg ();
  21         break;
  22
  23       case '?':
  24       default:
  25         ACE_ERROR_RETURN ((LM_ERROR,
  26                            "usage:  %s "
  27                            "-o <iorfile>"
  28                            "\n",
  29                            argv [0]),
  30                           -1);
  31       }
  32   // Indicates successful parsing of the command line
  33   return 0;
  34 }
  35
  36 int
  37 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  38 {
  39   int priority =
  40     (ACE_Sched_Params::priority_min (ACE_SCHED_FIFO)
  41      + ACE_Sched_Params::priority_max (ACE_SCHED_FIFO)) / 2;
  42   priority = ACE_Sched_Params::next_priority (ACE_SCHED_FIFO,
  43                                                   priority);
  44   // Enable FIFO scheduling
  45   if (ACE_OS::sched_params (ACE_Sched_Params (ACE_SCHED_FIFO,
  46                                               priority,
  47                                               ACE_SCOPE_PROCESS)) != 0)
  48     {
  49       if (ACE_OS::last_error () == EPERM)
  50         {
  51           ACE_DEBUG ((LM_DEBUG,
  52                       "server (%P|%t): user is not superuser, "
  53                       "test runs in time-shared class\n"));
  54         }
  55       else
  56         ACE_ERROR ((LM_ERROR,
  57                     "server (%P|%t): sched_params failed\n"));
  58     }
  59
  60   try
  61     {
  62       CORBA::ORB_var orb =
  63         CORBA::ORB_init (argc, argv);
  64
  65       CORBA::Object_var poa_object =
  66         orb->resolve_initial_references("RootPOA");
  67
  68       if (CORBA::is_nil (poa_object.in ()))
  69         ACE_ERROR_RETURN ((LM_ERROR,
  70                            " (%P|%t) Unable to initialize the POA.\n"),
  71                           1);
  72
  73       PortableServer::POA_var root_poa =
  74         PortableServer::POA::_narrow (poa_object.in ());
  75
  76       PortableServer::POAManager_var poa_manager =
  77         root_poa->the_POAManager ();
  78
  79       if (parse_args (argc, argv) != 0)
  80         return 1;
  81
  82       Roundtrip *roundtrip_impl;
  83       ACE_NEW_RETURN (roundtrip_impl,
  84                       Roundtrip (orb.in ()),
  85                       1);
  86       PortableServer::ServantBase_var owner_transfer(roundtrip_impl);
  87
  88       Test::Roundtrip_var roundtrip =
  89         roundtrip_impl->_this ();
  90
  91       CORBA::String_var ior =
  92         orb->object_to_string (roundtrip.in ());
  93
  94       // If the ior_output_file exists, output the ior to it
  95       FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
  96       if (output_file == 0)
  97         ACE_ERROR_RETURN ((LM_ERROR,
  98                            "Cannot open output file for writing IOR: %s",
  99                            ior_output_file),
 100                           1);
 101       ACE_OS::fprintf (output_file, "%s", ior.in ());
 102       ACE_OS::fclose (output_file);
 103
 104       poa_manager->activate ();
 105
 106       orb->run ();
 107
 108       ACE_DEBUG ((LM_DEBUG, "(%P|%t) server - event loop finished\n"));
 109
 110       root_poa->destroy (true, true);
 111
 112       orb->destroy ();
 113     }
 114   catch (const CORBA::Exception& ex)
 115     {
 116       ex._tao_print_exception ("Exception caught:");
 117       return 1;
 118     }
 119
 120   return 0;
 121 }