TAO/tests/POA/Object_Reactivation/Object_Reactivation.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file     Object_Reactivation.cpp
   5  *
   6  *   This program tests the reactivation of a servant that has been
   7  *   deactivated but not removed from the Active Object Map yet.
   8  *
   9  *  @author  Irfan Pyarali
  10  */
  11 //=============================================================================
  12
  13
  14 #include "testS.h"
  15 #include "ace/Task.h"
  16 #include "ace/Get_Opt.h"
  17 #include "ace/SString.h"
  18 #include "ace/Auto_Event.h"
  19 #include "ace/OS_NS_unistd.h"
  20
  21 static int debug = 1;
  22
  23 static int
  24 parse_args (int argc, ACE_TCHAR **argv)
  25 {
  26   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("d:"));
  27   int c;
  28
  29   while ((c = get_opts ()) != -1)
  30     switch (c)
  31       {
  32       case 'd':
  33         debug = ACE_OS::atoi (get_opts.opt_arg ());
  34         break;
  35
  36       default:
  37         ACE_ERROR_RETURN ((LM_ERROR,
  38                            "usage:  %s "
  39                            "-d debug "
  40                            "\n",
  41                            argv [0]),
  42                           -1);
  43       }
  44
  45   return 0;
  46 }
  47
  48 class test_i : public POA_test
  49 {
  50 public:
  51   test_i (ACE_Auto_Event &event);
  52
  53   void deactivate_self ();
  54
  55 private:
  56   ACE_Auto_Event &event_;
  57 };
  58
  59 test_i::test_i (ACE_Auto_Event &event)
  60   : event_ (event)
  61 {
  62 }
  63
  64 void
  65 test_i::deactivate_self ()
  66 {
  67   PortableServer::POA_var poa = this->_default_POA ();
  68
  69   PortableServer::ObjectId_var id = poa->servant_to_id (this);
  70
  71   if (debug)
  72     ACE_DEBUG ((LM_DEBUG, "(%t) Deactivating servant\n"));
  73
  74   poa->deactivate_object (id.in ());
  75
  76   if (debug)
  77     ACE_DEBUG ((LM_DEBUG, "(%t) Deactivation complete: signaling main thread and going to sleep\n"));
  78
  79   int result = this->event_.signal ();
  80   ACE_ASSERT (result == 0);
  81   ACE_UNUSED_ARG (result);
  82
  83   ACE_OS::sleep (3);
  84
  85   if (debug)
  86     ACE_DEBUG ((LM_DEBUG, "(%t) test_i::deactivate_self complete\n"));
  87 }
  88
  89 class Activator : public ACE_Task_Base
  90 {
  91 public:
  92   Activator (test_ptr t,
  93              ACE_Auto_Event &event,
  94              PortableServer::POA_ptr poa,
  95              PortableServer::Servant servant,
  96              const ACE_CString &task_id,
  97              const PortableServer::ObjectId &id);
  98   int svc ();
  99
 100 private:
 101   test_var test_;
 102   ACE_Auto_Event &event_;
 103   PortableServer::POA_var poa_;
 104   PortableServer::Servant servant_;
 105   ACE_CString task_id_;
 106   PortableServer::ObjectId id_;
 107 };
 108
 109 Activator::Activator (test_ptr t,
 110                       ACE_Auto_Event &event,
 111                       PortableServer::POA_ptr poa,
 112                       PortableServer::Servant servant,
 113                       const ACE_CString &task_id,
 114                       const PortableServer::ObjectId &id)
 115   : test_ (test::_duplicate (t)),
 116     event_ (event),
 117     poa_ (PortableServer::POA::_duplicate (poa)),
 118     servant_ (servant),
 119     task_id_ (task_id),
 120     id_ (id)
 121 {
 122 }
 123
 124 int
 125 Activator::svc ()
 126 {
 127   if (debug)
 128     ACE_DEBUG ((LM_DEBUG, "(%t) Waiting for deactivation to complete\n"));
 129
 130   int result = this->event_.wait ();
 131   ACE_ASSERT (result == 0);
 132   ACE_UNUSED_ARG (result);
 133
 134   if (debug)
 135     ACE_DEBUG ((LM_DEBUG, "(%t) Deactivation complete, trying to activate\n"));
 136
 137   try
 138     {
 139       if (this->task_id_ == "first thread")
 140         {
 141           PortableServer::ObjectId_var id =
 142             this->poa_->activate_object (this->servant_);
 143         }
 144       else
 145         {
 146           this->poa_->activate_object_with_id (this->id_,
 147                                                this->servant_);
 148         }
 149
 150       if (debug)
 151         ACE_DEBUG ((LM_DEBUG, "(%t) Activation complete\n"));
 152     }
 153   catch (const CORBA::Exception& ex)
 154     {
 155       ex._tao_print_exception ("Activator::svc");
 156       return -1;
 157     }
 158   return 0;
 159 }
 160
 161 class Deactivator : public ACE_Task_Base
 162 {
 163 public:
 164   Deactivator (test_ptr t);
 165   int svc ();
 166
 167 private:
 168   test_var test_;
 169 };
 170
 171 Deactivator::Deactivator (test_ptr t)
 172   : test_ (test::_duplicate (t))
 173 {
 174 }
 175
 176 int
 177 Deactivator::svc ()
 178 {
 179   try
 180     {
 181       this->test_->deactivate_self ();
 182     }
 183   catch (const CORBA::Exception& ex)
 184     {
 185       ex._tao_print_exception ("Deactivator::svc");
 186       return -1;
 187     }
 188   return 0;
 189 }
 190
 191 int
 192 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 193 {
 194   try
 195     {
 196       // Initialize the ORB first.
 197       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
 198
 199       int parse_args_result =
 200         parse_args (argc, argv);
 201
 202       if (parse_args_result != 0)
 203         return parse_args_result;
 204
 205       // Obtain the RootPOA.
 206       CORBA::Object_var obj =
 207         orb->resolve_initial_references ("RootPOA");
 208
 209       // Get the POA_var object from Object_var.
 210       PortableServer::POA_var root_poa =
 211         PortableServer::POA::_narrow (obj.in ());
 212
 213       // Get the POAManager of the RootPOA.
 214       PortableServer::POAManager_var poa_manager =
 215         root_poa->the_POAManager ();
 216
 217       poa_manager->activate ();
 218
 219       ACE_Auto_Event event1;
 220       test_i servant1 (event1);
 221
 222       ACE_Auto_Event event2;
 223       test_i servant2 (event2);
 224
 225       PortableServer::ObjectId_var id_act =
 226         root_poa->activate_object (&servant1);
 227
 228       CORBA::Object_var object_act = root_poa->id_to_reference (id_act.in ());
 229
 230       test_var test_object1 = test::_narrow (object_act.in ());
 231
 232       id_act = root_poa->activate_object (&servant2);
 233
 234       object_act = root_poa->id_to_reference (id_act.in ());
 235
 236       test_var test_object2 = test::_narrow (object_act.in ());
 237
 238       PortableServer::ObjectId_var id1 =
 239         root_poa->reference_to_id (test_object1.in ());
 240
 241       PortableServer::ObjectId_var id2 =
 242         root_poa->reference_to_id (test_object2.in ());
 243
 244       Activator activator1 (test_object1.in (),
 245                             event1,
 246                             root_poa.in (),
 247                             &servant1,
 248                             "first thread",
 249                             id1.in ());
 250
 251       Activator activator2 (test_object2.in (),
 252                             event2,
 253                             root_poa.in (),
 254                             &servant2,
 255                             "second thread",
 256                             id2.in ());
 257
 258       Deactivator deactivator1 (test_object1.in ());
 259
 260       Deactivator deactivator2 (test_object2.in ());
 261
 262       if (activator1.activate () != 0 ||
 263           activator2.activate () != 0 ||
 264           deactivator1.activate () != 0 ||
 265           deactivator2.activate () != 0)
 266         return -1;
 267
 268       int result = ACE_Thread_Manager::instance ()->wait ();
 269
 270       if (result != 0)
 271         return result;
 272
 273       root_poa->destroy (true, true);
 274
 275       orb->destroy ();
 276     }
 277   catch (const CORBA::Exception& ex)
 278     {
 279       ex._tao_print_exception ("Exception caught");
 280       return -1;
 281     }
 282
 283   return 0;
 284 }