Mailbox.h

Go to the documentation of this file.

#pragma once

#include <swarmio/Node.h>
#include <swarmio/Endpoint.h>
#include <swarmio/data/Message.pb.h>

namespace swarmio
{
    class SWARMIO_API Mailbox
    {
        private:

            Endpoint* _endpoint;

        protected:

            Mailbox()
                : _endpoint(nullptr) { }

            Mailbox(Endpoint* endpoint)
            {
                _endpoint = endpoint;
            }

            void FinishConstruction()
            {
                if (_endpoint != nullptr)
                {
                    _endpoint->RegisterMailbox(this);
                }
            }
            
            Mailbox(Mailbox&& other)
            {
                // Copy over endpoint reference
                _endpoint = other.GetEndpoint();
            }

            void FinishMovingTo(Mailbox* other)
            {
                if (_endpoint != nullptr)
                {
                    _endpoint->ReplaceMailbox(this, other);
                    _endpoint = nullptr;
                }
            }

        public:

            Mailbox(const Mailbox&) = delete;

            Mailbox& operator=(const Mailbox&) = delete;

            virtual bool ReceiveMessage(const Node* sender, const data::Message* message) { return false; }

            virtual void NodeWasDiscovered(const Node* node) noexcept { }

            virtual void NodeDidJoin(const Node* node) noexcept { }

            virtual void NodeWillLeave(const Node* node) noexcept { }

            virtual void MailboxWasConnected() noexcept { }

            virtual void MailboxWillBeDisconnected() noexcept { }

            Endpoint* GetEndpoint()
            {
                return _endpoint;
            }

            virtual void Disconnect()
            {
                if (_endpoint != nullptr)
                {
                    _endpoint->UnregisterMailbox(this);
                    _endpoint = nullptr;
                }
            }

            virtual ~Mailbox() 
            {
                Disconnect();
            }
    };
}