forwarderGroup.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2007 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/
/*********************************************************************
** $RCSfile: forwarderGroup.h,v $ $Revision: 1.3 $ $State: Exp $
*********************************************************************/
#ifndef DtForwarderGroup_H_
#define DtForwarderGroup_H_


#include "rtiMsConfig.h"
#include <vlutil/vlString.h>
#include "peerToPeerSockMgr.h"
#include <rtiUtil.h>
#include <list>
#include <map>

class DT_DLL_FORWARDER DtRtiForwarderGroup
{
protected:
   DtRtiForwarderGroup( const MAKRti::DtString& name, int defaultPort );

public:
   virtual ~DtRtiForwarderGroup();

public: 
   class DtRtiForwarderGroupMember
   {
   public:
      DtRtiForwarderGroupMember( const MAKRti::DtString& name, const MAKRti::DtInetAddr& addr,
         const MAKRti::DtString& host, unsigned int port, DtTcpMsgSocket* socket,
         const MAKRti::DtString& url, bool isPortal ) :
         myName( name ),
         myAddr( addr ),
         myHost( host ),
         myPort( port ),
         mySocket( socket ),
         myUrl( url ),
         myAcceptRemoteConnection( isPortal )
      {}

      ~DtRtiForwarderGroupMember(){}

   private:
      DtRtiForwarderGroupMember( );

   public:
      MAKRti::DtString     myName;
      MAKRti::DtInetAddr   myAddr;
      MAKRti::DtString     myHost;
      unsigned int myPort;
      DtTcpMsgSocket* mySocket;
      MAKRti::DtString     myUrl;
      bool         myAcceptRemoteConnection;
   };

   typedef std::map<DtTcpMsgSocket*, DtRtiForwarderGroupMember*> DtForwarderGroupConnectionList;
   typedef std::map<MAKRti::DtString, DtRtiForwarderGroupMember*> DtForwarderGroupMemberList;

public:
   virtual bool addForwarderToGroup( const MAKRti::DtString& fwderName,
      const MAKRti::DtInetAddr& addr, const MAKRti::DtString& host,
      int connectPort = -1, const MAKRti::DtString& url = NULL,
      bool isPortal = false);

   virtual unsigned int connectToPortal( const MAKRti::DtString& fwderName = NULL, 
                                         bool ignorePortalFlag = false );
   virtual unsigned int connectToPortal( DtTcpMsgSocket* socket, 
                                         bool ignorePortalFlag = false );

   virtual bool connectToForwarder( const MAKRti::DtString& fwderName ); 
   virtual bool connectToForwarder( DtTcpMsgSocket* socket );

   virtual bool disconnectFromForwarder( DtTcpMsgSocket* socket );

   virtual void disconnectFromGroup();

   virtual bool removeForwarder( const MAKRti::DtString& fwderName );
   virtual bool removeForwarder( DtTcpMsgSocket* socket );

   virtual void dissolve();

   virtual const DtForwarderGroupMemberList members() const;

   virtual bool isMember( const MAKRti::DtString& fwderName );
   virtual bool isMember( const DtInetAddrAndPort& addr );
   virtual bool isMember( DtTcpMsgSocket* sock );

   virtual const DtForwarderGroupConnectionList& portalConnections() const;

   virtual const DtForwarderGroupConnectionList& connections() const;

   virtual void setConnectionManager( DtPeerToPeerTcpSockMgr* connMgr );

   virtual void setDefaultPort( int port );

   virtual void setForwarderInfo( const DtInetAddrAndPort& addr, const MAKRti::DtString& url = NULL, 
                                  bool isPortal = false);
   virtual void setForwarderInfo( DtTcpMsgSocket* socket, const MAKRti::DtString& url = NULL, 
                                  bool isPortal = false);

   MAKRti::DtString getForwarderNameByAddr( const DtInetAddrAndPort& addr ) const;

   DtInetAddrAndPort getForwarderAddrByName( const MAKRti::DtString& fwderName ) const;

   virtual bool forwarderIsAPortal( const MAKRti::DtString& fwderName ) const;

   virtual unsigned int portalCount() const;

   virtual unsigned int connectionCount() const;

   virtual unsigned int memberCount() const;

   virtual const MAKRti::DtString& groupName() const;

   virtual DtRtiForwarderGroup::DtRtiForwarderGroupMember* nextAvailablePortal();

   virtual bool updateNextAvailablePortal();

public:
   typedef DtRtiForwarderGroup* (*DtForwarderGroupCreatorFcn)( const MAKRti::DtString&, int );
   
   static DtRtiForwarderGroup* create( const MAKRti::DtString& name, int defaultPort );
   
   static void setCreatorFunction( DtForwarderGroupCreatorFcn creator );

protected:
   MAKRti::DtString                myName;
   DtPeerToPeerTcpSockMgr* myConnectionMgr;
   int                     myDefaultPort;

   DtForwarderGroupConnectionList myConnectionSocketMap;
   DtForwarderGroupMemberList     myConnectionRoster;    
   DtForwarderGroupMemberList     myRoster;              

   DtForwarderGroupConnectionList myPortalForwarders;
   DtForwarderGroupConnectionList myConnectedForwarders;

   DtRtiForwarderGroup::DtRtiForwarderGroupMember* myNextAvailablePortal;

protected:
   
   static DtForwarderGroupCreatorFcn theCreator;
};


inline void DtRtiForwarderGroup::setDefaultPort( int port )
{
   myDefaultPort = port;
}

inline const DtRtiForwarderGroup::DtForwarderGroupConnectionList& DtRtiForwarderGroup::portalConnections() const
{
   return myPortalForwarders;
}

inline const DtRtiForwarderGroup::DtForwarderGroupConnectionList& DtRtiForwarderGroup::connections() const
{
   return myConnectedForwarders;
}

inline unsigned int DtRtiForwarderGroup::portalCount() const
{
   return myPortalForwarders.size();
}

inline unsigned int DtRtiForwarderGroup::connectionCount() const
{
   return myConnectedForwarders.size();
}

inline unsigned int DtRtiForwarderGroup::memberCount() const
{
   return myRoster.size();
}

inline const MAKRti::DtString& DtRtiForwarderGroup::groupName() const
{
   return myName;
}

inline DtRtiForwarderGroup::DtRtiForwarderGroupMember* DtRtiForwarderGroup::nextAvailablePortal()
{
   if ( !myNextAvailablePortal )
   {
      myNextAvailablePortal = members().begin()->second;
   }
   return myNextAvailablePortal;
}

#endif //! DtForwarderGroup_H_