netStatsConnMgr.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2012 MaK Technologies, Inc.
** All rights reserved.
*******************************************************************************/

#ifndef netStatsConnMgr_h__
#define netStatsConnMgr_h__


#include <vlutil/vlThreadSafe.h>
#include <asyConMgr.h>

// Forward References
class DtNetStatsMonitor;
class MAKRti::DtClock;
class DtRIDParameters;
class DtFederateMgr;

// The DtNetStatsAsyncConnMgr class adds statistics gathering capabilities
// to the DtAsyncConnectionMgr class.

class DtNetStatsAsyncConnMgr : public DtAsyncConnectionMgr
{
protected:

   DtNetStatsAsyncConnMgr( DtRIDParameters* params, bool isFederate = false, 
      std::shared_ptr<DtRtiAssistantLrcClient> assistantClient = 
      std::shared_ptr<DtRtiAssistantLrcClient>() );

public:

   virtual ~DtNetStatsAsyncConnMgr();

   // Set the handle to the Network Traffic Monitor
   virtual void setTrafficMonitor( DtNetStatsMonitor* stats );


   // Sending

   // Send a message
   // Returns the size of the data sent
   virtual int send( const DtRtiMsg& msg, 
                     DtTransportType transport,
                     const DtInetAddr& addr = DtInetAddr::inaddrAny() );

   // Send a message and delete the message after it has been sent
   virtual int sendAndDelete( DtRtiMsg* msg, 
                              DtTransportType transport,
                              const DtInetAddr& addr = DtInetAddr::inaddrAny() );


   // Receiving

   // Record received messages here for synch. I/O
   // Read from the given connection and process the message (if received)
   // Return boolean: message was received? (i.e. could be more input).
   virtual bool processConnection( DtRtiConnection* conn,
                                   DtTransportType trans);

   // Get a message if one exists
   // Message must be destroyed by caller
   virtual DtRtiMsg* getMessage();


   // Record received messages here for asynch. I/O
   // this function applies the callbacks to the message
   virtual DtNetReadStatus processMsg( const DtRtiMsg& msg );

   virtual void shutdown();

   virtual bool netStatsEnabled() const;
   virtual void setNetStatsEnabled(bool shouldEnable);

public:
 
   // Creates an instance of a DtAsyncConnectionMgr;  If an alternate
   // creator function has been set, that creator function will be called
   // from within this function.
   static DtConnectionMgr* create( DtRIDParameters* p, bool isFederate = false, 
      std::shared_ptr<DtRtiAssistantLrcClient> assistantClient = 
      std::shared_ptr<DtRtiAssistantLrcClient>() );
   
  

protected:

   // Statistics Log
   DtNetStatsMonitor* myNetStats;

   // Network Call Timer
   MAKRti::DtClock   myTimer;

   // Enable/Disable flag for network statistics
   bool myNetStatsEnabled;

};
#endif // netStatsConnMgr_h__