BasicEndpoint.h

Go to the documentation of this file.

#pragma once

#include <swarmio/Endpoint.h>
#include <swarmio/Mailbox.h>
#include <mutex>
#include <set>
#include <atomic>

namespace swarmio::transport
{
    class SWARMIO_API BasicEndpoint : public Endpoint
    {
        private:

            bool _isRunning;

            std::set<Mailbox*> _mailboxes;

            std::recursive_mutex _mutex;

            std::atomic<uint64_t> _counter;

            void ReplyWithError(const Node* sender, const data::Message* message, data::Error error);

        protected:

            BasicEndpoint() : _counter(1), _isRunning(false) { }

            virtual void Send(const void* data, size_t size, const Node* node) = 0;

            virtual bool ReceiveMessage(const Node* sender, const data::Message* message) noexcept;

            virtual bool ReceiveMessage(const Node* sender, const void* data, size_t size) noexcept;

            virtual void NodeWasDiscovered(const Node* node) noexcept;

            virtual void NodeDidJoin(const Node* node) noexcept;

            virtual void NodeWillLeave(const Node* node) noexcept;

            virtual void RegisterMailbox(Mailbox* mailbox) override;

            virtual void UnregisterMailbox(Mailbox* mailbox) override;

            virtual void ReplaceMailbox(Mailbox* oldMailbox, Mailbox* newMailbox) override;

        public:

            virtual void Start() override;

            virtual void Stop() override;

            bool IsRunning()
            {
                return _isRunning;
            }

            virtual void Send(data::Message* message, const Node* node) override;

            virtual void Tag(data::Message* message) override;
    };
}