TAO/orbsvcs/tests/FT_App/FT_ReplicationManagerController.cpp

   1 #include "orbsvcs/FT_ReplicationManagerC.h"
   2 #include <ace/Get_Opt.h>
   3 // FUZZ: disable check_for_streams_include
   4 #include "ace/streams.h"
   5 #include "ace/OS_NS_unistd.h"
   6 #include "ace/OS_NS_stdio.h"
   7
   8 // A simple class for building a client that "controls' the
   9 // Replication Manager (right now it just shuts it down).
  10 class TAO_FT_ReplicationManagerController
  11 {
  12 /////////////////////////////
  13 // Public interface.
  14 public:
  15   TAO_FT_ReplicationManagerController ();
  16   virtual ~TAO_FT_ReplicationManagerController ();
  17
  18   int init (int &argc, ACE_TCHAR *argv[]);
  19   int parse_args (int &argc, ACE_TCHAR *argv[]);
  20   int run ();
  21
  22 /////////////////////////////
  23 // Forbidden.
  24 private:
  25   TAO_FT_ReplicationManagerController (
  26     const TAO_FT_ReplicationManagerController & rhs);
  27   TAO_FT_ReplicationManagerController & operator = (
  28     const TAO_FT_ReplicationManagerController & rhs);
  29
  30 /////////////////////////////
  31 // Private implementation.
  32 private:
  33   void usage (FILE* out) const;
  34
  35 /////////////////////////////
  36 // Data members.
  37 private:
  38   CORBA::ORB_var orb_;
  39   FT::ReplicationManager_var replication_manager_;
  40   const ACE_TCHAR *rm_ior_;
  41   int shutdown_;
  42 };
  43
  44 TAO_FT_ReplicationManagerController::TAO_FT_ReplicationManagerController ()
  45   : orb_ (CORBA::ORB::_nil())
  46   , replication_manager_ (FT::ReplicationManager::_nil())
  47   , rm_ior_ (0)
  48   , shutdown_ (0)
  49 {
  50 }
  51
  52 TAO_FT_ReplicationManagerController::~TAO_FT_ReplicationManagerController ()
  53 {
  54 }
  55
  56 int TAO_FT_ReplicationManagerController::init (int &argc, ACE_TCHAR *argv[])
  57 {
  58   int result = 0;
  59
  60   try
  61   {
  62     // Initialize the ORB.
  63     this->orb_ = CORBA::ORB_init (argc, argv);
  64
  65     // Parse arguments.
  66     result = this->parse_args (argc, argv);
  67     if (result == 0)
  68     {
  69       CORBA::Object_var obj = CORBA::Object::_nil ();
  70       if (this->rm_ior_ != 0)
  71       {
  72         obj = this->orb_->string_to_object (
  73           ACE_TEXT_ALWAYS_CHAR (this->rm_ior_));
  74       }
  75       else
  76       {
  77         obj = this->orb_->resolve_initial_references (
  78           "ReplicationManager");
  79       }
  80       this->replication_manager_ = FT::ReplicationManager::_narrow (
  81         obj.in());
  82       if (CORBA::is_nil (this->replication_manager_.in()))
  83       {
  84         ACE_ERROR ((LM_ERROR,
  85           ACE_TEXT ("TAO_FT_ReplicationManagerController::init: ")
  86           ACE_TEXT ("Could not get Replication Manager's IOR.\n")
  87         ));
  88         result = -1;
  89       }
  90     }
  91   }
  92   catch (const CORBA::Exception& ex)
  93   {
  94     ex._tao_print_exception (
  95       ACE_TEXT (
  96         "TAO_FT_ReplicationManagerController::init:\n"));
  97     result = -1;
  98   }
  99
 100   return result;
 101 }
 102
 103
 104 int TAO_FT_ReplicationManagerController::parse_args (int &argc, ACE_TCHAR *argv[])
 105 {
 106   int result = 0;
 107
 108   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:x"));
 109   int c;
 110
 111   while (result == 0 && (c = get_opts ()) != -1)
 112   {
 113     switch (c)
 114     {
 115       case 'k':
 116       {
 117         this->rm_ior_ = get_opts.opt_arg ();
 118         break;
 119       }
 120
 121       case 'x':
 122       {
 123         this->shutdown_ = 1;
 124         break;
 125       }
 126
 127       default:
 128         ACE_OS::fprintf (stderr, "%s: Unknown argument - %c\n",
 129                          ACE_TEXT_ALWAYS_CHAR (argv[0]), c);
 130         this->usage(stderr);
 131         result = -1;
 132         break;
 133
 134       case '?':
 135         this->usage(stderr);
 136         result = -1;
 137     }
 138   }
 139
 140   return result;
 141 }
 142
 143 void TAO_FT_ReplicationManagerController::usage (FILE* out) const
 144 {
 145   ACE_OS::fprintf (out, "usage"
 146                         " -k <replication manager ior file>"
 147                         " -x (shutdown the Replication Manager)\n");
 148 }
 149
 150 int TAO_FT_ReplicationManagerController::run ()
 151 {
 152   int result = 0;
 153
 154   try
 155   {
 156     if (this->shutdown_ == 1)
 157     {
 158       this->replication_manager_->shutdown ();
 159       ACE_Time_Value tv (0, 500000);
 160       ACE_OS::sleep (tv);
 161     }
 162   }
 163   catch (const CORBA::Exception& ex)
 164   {
 165     ex._tao_print_exception (
 166       ACE_TEXT ("TAO_FT_ReplicationManagerController::run:\n"));
 167     result = -1;
 168   }
 169
 170   return result;
 171 }
 172
 173 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 174 {
 175   int result = 0;
 176   TAO_FT_ReplicationManagerController rmctrl;
 177   result = rmctrl.init (argc, argv);
 178   if (result == 0)
 179   {
 180     result = rmctrl.run ();
 181   }
 182   return result;
 183 }