TAO/tests/Smart_Proxies/Benchmark/server.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file     server.cpp
   5  *
   6  *   This is the server program that tests TAO's Smart Proxy extension.
   7  *
   8  *  @author  Kirthika Parameswaran <kirthika@cs.wustl.edu>
   9  */
  10 //=============================================================================
  11
  12
  13 #include "testS.h"
  14 #include "ace/Get_Opt.h"
  15 #include "ace/OS_NS_string.h"
  16
  17 // The servant
  18
  19 class Test_i : public POA_Test
  20 {
  21 public:
  22   Test_i (CORBA::ORB_ptr orb);
  23
  24   CORBA::Short box_prices  ();
  25
  26   CORBA::Long tickets  (CORBA::Short number);
  27
  28   //FUZZ: disable check_for_lack_ACE_OS
  29   ///FUZZ: enable check_for_lack_ACE_OS
  30   void shutdown  ();
  31
  32 private:
  33   CORBA::ORB_var orb_;
  34 };
  35
  36 Test_i::Test_i (CORBA::ORB_ptr orb)
  37   : orb_ (CORBA::ORB::_duplicate (orb))
  38 {
  39 }
  40
  41 CORBA::Short
  42 Test_i::box_prices ()
  43 {
  44   return 125;
  45 }
  46
  47 CORBA::Long
  48 Test_i::tickets (CORBA::Short number)
  49 {
  50   return 125 * number;
  51 }
  52
  53 void
  54 Test_i::shutdown ()
  55 {
  56   this->orb_->shutdown (false);
  57 }
  58
  59 static const ACE_TCHAR *ior_output_file = 0;
  60
  61 int
  62 parse_args (int argc, ACE_TCHAR *argv[])
  63 {
  64   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:"));
  65   int c;
  66
  67   while ((c = get_opts ()) != -1)
  68     switch (c)
  69       {
  70       case 'o':
  71         ior_output_file = get_opts.opt_arg ();
  72         break;
  73       case '?':
  74       default:
  75         ACE_ERROR_RETURN ((LM_ERROR,
  76                            "usage:  %s "
  77                            "-o <iorfile>"
  78                            "\n",
  79                            argv [0]),
  80                           -1);
  81       }
  82   // Indicates successful parsing of the command line
  83   return 0;
  84 }
  85
  86 int
  87 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  88 {
  89   try
  90     {
  91       CORBA::ORB_var orb = CORBA::ORB_init (argc,
  92                                             argv);
  93       if (parse_args (argc, argv) != 0)
  94         return 1;
  95
  96       Test_i servant (orb.in ());
  97       // Obtain RootPOA.
  98       CORBA::Object_var object =
  99         orb->resolve_initial_references ("RootPOA");
 100
 101       PortableServer::POA_var root_poa =
 102         PortableServer::POA::_narrow (object.in ());
 103
 104
 105       // Get the POAManager of the RootPOA.
 106       PortableServer::POAManager_var poa_manager =
 107         root_poa->the_POAManager ();
 108
 109       PortableServer::ObjectId_var id =
 110         root_poa->activate_object (&servant);
 111
 112       CORBA::Object_var object_act = root_poa->id_to_reference (id.in ());
 113
 114       Test_var Test_object =
 115         Test::_narrow (object_act.in ());
 116
 117       CORBA::String_var ior =
 118         orb->object_to_string (Test_object.in ());
 119
 120       // If the ior_output_file exists, output the ior to it
 121       if (ior_output_file != 0)
 122         {
 123           FILE *output_file =
 124             ACE_OS::fopen (ior_output_file, "w");
 125
 126           if (output_file == 0)
 127             ACE_ERROR_RETURN ((LM_ERROR,
 128                                "Cannot open output file for writing IOR: %s",
 129                                ior_output_file),
 130                               1);
 131
 132           ACE_OS::fprintf (output_file,
 133                            "%s",
 134                            ior.in ());
 135           ACE_OS::fclose (output_file);
 136         }
 137
 138       poa_manager->activate ();
 139
 140       orb->run ();
 141
 142       ACE_DEBUG ((LM_DEBUG,
 143                   "event loop finished\n"));
 144
 145       root_poa->destroy (true, true);
 146
 147       orb->destroy ();
 148     }
 149   catch (const CORBA::Exception& ex)
 150     {
 151       ex._tao_print_exception ("Exception in setting up server");
 152       ACE_ASSERT (0);
 153     }
 154   return 0;
 155 }