peer.hpp

Go to the documentation of this file.

// Copyright (c) 2021 MIT Digital Currency Initiative,
//                    Federal Reserve Bank of Boston
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#ifndef OPENCBDC_TX_SRC_NETWORK_PEER_H_
#define OPENCBDC_TX_SRC_NETWORK_PEER_H_
 
#include "tcp_socket.hpp"
#include "util/common/blocking_queue.hpp"
 
#include <atomic>
#include <thread>
 
namespace cbdc::network {
    class peer {
      public:
        using callback_type
            = std::function<void(std::shared_ptr<cbdc::buffer>)>;
 
        peer(std::unique_ptr<tcp_socket> sock,
             callback_type cb,
             bool attempt_reconnect);
 
        ~peer();
 
        peer(const peer&) = delete;
        auto operator=(const peer&) -> peer& = delete;
 
        peer(peer&&) = delete;
        auto operator=(peer&&) -> peer& = delete;
 
        void send(const std::shared_ptr<cbdc::buffer>& data);
 
        void shutdown();
 
        [[nodiscard]] auto connected() const -> bool;
 
      private:
        std::unique_ptr<tcp_socket> m_sock;
 
        blocking_queue<std::shared_ptr<cbdc::buffer>> m_send_queue;
 
        std::thread m_recv_thread;
        std::thread m_send_thread;
 
        std::thread m_reconnect_thread;
        std::mutex m_reconnect_mut;
        std::condition_variable m_reconnect_cv;
        bool m_reconnect{false};
        bool m_attempt_reconnect{};
 
        std::atomic_bool m_running{true};
        std::atomic_bool m_shut_down{false};
 
        callback_type m_recv_cb;
 
        void do_send();
 
        void do_recv();
 
        void do_reconnect();
 
        void close();
 
        void signal_reconnect();
    };
}
 
#endif