TAO/orbsvcs/tests/FT_Naming/FaultTolerant/LB_server.h

   1 // -*- C++ -*-
   2
   3 //=============================================================================
   4 /**
   5  *  @file   LB_server.h
   6  *
   7  *  @author Jaiganesh Balasubramanian <jai@doc.ece.uci.edu>
   8  *  @author Ossama Othman <ossama@uci.edu>
   9  */
  10 //=============================================================================
  11
  12
  13 #include "orbsvcs/FT_NamingManagerC.h"
  14 #include "orbsvcs/PortableGroupC.h"
  15 #include "tao/PortableServer/PortableServer.h"
  16 #include "ace/Get_Opt.h"
  17 #include "orbsvcs/Naming/Naming_Client.h"
  18
  19 // Forward Declaration of the kind of servant we would like to handle
  20 // by the server.
  21 class Basic;
  22
  23 /**
  24  * @class LB_server
  25  *
  26  * @brief Encapsulate the test in class.
  27  *
  28  * TODO:  Fix description
  29  * This is a class used to retrieve the NamingManager reference and
  30  * create the Object Group.  Servants add themselves to the object
  31  * group and the object group reference is published to the clients.
  32  * Clients then invoke the "servant operations" on the NamingManager
  33  * itself.  At this point the Load Balancing cycle starts.
  34  */
  35 class LB_server
  36 {
  37 public:
  38   /// Constructor
  39   LB_server (int argc, ACE_TCHAR **argv);
  40
  41   /// destroys NamingManager, ORB and POA.
  42   int destroy ();
  43
  44   /// start the ORB.
  45   int start_orb_and_poa ();
  46
  47   /// Get the NamingManager Interface.
  48   int init (int argc, ACE_TCHAR **argv);
  49
  50   /// Create the Object Group using the Load Manager Reference.
  51   /// Returns 0 on successful creation. Returns 1 if object
  52   /// group already exists. Returns -1 if object group cannot
  53   /// be created for some reason.
  54   int create_object_group (const char *group_name);
  55
  56   /// register the servants with the object group.
  57   int register_servant (Basic *servant, const char *loc);
  58
  59   /// remove a servant from the object group.
  60   int remove_servant (const char *loc);
  61
  62   /// obtain the name service facade
  63   TAO_Naming_Client& name_svc ();
  64
  65   /// for servants to register to the initialised ORB.
  66   CORBA::ORB_ptr orb ();
  67
  68   /// for servants to get the reference for object_group.
  69   CORBA::Object_ptr object_group ();
  70
  71   /// for servants to get the reference for object_group.
  72   FT_Naming::NamingManager_ptr naming_manager ();
  73
  74 private:
  75     int parse_args (int, ACE_TCHAR **);
  76
  77 private:
  78   /// Load Manager
  79   FT_Naming::NamingManager_var naming_manager_;
  80
  81   /// Object Group.
  82   CORBA::Object_var object_group_;
  83
  84   /// factory id for the object group.
  85   PortableGroup::GenericFactory::FactoryCreationId_var fcid_;
  86
  87   /// Used to force the creation of the object group only once.
  88   static int called_once_;
  89
  90   /// write the IOR to a file so that it can be read later.
  91   int write_ior_to_file (const ACE_TCHAR *filename,
  92                          const char *ior);
  93
  94   CORBA::ORB_var orb_;
  95   int argc_;
  96   ACE_TCHAR **argv_;
  97   PortableServer::POA_var root_poa_;
  98   const ACE_TCHAR *ior_output_file_;
  99   TAO_Naming_Client name_svc_;
 100 };