TAO/examples/RTScheduling/MIF_Scheduler/test.cpp

   1 #include "test.h"
   2 #include "../Task_Stats.h"
   3 #include "../Synch_i.h"
   4
   5 #include "tao/RTScheduling/RTScheduler_Manager.h"
   6 #include "tao/ORB_Core.h"
   7
   8 #include "ace/Arg_Shifter.h"
   9
  10 #include "ace/Event_Handler.h"
  11 #include "ace/Sig_Handler.h"
  12
  13 class TestShutdown : public ACE_Event_Handler
  14 {
  15 public:
  16   TestShutdown (CORBA::ORB_ptr orb)
  17     : orb_(CORBA::ORB::_duplicate (orb))
  18   {
  19 #if !defined(ACE_LACKS_UNIX_SIGNALS)
  20     this->shutdown_.register_handler (SIGTERM, this);
  21     this->shutdown_.register_handler (SIGINT, this);
  22 #elif defined(ACE_WIN32)
  23     this->shutdown_.register_handler (SIGINT, this);
  24 #endif
  25   }
  26
  27   ~TestShutdown ()
  28   {
  29 #if !defined(ACE_LACKS_UNIX_SIGNALS)
  30     this->shutdown_.remove_handler (SIGTERM);
  31     this->shutdown_.remove_handler (SIGINT);
  32 #elif defined(ACE_WIN32)
  33     this->shutdown_.remove_handler (SIGINT);
  34 #endif
  35   }
  36
  37   virtual int handle_signal (int, siginfo_t*, ucontext_t*)
  38   {
  39     ACE_DEBUG ((LM_DEBUG, "Shutting down...\n"));
  40     this->orb_->shutdown ();
  41     return 0;
  42   }
  43
  44 private:
  45   CORBA::ORB_var orb_;
  46
  47   ACE_Sig_Handler shutdown_;
  48 };
  49
  50 DT_Test::DT_Test ()
  51 {
  52 }
  53
  54 int
  55 DT_Test::init (int argc, ACE_TCHAR *argv [])
  56 {
  57   orb_ = CORBA::ORB_init (argc, argv);
  58
  59   dt_creator_->orb (orb_.in ());
  60
  61   CORBA::Object_var manager_obj = orb_->resolve_initial_references ("RTSchedulerManager");
  62
  63   TAO_RTScheduler_Manager_var manager = TAO_RTScheduler_Manager::_narrow (manager_obj.in ());
  64
  65
  66   ACE_NEW_RETURN (scheduler_,
  67       MIF_Scheduler (orb_.in ()), -1);
  68
  69   manager->rtscheduler (scheduler_.in ());
  70
  71   CORBA::Object_var object =
  72     orb_->resolve_initial_references ("RTScheduler_Current");
  73
  74   current_  =
  75     RTScheduling::Current::_narrow (object.in ());
  76
  77   return 0;
  78 }
  79
  80 void
  81 DT_Test::run (int argc, ACE_TCHAR* argv [])
  82 {
  83   init (argc,argv);
  84
  85   TestShutdown killer (this->orb_.in ());
  86
  87   TASK_STATS::instance ()->init (this->dt_creator_->total_load ());
  88   if (this->dt_creator_->resolve_naming_service () == -1)
  89     return;
  90
  91   this->dt_creator_->activate_root_poa ();
  92
  93   this->dt_creator_->activate_poa_list ();
  94   this->dt_creator_->activate_job_list ();
  95   this->dt_creator_->activate_schedule ();
  96
  97   this->dt_creator_->register_synch_obj ();
  98
  99   ACE_DEBUG ((LM_DEBUG,
 100         "Registered Synch Object\n"));
 101
 102   this->dt_creator_->create_distributable_threads (current_.in ());
 103
 104   ACE_Thread_Manager::instance ()->wait ();
 105
 106   this->orb_->destroy ();
 107
 108   ACE_DEBUG ((LM_DEBUG,
 109         "Test Terminating......\n"));
 110 }
 111
 112
 113 void
 114 DT_Test::dt_creator (MIF_DT_Creator* dt_creator)
 115 {
 116   this->dt_creator_ = dt_creator;
 117 }
 118
 119
 120 MIF_Scheduler*
 121 DT_Test::scheduler ()
 122 {
 123   return this->scheduler_.in ();
 124 }
 125 /*
 126 int
 127 DT_Test::activate_task ()
 128 {
 129 long flags;
 130   flags = THR_NEW_LWP | THR_JOINABLE;
 131   flags |=
 132     orb_->orb_core ()->orb_params ()->scope_policy () |
 133     orb_->orb_core ()->orb_params ()->sched_policy ();
 134
 135   if (this->activate (flags,
 136           1) == -1)
 137     {
 138       if (ACE_OS::last_error () == EPERM)
 139   ACE_ERROR_RETURN ((LM_ERROR,
 140          ACE_TEXT ("Insufficient privilege to run this test.\n")),
 141         -1);
 142     }
 143   return 0;
 144 }
 145
 146 int
 147 DT_Test::svc ()
 148 {
 149   try
 150     {
 151       dt_creator_->create_distributable_threads (current_.in ());
 152     }
 153   catch (const CORBA::Exception& ex)
 154     {
 155       ex._tao_print_exception ("Caught exception:");
 156       return -1;
 157     }
 158
 159   return 0;
 160 }
 161 */
 162
 163 int
 164 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 165 {
 166   int status = 0;
 167
 168   ACE_Service_Config::static_svcs ()->insert (&ace_svc_desc_MIF_DT_Creator);
 169
 170   try
 171     {
 172       ACE_DEBUG ((LM_DEBUG,
 173          "%t Main Thread\n"));
 174       DT_TEST::instance ()->run (argc, argv);
 175     }
 176   catch (const CORBA::Exception& ex)
 177     {
 178       ex._tao_print_exception ("Caught exception:");
 179       status = 1;
 180     }
 181
 182   ACE_Service_Config::static_svcs ()->remove (ACE_TEXT ("MIF_DT_Creator"));
 183
 184   ACE_DEBUG ((LM_DEBUG, "END\n"));
 185   return status;
 186 }
 187
 188 ACE_SINGLETON_TEMPLATE_INSTANTIATE(ACE_Singleton, DT_Test,  TAO_SYNCH_MUTEX);