NodeContainer.h

Go to the documentation of this file.

 
#pragma once
#include "Node.h"
#include "NodeInfo.h"
#include "catapult/utils/ArraySet.h"
#include "catapult/utils/SpinReaderWriterLock.h"
#include <unordered_map>
 
namespace catapult {
    namespace ionet {
        struct NodeContainerData;
        struct NodeData;
        struct NodeInteractionResult;
    }
}
 
namespace catapult { namespace ionet {
 
    class NodeContainerView : utils::MoveOnly {
    public:
        NodeContainerView(const NodeContainerData& nodeContainerData, utils::SpinReaderWriterLock::ReaderLockGuard&& readLock);
 
    public:
        size_t size() const;
 
        Timestamp time() const;
 
        bool contains(const Key& identityKey) const;
 
        const NodeInfo& getNodeInfo(const Key& identityKey) const;
 
        void forEach(const consumer<const Node&, const NodeInfo&>& consumer) const;
 
    private:
        const NodeContainerData& m_nodeContainerData;
        utils::SpinReaderWriterLock::ReaderLockGuard m_readLock;
    };
 
    class NodeContainerModifier : utils::MoveOnly {
    public:
        NodeContainerModifier(NodeContainerData& nodeContainerData, utils::SpinReaderWriterLock::ReaderLockGuard&& readLock);
 
    public:
        bool add(const Node& node, NodeSource source);
 
        void addConnectionStates(ServiceIdentifier serviceId, ionet::NodeRoles role);
 
        ConnectionState& provisionConnectionState(ServiceIdentifier serviceId, const Key& identityKey);
 
        void ageConnections(ServiceIdentifier serviceId, const utils::KeySet& identities);
 
        void ageConnectionBans(ServiceIdentifier serviceId, uint32_t maxConnectionBanAge, uint32_t numConsecutiveFailuresBeforeBanning);
 
        void incrementSuccesses(const Key& identityKey);
 
        void incrementFailures(const Key& identityKey);
 
    private:
        void autoProvisionConnectionStates(NodeData& nodeData);
 
        bool ensureAtLeastOneEmptySlot();
 
        void incrementInteraction(const Key& identityKey, const consumer<NodeInfo&>& incrementer);
 
    private:
        NodeContainerData& m_nodeContainerData;
        utils::SpinReaderWriterLock::ReaderLockGuard m_readLock;
        utils::SpinReaderWriterLock::WriterLockGuard m_writeLock;
    };
 
    class NodeContainer {
    public:
        NodeContainer();
 
        NodeContainer(size_t maxNodes, const supplier<Timestamp>& timeSupplier);
 
        ~NodeContainer();
 
    public:
        NodeContainerView view() const;
 
        NodeContainerModifier modifier();
 
    private:
        std::unique_ptr<NodeContainerData> m_pImpl;
        mutable utils::SpinReaderWriterLock m_lock;
    };
 
    NodeSet FindAllActiveNodes(const NodeContainerView& view);
}}