uhs/atomizer/atomizer/controller.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_ATOMIZER_CONTROLLER_H_
#define OPENCBDC_TX_SRC_ATOMIZER_CONTROLLER_H_
 
#include "atomizer_raft.hpp"
#include "uhs/atomizer/atomizer/block.hpp"
#include "util/common/config.hpp"
#include "util/network/connection_manager.hpp"
 
#include <memory>
 
namespace cbdc::atomizer {
    class controller {
      public:
        controller() = delete;
        controller(const controller&) = delete;
        auto operator=(const controller&) -> controller& = delete;
        controller(controller&&) = delete;
        auto operator=(controller&&) -> controller& = delete;
 
        controller(uint32_t atomizer_id,
                   const config::options& opts,
                   std::shared_ptr<logging::log> log);
 
        ~controller();
 
        auto init() -> bool;
 
      private:
        uint32_t m_atomizer_id;
        cbdc::config::options m_opts;
        std::shared_ptr<logging::log> m_logger;
 
        atomizer_raft m_raft_node;
        std::atomic_bool m_running{true};
 
        cbdc::network::connection_manager m_watchtower_network;
        cbdc::network::connection_manager m_atomizer_network;
 
        std::thread m_atomizer_server;
        std::thread m_tx_notify_thread;
        std::thread m_main_thread;
 
        blocking_queue<tx_notify_request> m_notification_queue;
        std::vector<std::thread> m_notification_threads;
 
        auto server_handler(cbdc::network::message_t&& pkt)
            -> std::optional<cbdc::buffer>;
        void tx_notify_handler();
        void main_handler();
        void raft_result_handler(raft::result_type& r,
                                 nuraft::ptr<std::exception>& err);
        void err_return_handler(raft::result_type& r,
                                nuraft::ptr<std::exception>& err);
        auto raft_callback(nuraft::cb_func::Type type,
                           nuraft::cb_func::Param* param)
            -> nuraft::cb_func::ReturnCode;
        void notification_consumer();
    };
}
 
#endif // OPENCBDC_TX_SRC_ATOMIZER_CONTROLLER_H_