TAO/tests/RTScheduling/Scheduling_Interceptor/test_client.cpp

   1 #include "../Scheduler.h"
   2 #include "tao/RTScheduling/RTScheduler_Manager.h"
   3 #include "testC.h"
   4 #include "ace/Get_Opt.h"
   5 #include "ace/SString.h"
   6
   7 ACE_TString ior = ACE_TEXT("file://test.ior");
   8
   9 int
  10 parse_args (int argc,
  11             ACE_TCHAR* argv [])
  12 {
  13   // Parse command line arguments
  14   ACE_Get_Opt opts (argc, argv, ACE_TEXT("f:"));
  15
  16   int c;
  17   while ((c= opts ()) != -1)
  18     {
  19       switch (c)
  20         {
  21         case 'f':
  22           ior = ACE_TEXT("file://");
  23           ior += opts.opt_arg ();
  24           break;
  25         default:
  26           ACE_DEBUG ((LM_DEBUG, "Unknown Option\n"));
  27           return -1;
  28         }
  29     }
  30   return 0;
  31 }
  32
  33 #if defined (THREAD_CANCELLED)
  34 #undef THREAD_CANCELLED
  35 #endif /* THREAD_CANCELLED */
  36
  37 int
  38 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  39 {
  40   CORBA::ORB_var orb;
  41   test_var server;
  42   RTScheduling::Scheduler_var safe_scheduler;
  43
  44   try
  45     {
  46       try
  47         {
  48           orb = CORBA::ORB_init (argc, argv);
  49
  50           if (parse_args (argc, argv) == -1)
  51             return (-1);
  52
  53           CORBA::Object_var manager_obj = orb->resolve_initial_references ("RTSchedulerManager");
  54
  55           TAO_RTScheduler_Manager_var manager = TAO_RTScheduler_Manager::_narrow (manager_obj.in ());
  56
  57           TAO_Scheduler* scheduler = 0;
  58           ACE_NEW_RETURN (scheduler,
  59                           TAO_Scheduler (orb.in ()),
  60                           -1);
  61           safe_scheduler = scheduler;
  62
  63           manager->rtscheduler (scheduler);
  64
  65
  66           CORBA::Object_var object =
  67             orb->string_to_object (ior.c_str ());
  68
  69           server = test::_narrow (object.in ());
  70
  71           if (CORBA::is_nil (server.in ()))
  72             {
  73               ACE_ERROR_RETURN ((LM_ERROR,
  74                                 "ERROR: Object reference <%s> is nil\n",
  75                                 ior.c_str ()),
  76                                 1);
  77             }
  78
  79           CORBA::Object_var current_obj = orb->resolve_initial_references ("RTScheduler_Current");
  80
  81           RTScheduling::Current_var current = RTScheduling::Current::_narrow (current_obj.in ());
  82
  83           const char * name = 0;
  84           CORBA::Policy_ptr sched_param = 0;
  85           CORBA::Policy_ptr implicit_sched_param = 0;
  86
  87           current->begin_scheduling_segment (name,
  88                                             sched_param,
  89                                             implicit_sched_param);
  90
  91           ACE_DEBUG ((LM_DEBUG,
  92                       "Making a one-way request\n"));
  93           server->one_way (ACE_TEXT_ALWAYS_CHAR(ior.c_str ()));
  94
  95           ACE_DEBUG ((LM_DEBUG,
  96                       "Making a two-way request\n"));
  97           server->two_way (ACE_TEXT_ALWAYS_CHAR(ior.c_str ()));
  98
  99           current->end_scheduling_segment (name);
 100         }
 101       catch (const CORBA::THREAD_CANCELLED& )
 102         {
 103           ACE_DEBUG ((LM_DEBUG,
 104             "Distributable Thread Cancelled - Expected Exception\n"));
 105           server->shutdown ();
 106         }
 107       catch (const CORBA::Exception& ex)
 108         {
 109           ex._tao_print_exception ("Caught exception:");
 110
 111           return 0;
 112         }
 113
 114       orb->destroy ();
 115     }
 116   catch (const CORBA::Exception& ex)
 117     {
 118       ex._tao_print_exception ("Caught unexpected exception:");
 119
 120       return 1;
 121     }
 122
 123   return 0;
 124 }