federateCounterProcessor.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2005 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/
/*********************************************************************
** $RCSfile: federateCounterProcessor.h,v $ $Revision: 1.4 $ $State: Exp $
*********************************************************************/

#ifndef DtFederateCounterProcessor_H_
#define DtFederateCounterProcessor_H_

#include "forwarderNode.h"
#include "forwarderCounterMsg.h"
#include "routeMap.h"
#include "rtiMsConfig.h"

#include <vlutil/vlUtil.h>
#include <map>
#include <set>

class DtFederateCounterFilter;
class DtRtiExecDisconnectFilter;

class DT_DLL_FORWARDER DtFederateCounterProcessor
{
public:
   typedef bool (DtFederateCounterProcessor::* CounterMsgHandlerMemberCb)(
      const DtInetAddrAndPort& src, const DtForwarderCounterMsg& msg  );
   typedef bool (DtFederateCounterProcessor::* DestinationProcessingMemberCb)(
      const DtInetAddrAndPort& src, DtForwarderPort::DtAddressPortList& destinations );

   DtFederateCounterProcessor( DtBoost::shared_ptr<DtForwarderNode> parent );
   
   virtual ~DtFederateCounterProcessor();

   virtual bool gotMsgToOutPort( const DtInetAddrAndPort& src,
                                 const DtForwarderCounterMsg& msg );

   virtual bool processCounterMsgDestinations ( const DtInetAddrAndPort& src, 
      DtForwarderPort::DtAddressPortList& destinations );

public:

   static int lostConnection( const MAKRti::DtString& connName, const MAKRti::DtString& hostName,
      unsigned int connID, const std::pair<DtInetAddrAndPort, DtInetAddrAndPort>& connAddr, void* usr );

   static int reconnectBegun(const MAKRti::DtString& connName, const MAKRti::DtString& hostName,
      unsigned int connID, const DtInetAddrAndPort& connAddr, void* usr);

protected:

   virtual int lostConnection( const MAKRti::DtString& connName, const MAKRti::DtString& hostName,
      unsigned int connID, const std::pair<DtInetAddrAndPort, DtInetAddrAndPort>& connAddr);

   virtual int reconnectBegun(const MAKRti::DtString& connName, const MAKRti::DtString& hostName,
      unsigned int connID, const DtInetAddrAndPort& connAddr);

   virtual void sendResponseMsg( const DtInetAddrAndPort& destination,
      unsigned int federateCount, const DtInetAddrAndPort* addr = 0 );

   virtual void requestDownstreamCount( const DtInetAddrAndPort& destination );   
   virtual void requestDownstreamCount( const DtInetAddrAndPort& destination,
                                        const DtRouteMap::DtAddressPortList& destinations );

   virtual bool processInToOutMsg( const DtInetAddrAndPort& src,
      const DtForwarderCounterMsg& msg );
   
   virtual bool processLanToOutMsg( const DtInetAddrAndPort& src,
      const DtForwarderCounterMsg& msg );

   
protected:

   class PendingRequest
   {
   public:
      PendingRequest() :
         currentCount( 0 ),
         destinationAddress(),
         sourceAddress(),
         pendingOutResponses() {}
         
      PendingRequest( const DtInetAddrAndPort& src, const DtInetAddrAndPort& dest,
                      const DtRouteMap::DtAddressPortList& expectedOutResponses ) :
         currentCount( 0 ),
         destinationAddress( dest ),
         sourceAddress( src ),
         pendingOutResponses( expectedOutResponses ) {}
         
      unsigned int                  currentCount;
      DtInetAddrAndPort             destinationAddress;
      DtInetAddrAndPort             sourceAddress;
      DtRouteMap::DtAddressPortList pendingOutResponses;
   };

   typedef std::map< DtInetAddrAndPort, PendingRequest > DtDestToRequestMap;
   
protected:

   DtBoost::shared_ptr<DtForwarderNode> myParent;
   DtFederateCounterFilter* myMsgToOutFilter;

   DtDestToRequestMap       myPendingRequests;
   
};

#endif