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 (void)
  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 (void)
  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 (void)
 122 {
 123   return this->scheduler_.in ();
 124 }
 125 /*
 126 int
 127 DT_Test::activate_task (void)
 128 {
 129
 130 long flags;
 131   flags = THR_NEW_LWP | THR_JOINABLE;
 132   flags |=
 133     orb_->orb_core ()->orb_params ()->scope_policy () |
 134     orb_->orb_core ()->orb_params ()->sched_policy ();
 135
 136   if (this->activate (flags,
 137           1) == -1)
 138     {
 139       if (ACE_OS::last_error () == EPERM)
 140   ACE_ERROR_RETURN ((LM_ERROR,
 141          ACE_TEXT ("Insufficient privilege to run this test.\n")),
 142         -1);
 143     }
 144   return 0;
 145 }
 146
 147 int
 148 DT_Test::svc (void)
 149 {
 150   try
 151     {
 152       dt_creator_->create_distributable_threads (current_.in ());
 153
 154     }
 155   catch (const CORBA::Exception& ex)
 156     {
 157       ex._tao_print_exception ("Caught exception:");
 158       return -1;
 159     }
 160
 161   return 0;
 162 }
 163 */
 164
 165 int
 166 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 167 {
 168   int status = 0;
 169
 170   ACE_Service_Config::static_svcs ()->insert (&ace_svc_desc_MIF_DT_Creator);
 171
 172   try
 173     {
 174       ACE_DEBUG ((LM_DEBUG,
 175          "%t Main Thread\n"));
 176       DT_TEST::instance ()->run (argc, argv);
 177     }
 178   catch (const CORBA::Exception& ex)
 179     {
 180       ex._tao_print_exception ("Caught exception:");
 181       status = 1;
 182     }
 183
 184   ACE_Service_Config::static_svcs ()->remove (ACE_TEXT ("MIF_DT_Creator"));
 185
 186   ACE_DEBUG ((LM_DEBUG, "END\n"));
 187   return status;
 188 }
 189
 190 ACE_SINGLETON_TEMPLATE_INSTANTIATE(ACE_Singleton, DT_Test,  TAO_SYNCH_MUTEX);