RTFXPostOffice.h

Go to the documentation of this file.

#ifndef __RTFX_POSTOFFICE__
#define __RTFX_POSTOFFICE__

#include <set>
#include <queue>
#include <boost/bind.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/asio.hpp>

#include "RTFXDebug.h"

#include "RTFXParcel.h"
#include "RTFXParticipant.h"
#include "RTFXMailbox.h"
#include "RTFXBoostIncludes.h"


using boost::asio::ip::tcp;

namespace RTFX
{
    typedef boost::shared_ptr<RTFXParticipant> RTFXParticipant_ptr;
    typedef std::deque<RTFXParcel> RTFXParcelQueue;

    struct MailboxNameComparison
    {
        bool operator() ( const char * a, const char * b ) const
        {
            if ( memcmp( a, b, 16 ) < 0 ) return true;
            return false;
        }
    };

    class RTFXAPI RTFXPostOffice
    {
        public:

            RTFXPostOffice()
            {
                debug( ostringstream().flush() << "RTFXPostOffice Constructor" << endl );
                compressionType = NONE;
            }

            ~RTFXPostOffice()
            {
                map<const char *, RTFXMailbox *, MailboxNameComparison>::iterator iter_ = mailboxes.begin();
                while ( iter_ != mailboxes.end() )
                {
                    RTFXMailbox * ptr_ = ( *iter_ ).second;
                    mailboxes.erase( iter_ );
                    delete ptr_;

                    iter_ ++;
                }
            }

            void Join( RTFXParticipant_ptr _participant );

            void Leave( RTFXParticipant_ptr _particpant );

            void Deliver( RTFXParcel &_parcel );

            void RemoveAll( boost::uuids::uuid _owner, char * _mailboxName );

            void SendMailbox( RTFXParticipant_ptr _participant );

            RTFXMailbox * GetMailbox( char * _mailboxName )
            {
                if ( mailboxes.find( _mailboxName ) == mailboxes.end() )
                    return NULL;
                return mailboxes[ _mailboxName ];
            }

            void SetDefaultCompressionType( COMPRESSIONTYPE _tmp )
            {
                compressionType = _tmp;
                map<const char *, RTFXMailbox *, MailboxNameComparison >::iterator iter_ = mailboxes.begin();
                while ( iter_ != mailboxes.end() )
                {
                    ((*iter_).second)->SetDefaultCompressionType( compressionType );
                    iter_ ++;
                }
            }

            void SetOwnerID( boost::uuids::uuid _id )
            {
                ownerID = _id;
            }

            boost::uuids::uuid GetOwnerID()
            {
                return ownerID;
            }

        private:
            boost::uuids::uuid ownerID;
        
            COMPRESSIONTYPE compressionType;

            std::string postOfficeName;

            boost::uuids::uuid id;

            std::set<RTFXParticipant_ptr> participants;

            std::map<const char *, RTFXMailbox *, MailboxNameComparison > mailboxes;

            RTFXParticipant_ptr newParticipant;

            RTFXParcelQueue availableData;

            bool isBinaryData;
    };
}

#endif