atomizer_raft.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_ATOMIZER_RAFT_H_
#define OPENCBDC_TX_SRC_ATOMIZER_ATOMIZER_RAFT_H_
 
#include "messages.hpp"
#include "state_machine.hpp"
#include "util/network/connection_manager.hpp"
#include "util/raft/node.hpp"
#include "util/raft/state_manager.hpp"
 
namespace cbdc::atomizer {
    class atomizer_raft : public cbdc::raft::node {
      public:
        atomizer_raft(uint32_t atomizer_id,
                      std::vector<network::endpoint_t> raft_endpoints,
                      size_t stxo_cache_depth,
                      std::shared_ptr<logging::log> logger,
                      config::options opts,
                      nuraft::cb_func::func_type raft_callback);
 
        auto make_request(const state_machine::request& r,
                          const raft::callback_type& result_fn) -> bool;
 
        [[nodiscard]] auto get_sm() -> state_machine*;
 
        [[nodiscard]] auto tx_notify_count() -> uint64_t;
 
        void tx_notify(tx_notify_request&& notif);
 
        [[nodiscard]] auto
        send_complete_txs(const raft::callback_type& result_fn) -> bool;
 
      private:
        static constexpr const auto m_node_type = "atomizer";
 
        using attestation = std::pair<uint64_t, uint64_t>;
 
        struct attestation_hash {
            auto operator()(const attestation& pair) const -> size_t;
        };
 
        struct attestation_cmp {
            auto operator()(const attestation& a, const attestation& b) const
                -> bool;
        };
 
        using attestation_set = std::
            unordered_set<attestation, attestation_hash, attestation_cmp>;
 
        std::unordered_map<transaction::compact_tx,
                           attestation_set,
                           transaction::compact_tx_hasher>
            m_txs;
        std::mutex m_complete_mut;
        std::vector<aggregate_tx_notification> m_complete_txs;
        std::shared_ptr<logging::log> m_log;
        config::options m_opts;
 
        std::mutex m_txs_mut;
    };
}
 
#endif // OPENCBDC_TX_SRC_ATOMIZER_ATOMIZER_RAFT_H_