PeersConnectionTasks.h

Go to the documentation of this file.

 
#pragma once
#include "NodeSelector.h"
#include "catapult/config/NodeConfiguration.h"
#include "catapult/ionet/NodeContainer.h"
#include "catapult/thread/Task.h"
#include <vector>
 
namespace catapult {
    namespace cache { class CatapultCache; }
    namespace net {
        class ConnectionContainer;
        class PacketWriters;
    }
}
 
namespace catapult { namespace extensions {
 
    // region NodeAger
 
    // A node ager.
    using NodeAger = consumer<const utils::KeySet&>;
 
    NodeAger CreateNodeAger(
            ionet::ServiceIdentifier serviceId,
            const config::NodeConfiguration::ConnectionsSubConfiguration& config,
            ionet::NodeContainer& nodes);
 
    // endregion
 
    // region SelectorSettings
 
    struct SelectorSettings {
    public:
        SelectorSettings(
                const cache::CatapultCache& cache,
                Importance totalChainImportance,
                ionet::NodeContainer& nodes,
                ionet::ServiceIdentifier serviceId,
                ionet::NodeRoles requiredRole,
                const config::NodeConfiguration::ConnectionsSubConfiguration& config);
 
        SelectorSettings(
                const cache::CatapultCache& cache,
                Importance totalChainImportance,
                ionet::NodeContainer& nodes,
                ionet::ServiceIdentifier serviceId,
                const config::NodeConfiguration::ConnectionsSubConfiguration& config);
 
    public:
        ionet::NodeContainer& Nodes;
 
        ionet::ServiceIdentifier ServiceId;
 
        ionet::NodeRoles RequiredRole;
 
        config::NodeConfiguration::ConnectionsSubConfiguration Config;
 
        extensions::ImportanceRetriever ImportanceRetriever;
    };
 
    // endregion
 
    // region NodeSelector / ConnectPeersTask
 
    using NodeSelector = supplier<NodeSelectionResult>;
 
    NodeSelector CreateNodeSelector(const SelectorSettings& settings);
 
    thread::Task CreateConnectPeersTask(const SelectorSettings& settings, net::PacketWriters& packetWriters);
 
    thread::Task CreateConnectPeersTask(const SelectorSettings& settings, net::PacketWriters& packetWriters, const NodeSelector& selector);
 
    // endregion
 
    // region RemoveOnlyNodeSelector / AgePeersTask
 
    using RemoveOnlyNodeSelector = supplier<utils::KeySet>;
 
    RemoveOnlyNodeSelector CreateRemoveOnlyNodeSelector(const SelectorSettings& settings);
 
    thread::Task CreateAgePeersTask(const SelectorSettings& settings, net::ConnectionContainer& connectionContainer);
 
    thread::Task CreateAgePeersTask(
            const SelectorSettings& settings,
            net::ConnectionContainer& connectionContainer,
            const RemoveOnlyNodeSelector& selector);
 
    // endregion
}}