Connector.hpp

Go to the documentation of this file.

/***************************************************
 *       _   _            _                        *
 *      | | | |          (_)                       *
 *      | |_| | ___  _ __ _ _______  _ __          *
 *      |  _  |/ _ \| '__| |_  / _ \| '_  \        *
 *      | | | | (_) | |  | |/ / (_) | | | |        *
 *      \_| |_/\___/|_|  |_/___\___/|_| |_|        *
 ***************************************************
 * This file is part of Horizon (c).
 *
 * Copyright (c) 2019 Sagun K. (sagunxp@gmail.com).
 * Copyright (c) 2019 Horizon Dev Team.
 *
 * Base Author - Sagun K. (sagunxp@gmail.com)
 *
 * This library is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this library.  If not, see <http://www.gnu.org/licenses/>.
 **************************************************/
 
#ifndef HORIZON_NETWORKING_CONNECTOR_HPP
#define HORIZON_NETWORKING_CONNECTOR_HPP
 
#include "Core/Logging/Logger.hpp"
#include "Server/Common/Server.hpp"
 
#include <boost/asio.hpp>
#include <boost/bind/bind.hpp>
#include <thread>
#include <functional>
 
using namespace boost::asio::ip;
 
namespace Horizon
{
namespace Networking
{
class Connector
{
public:
    typedef std::function<void(std::string &, std::shared_ptr<tcp::socket>, uint32_t)> ConnectorCallback;
 
    Connector(std::string const &connection_name, Server *server, std::string const &connect_ip, uint16_t port)
    : server(server), _connection_name(connection_name), _endpoint(boost::asio::ip::address::from_string(connect_ip), port),
    _socket_factory(std::bind(&Connector::default_socket_factory, this))
    {
    }
 
    Connector()
    {
    }
 
    void connect_with_callback(ConnectorCallback callback, int connections = 1)
    {
        for (int i = 0; i < connections; i++) {
            std::shared_ptr<tcp::socket> socket;
            uint32_t network_thread_idx;
            boost::system::error_code error;
 
            // Get a new socket from a thread with the minimum connections.
            std::tie(socket, network_thread_idx) = _socket_factory();
 
            do {
                HLog(info) << "Trying to establish connection for '" << _connection_name << "' at tcp://" << _endpoint.address().to_string() << ":" << _endpoint.port();
 
                // Try connecting to the endpoint.
                socket->connect(_endpoint, error);
 
                if (error.value() != 0) {
                    HLog(info) << "Error connecting to '" << _connection_name << "' at tcp://" << _endpoint.address().to_string() << ":" << _endpoint.port();
                    std::this_thread::sleep_for(std::chrono::seconds(10));
                    socket->close();
                    error.clear();
                } else {
                    callback(_connection_name, socket, network_thread_idx);
                    HLog(info) << "Successfully connected to '" << _connection_name << "' at tcp://" << _endpoint.address().to_string() << ":" << _endpoint.port();
                }
            } while (!socket->is_open());
        }
    }
 
    void set_socket_factory(std::function<std::pair<std::shared_ptr<tcp::socket>, uint32_t>()> &&func) { _socket_factory = func; }
 
private:
    std::pair<std::shared_ptr<tcp::socket>, uint32_t> default_socket_factory() { return std::make_pair(nullptr, 0); }
 
    Server *server;
    std::string _connection_name;
    tcp::endpoint _endpoint;
    std::function<std::pair<std::shared_ptr<tcp::socket>, uint32_t>()> _socket_factory;
};
}
}
 
#endif //HORIZON_NETWORKCONNECTOR_HPP