TAO/orbsvcs/tests/ImplRepo/MT_stress/server_i.cpp

   1 #include "server_i.h"
   2
   3 #include "tao/ImR_Client/ImR_Client.h"
   4
   5 #include "orbsvcs/CosNamingC.h"
   6
   7 #include "ace/Get_Opt.h"
   8 #include "ace/Read_Buffer.h"
   9 #include "ace/streams.h"
  10
  11 class test_i : public virtual POA_test
  12 {
  13   int n_;
  14 public:
  15   test_i ()
  16     : n_(0)
  17   {
  18   }
  19
  20   virtual ~test_i ()
  21   {
  22   }
  23
  24   virtual CORBA::Long get ()
  25   {
  26     ++n_;
  27     return n_;
  28   }
  29 };
  30
  31 Server_i::Server_i ()
  32   : server_name_()
  33   , second_name_()
  34 {
  35 }
  36
  37 Server_i::~Server_i()
  38 {
  39   this->root_poa_->destroy(1, 1);
  40   this->orb_->destroy();
  41 }
  42
  43 int
  44 Server_i::parse_args (int argc, ACE_TCHAR* argv[])
  45 {
  46   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("s:d:"));
  47   int c;
  48
  49   while ((c = get_opts ()) != -1)
  50     {
  51       switch (c)
  52         {
  53         case 'd':
  54           this->second_name_ = ACE_TEXT_ALWAYS_CHAR(get_opts.opt_arg ());
  55           break;
  56         case 's':
  57           this->server_name_ = ACE_TEXT_ALWAYS_CHAR(get_opts.opt_arg ());
  58           break;
  59         default:
  60           ACE_ERROR_RETURN ((LM_ERROR,
  61                              "usage:  %s"
  62                              " [-d <name>] name of the server to depend on."
  63                              " -s <name> name of this server."
  64                              "\n",
  65                              argv[0]),1);
  66         }
  67     }
  68
  69   return 0;
  70 }
  71
  72 int
  73 Server_i::init (int argc, ACE_TCHAR** argv)
  74 {
  75   try
  76     {
  77       this->orb_ = CORBA::ORB_init (argc, argv);
  78
  79       int retval = this->parse_args (argc, argv);
  80       if (retval != 0)
  81         return retval;
  82
  83       CORBA::Object_var obj =
  84         this->orb_->resolve_initial_references ("RootPOA");
  85       this->root_poa_ = PortableServer::POA::_narrow (obj.in ());
  86       ACE_ASSERT(! CORBA::is_nil(this->root_poa_.in()));
  87
  88       PortableServer::POAManager_var poa_manager =
  89         this->root_poa_->the_POAManager ();
  90
  91       // If -orbuseimr 1 is specified then all persistent poas will be
  92       // registered with the imr.
  93       CORBA::PolicyList policies (2);
  94       policies.length (2);
  95       policies[0] = this->root_poa_->create_id_assignment_policy (PortableServer::USER_ID);
  96       policies[1] = this->root_poa_->create_lifespan_policy (PortableServer::PERSISTENT);
  97
  98       test_i* test_svt;
  99       ACE_NEW_RETURN (test_svt, test_i, -1);
 100       PortableServer::ServantBase_var scoped_svt(test_svt);
 101
 102       PortableServer::ObjectId_var server_id =
 103         PortableServer::string_to_ObjectId ("Test");
 104
 105       PortableServer::POA_var poa =
 106         this->root_poa_->create_POA (this->server_name_.c_str(),
 107                                      poa_manager.in (),
 108                                      policies);
 109
 110       poa->activate_object_with_id (server_id.in (), test_svt);
 111       obj = poa->id_to_reference (server_id.in ());
 112
 113       policies[0]->destroy();
 114       policies[1]->destroy();
 115
 116       CORBA::Object_var nsobj =
 117         this->orb_->resolve_initial_references ("NameService");
 118       CosNaming::NamingContext_var nc =
 119         CosNaming::NamingContext::_narrow (nsobj.in ());
 120       CosNaming::Name n(1);
 121       n.length (1);
 122       n[0].id = CORBA::string_dup (this->server_name_.c_str ());
 123
 124       try
 125         {
 126           nsobj = nc->resolve (n);
 127         }
 128       catch (CosNaming::NamingContext::NotFound &)
 129         {
 130           nc->bind (n, obj.in ());
 131           return 1;
 132         }
 133       if (this->second_name_.length ())
 134         {
 135           n[0].id = CORBA::string_dup (this->second_name_.c_str ());
 136
 137           try
 138             {
 139               nsobj = nc->resolve (n);
 140             }
 141           catch (CosNaming::NamingContext::NotFound &)
 142             {
 143               return 1;
 144             }
 145           test_var test = test::_narrow(nsobj.in());
 146           test->get();
 147         }
 148     }
 149   catch (const CORBA::Exception& ex)
 150     {
 151       ex._tao_print_exception ("Server_i::init");
 152       throw;
 153     }
 154
 155
 156   return 0;
 157 }
 158
 159 int
 160 Server_i::run ()
 161 {
 162   try
 163     {
 164       PortableServer::POAManager_var poa_manager =
 165         this->root_poa_->the_POAManager ();
 166
 167       poa_manager->activate ();
 168
 169       {
 170         ACE_CString status = this->server_name_ + ACE_CString(".status");
 171         ofstream out(status.c_str());
 172         out << "started" << endl;
 173       }
 174
 175       this->orb_->run ();
 176     }
 177   catch (const CORBA::Exception& ex)
 178     {
 179       ex._tao_print_exception ("Server_i::run");
 180       throw;
 181     }
 182
 183
 184   return 0;
 185 }