NetworkUtils.h

Go to the documentation of this file.

 
#pragma once
#include "catapult/config/CatapultConfiguration.h"
#include "catapult/net/AsyncTcpServer.h"
#include "catapult/net/ConnectionSettings.h"
#include "catapult/net/PeerConnectResult.h"
#include "catapult/subscribers/NodeSubscriber.h"
#include "catapult/thread/MultiServicePool.h"
 
namespace catapult { namespace extensions {
 
    net::ConnectionSettings GetConnectionSettings(const config::CatapultConfiguration& config);
 
    void UpdateAsyncTcpServerSettings(net::AsyncTcpServerSettings& settings, const config::CatapultConfiguration& config);
 
    uint32_t GetMaxIncomingConnectionsPerIdentity(ionet::NodeRoles roles);
 
    template<typename TAcceptor>
    std::shared_ptr<net::AsyncTcpServer> BootServer(
            thread::MultiServicePool::ServiceGroup& serviceGroup,
            unsigned short port,
            ionet::ServiceIdentifier serviceId,
            const config::CatapultConfiguration& config,
            subscribers::NodeSubscriber& nodeSubscriber,
            TAcceptor acceptor) {
        auto endpoint = boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), port);
        auto settings = net::AsyncTcpServerSettings([acceptor, port, serviceId, &nodeSubscriber](const auto& socketInfo) {
            acceptor(socketInfo, [port, serviceId, &nodeSubscriber](const auto& connectResult) {
                CATAPULT_LOG(info) << "accept result to local node port " << port << ": " << connectResult.Code;
                if (net::PeerConnectCode::Accepted == connectResult.Code)
                    nodeSubscriber.notifyIncomingNode(connectResult.IdentityKey, serviceId);
            });
        });
 
        UpdateAsyncTcpServerSettings(settings, config);
        return serviceGroup.pushService(net::CreateAsyncTcpServer, endpoint, settings);
    }
}}