subscription.h

Go to the documentation of this file.

 
#pragma once
#include <knp/core/uid.h>
 
#include <algorithm>
#include <string>
#include <unordered_set>
#include <vector>
 
#include <boost/functional/hash.hpp>
 
 
namespace knp::core
{
 
template <class MessageT>
class Subscription final
{
public:
    using MessageType = MessageT;
    using MessageContainerType = std::vector<MessageType>;
 
    using UidSet = std::unordered_set<::boost::uuids::uuid, boost::hash<boost::uuids::uuid>>;
    // Subscription(const Subscription &) = delete;
 
public:
    Subscription(const UID &receiver, const std::vector<UID> &senders) : receiver_(receiver) { add_senders(senders); }
 
    [[nodiscard]] const UidSet &get_senders() const { return senders_; }
 
    [[nodiscard]] UID get_receiver_uid() const { return receiver_; }
 
    size_t remove_sender(const UID &uid) { return senders_.erase(static_cast<boost::uuids::uuid>(uid)); }
 
    size_t add_sender(const UID &uid) { return senders_.insert(static_cast<boost::uuids::uuid>(uid)).second; }
 
    size_t add_senders(const std::vector<UID> &senders)
    {
        size_t size_before = senders_.size();
        std::copy(senders.begin(), senders.end(), std::inserter(senders_, senders_.end()));
        return senders_.size() - size_before;
    }
 
    [[nodiscard]] bool has_sender(const UID &uid) const
    {
        return senders_.find(static_cast<boost::uuids::uuid>(uid)) != senders_.end();
    }
 
public:
    void add_message(MessageType &&message) { messages_.push_back(message); }
    void add_message(const MessageType &message) { messages_.push_back(message); }
 
    MessageContainerType &get_messages() { return messages_; }
    const MessageContainerType &get_messages() const { return messages_; }
 
    void clear_messages() { messages_.clear(); }
 
private:
    const UID receiver_;
 
    std::unordered_set<::boost::uuids::uuid, boost::hash<boost::uuids::uuid>> senders_;
    MessageContainerType messages_;
};
 
}  // namespace knp::core