NodeInfo.h

Go to the documentation of this file.

 
#pragma once
#include "NodeInteractionResultCode.h"
#include "NodeInteractionsContainer.h"
#include "catapult/utils/Hashers.h"
#include "catapult/types.h"
#include <unordered_set>
#include <vector>
 
namespace catapult { namespace ionet {
 
#define NODE_SOURCE_LIST \
    /* Incoming connection from another node. */ \
    ENUM_VALUE(Dynamic_Incoming) \
    \
    /* Forwarded from a peer node. */ \
    ENUM_VALUE(Dynamic) \
    \
    /* Loaded from a local peers file. */ \
    ENUM_VALUE(Static) \
    \
    /* Local node. */ \
    ENUM_VALUE(Local)
 
#define ENUM_VALUE(LABEL) LABEL,
    enum class NodeSource : uint32_t {
        NODE_SOURCE_LIST
    };
#undef ENUM_VALUE
 
    std::ostream& operator<<(std::ostream& out, NodeSource value);
 
    struct ServiceIdentifier_tag {};
    using ServiceIdentifier = utils::BaseValue<uint32_t, ServiceIdentifier_tag>;
 
    struct ConnectionState {
    public:
        ConnectionState()
                : Age(0)
                , NumConsecutiveFailures(0)
                , BanAge(0)
        {}
 
    public:
        uint32_t Age;
 
        uint32_t NumConsecutiveFailures;
 
        uint32_t BanAge;
    };
 
    struct NodeInfo {
    public:
        using ServiceIdentifiers = std::unordered_set<ServiceIdentifier, utils::BaseValueHasher<ServiceIdentifier>>;
 
    public:
        explicit NodeInfo(NodeSource source);
 
    public:
        NodeSource source() const;
 
        NodeInteractions interactions(Timestamp timestamp) const;
 
        size_t numConnectionStates() const;
 
        ServiceIdentifiers services() const;
 
        bool hasActiveConnection() const;
 
        const ConnectionState* getConnectionState(ServiceIdentifier serviceId) const;
 
    public:
        void source(NodeSource source);
 
        void incrementSuccesses(Timestamp timestamp);
 
        void incrementFailures(Timestamp timestamp);
 
        ConnectionState& provisionConnectionState(ServiceIdentifier serviceId);
 
        void clearAge(ServiceIdentifier serviceId);
 
        void updateBan(ServiceIdentifier serviceId, uint32_t maxConnectionBanAge, uint32_t numConsecutiveFailuresBeforeBanning);
 
    private:
        NodeSource m_source;
        NodeInteractionsContainer m_interactions;
        std::vector<std::pair<ServiceIdentifier, ConnectionState>> m_connectionStates;
    };
}}