ResponseTracker.h

/*
 * Copyright 2003 Michael A. Marsh, Cornell University. All rights reserved.
 * This software is released under the modified BSD license.
 * See the file LICENSE in the top-level directory for details.
 */
//
// $Id: ResponseTracker.h,v 1.4 2005/01/21 19:44:17 mmarsh Exp $
//
// $Log: ResponseTracker.h,v $
// Revision 1.4  2005/01/21 19:44:17  mmarsh
// Updated for compatibility with Doxygen 1.4.1
//
// Revision 1.3  2004/05/19 15:56:56  mmarsh
// *** empty log message ***
//
// Revision 1.2  2003/11/04 22:17:23  mmarsh
// General code cleanup.
//
//

#ifndef __CODEX_QUORUM_RESPONSETRACKER_H__
#define __CODEX_QUORUM_RESPONSETRACKER_H__

#include "ResponseInfo.h"

namespace CODEX_Quorum
{
   template <size_t _Tn>
   struct memless : public binary_function<const void*,const void*,bool>
   {
         bool operator()(const void* __x, const void* __y) const {
            return memcmp(__x,__y,_Tn) < 0; }
   };

   class ResponseTracker
   {
      public :
         virtual ~ResponseTracker() {}

         virtual void insert(unsigned char* key,ResponseInfo* m) = 0;

         virtual bool remove(const unsigned char* key, bool del=false) = 0;

         virtual bool check(const unsigned char* data,
                            int length,
                            unsigned int server)
            = 0;

         virtual ResponseInfo* operator()(const unsigned char* msgID) = 0;

         virtual unsigned char* extractKey(const unsigned char* data) = 0;
   };

   template< size_t ST, size_t MT >
   class ConcreteResponseTracker : public ResponseTracker
   {
      public :
         typedef map< const unsigned char*, ResponseInfo*, memless< MT > >
         MessageResponseMap;
         typedef typename MessageResponseMap::iterator  MessageResponseMapItr;

         ConcreteResponseTracker( unsigned char* self ) :
            m_self( self )
         {
         }

         virtual ~ConcreteResponseTracker() {
            delete [] m_self;
            while( m_map.size() > 0 )
            {
               MessageResponseMapItr item = m_map.begin();
               m_map.erase(item);
               delete [] item->first;
               delete item->second;
            }
         }

         void insert(unsigned char* key,ResponseInfo* m) {
            m_map.insert(MessageResponseMap::value_type(key,m));
         }

         bool remove(const unsigned char* key, bool del=false) {
            MessageResponseMapItr item = m_map.find(key);
            if ( m_map.end() == item )
            {
               return false;
            }
            m_map.erase(item);
            // We always need to delete the key, since we have the only
            // handle to it.
            delete [] item->first;
            if ( del )
            {
               delete item->second;
            }
            return true;
         }

         bool check( const unsigned char* data,
                     int length,
                     unsigned int server ) {
            // Is this a response to a request from this server?
            if ( 0 != memcmp(data,m_self,ST) )
            {
               return false;
            }
            // Has the request been registered to collect responses?
            MessageResponseMapItr item = m_map.find(data+ST);
            if ( item == m_map.end() )
            {
               return false;
            }
            // Pass the message along to the ResponseInfo object.
            Message* pMsg = new Message();
            pMsg->fill( data+ST+MT, length-ST-MT );
            item->second->add(server,pMsg);
            return true;
         }

         ResponseInfo* operator()(const unsigned char* msgID) {
            MessageResponseMapItr item = m_map.find(msgID);
            if ( item == m_map.end() )
            {
               return 0;
            }
            return item->second;
         }

         unsigned char* extractKey(const unsigned char* data) {
            unsigned char* retVal = new unsigned char[MT];
            memcpy(retVal,data+ST,MT);
            return retVal;
         }

      private :
         unsigned char*      m_self;
         MessageResponseMap  m_map;
   };

}

#endif /* __CODEX_QUORUM_RESPONSETRACKER_H__ */

---