TAO/tests/Big_Reply/server.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "Big_Reply_i.h"
   3 #include "ace/OS_NS_stdio.h"
   4 #include "ace/Task.h"
   5
   6
   7 const ACE_TCHAR *ior_output_file = ACE_TEXT("test.ior");
   8
   9 // We can change this value if wanted..
  10 const CORBA::ULong data_size = 4000000;
  11
  12 class OrbTask : public ACE_Task_Base
  13 {
  14 public:
  15   OrbTask(const CORBA::ORB_ptr orb)
  16       : orb_(CORBA::ORB::_duplicate(orb))
  17   {
  18   }
  19
  20   virtual int svc()
  21   {
  22       try
  23         {
  24           this->orb_->run ();
  25         }
  26       catch (const CORBA::Exception&)
  27         {
  28         }
  29       return 0;
  30   }
  31
  32 private:
  33   CORBA::ORB_var orb_;
  34 };
  35
  36 static int n_threads = 1;
  37
  38 int
  39 parse_args (int argc, ACE_TCHAR *argv[])
  40 {
  41   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:s:t:"));
  42   int c;
  43
  44   while ((c = get_opts ()) != -1)
  45     switch (c)
  46       {
  47       case 'o':
  48         ior_output_file = get_opts.opt_arg ();
  49         break;
  50       case 't':
  51         n_threads = ACE_OS::atoi(get_opts.opt_arg());
  52         break;
  53       case '?':
  54       default:
  55         ACE_ERROR_RETURN ((LM_ERROR,
  56                            "usage:  %s "
  57                            "-o <iorfile>"
  58                            "-i <no_iterations>"
  59                            "\n",
  60                            argv [0]),
  61                           -1);
  62       }
  63   // Indicates successful parsing of the command line
  64   return 0;
  65 }
  66
  67 int
  68 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  69 {
  70   ACE_DEBUG ((LM_DEBUG, "Starting server\n"));
  71
  72   try
  73     {
  74       CORBA::ORB_var orb =
  75         CORBA::ORB_init (argc, argv);
  76
  77       CORBA::Object_var poa_object =
  78         orb->resolve_initial_references ("RootPOA");
  79
  80       if (CORBA::is_nil (poa_object.in ()))
  81         ACE_ERROR_RETURN ((LM_ERROR,
  82                            " (%P|%t) Unable to initialize the POA.\n"),
  83                           1);
  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
  92       if (parse_args (argc, argv) != 0)
  93         return 1;
  94
  95       Big_Reply_i *big_reply_gen = 0;
  96
  97       ACE_NEW_RETURN (big_reply_gen,
  98                       Big_Reply_i (orb.in (),
  99                                    data_size),
 100                       1);
 101
 102
 103       PortableServer::ServantBase_var big_reply_owner_transfer(big_reply_gen);
 104
 105       PortableServer::ObjectId_var id =
 106         root_poa->activate_object (big_reply_gen);
 107
 108       CORBA::Object_var object = root_poa->id_to_reference (id.in ());
 109
 110       Test::Big_Reply_var big_reply =
 111         Test::Big_Reply::_narrow (object.in ());
 112
 113       CORBA::String_var ior =
 114         orb->object_to_string (big_reply.in ());
 115
 116       // If the ior_output_file exists, output the ior to it
 117       FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
 118       if (output_file == 0)
 119         ACE_ERROR_RETURN ((LM_ERROR,
 120                            "Cannot open output file for writing IOR: %s\n",
 121                            ior_output_file),
 122                               1);
 123       ACE_OS::fprintf (output_file, "%s", ior.in ());
 124       ACE_OS::fclose (output_file);
 125
 126       poa_manager->activate ();
 127
 128       OrbTask task(orb.in());
 129
 130       if (task.activate (THR_NEW_LWP | THR_JOINABLE,
 131                            n_threads) != 0)
 132         ACE_ERROR_RETURN ((LM_ERROR,
 133                            "Cannot activate threads\n"),
 134                           1);
 135       task.wait();
 136
 137       ACE_DEBUG ((LM_DEBUG, "event loop finished\n"));
 138
 139       root_poa->destroy (true, true);
 140     }
 141   catch (const CORBA::Exception& ex)
 142     {
 143       ex._tao_print_exception ("Caught exception:");
 144       return 1;
 145     }
 146
 147   ACE_DEBUG ((LM_DEBUG, "Ending server\n"));
 148
 149   return 0;
 150 }