dag_block_proposer.hpp

Go to the documentation of this file.

#pragma once
 
#include <atomic>
#include <thread>
#include <vector>
 
#include "dag/dag_block.hpp"
#include "logger/logger.hpp"
#include "network/network.hpp"
 
namespace taraxa {
 
class TransactionManager;
class KeyManager;
class DagManager;
struct FullNodeConfig;
 
namespace final_chain {
class FinalChain;
}
 
class DagBlockProposer {
 public:
  DagBlockProposer(const FullNodeConfig& config, std::shared_ptr<DagManager> dag_mgr,
                   std::shared_ptr<TransactionManager> trx_mgr, std::shared_ptr<final_chain::FinalChain> final_chain,
                   std::shared_ptr<DbStorage> db, std::shared_ptr<KeyManager> key_manager);
  ~DagBlockProposer() { stop(); }
  DagBlockProposer(const DagBlockProposer&) = delete;
  DagBlockProposer(DagBlockProposer&&) = delete;
  DagBlockProposer& operator=(const DagBlockProposer&) = delete;
  DagBlockProposer& operator=(DagBlockProposer&&) = delete;
 
  void start();
 
  void stop();
 
  bool proposeDagBlock();
 
  void setNetwork(std::weak_ptr<Network> network);
 
  uint64_t getProposedBlocksCount() const { return proposed_blocks_count_; }
 
  vec_blk_t selectDagBlockTips(const vec_blk_t& frontier_tips, uint64_t gas_limit) const;
 
 private:
  DagBlock createDagBlock(DagFrontier&& frontier, level_t level, const SharedTransactions& trxs,
                          std::vector<uint64_t>&& estimations, VdfSortition&& vdf) const;
 
  std::pair<SharedTransactions, std::vector<uint64_t>> getShardedTrxs(PbftPeriod proposal_period,
                                                                      uint64_t weight_limit) const;
 
  level_t getProposeLevel(blk_hash_t const& pivot, vec_blk_t const& tips) const;
 
  bool isValidDposProposer(PbftPeriod propose_period) const;
 
 private:
  uint16_t max_num_tries_{20};  // Wait 2000(ms)
  uint16_t num_tries_{0};
  uint64_t last_propose_level_{0};
  util::ThreadPool executor_{1};
 
  std::atomic<uint64_t> proposed_blocks_count_{0};
  std::atomic<bool> stopped_{true};
 
  const DagBlockProposerConfig bp_config_;
  const uint16_t total_trx_shards_;
  uint16_t my_trx_shard_;
 
  std::shared_ptr<DagManager> dag_mgr_;
  std::shared_ptr<TransactionManager> trx_mgr_;
  std::shared_ptr<final_chain::FinalChain> final_chain_;
  std::shared_ptr<KeyManager> key_manager_;
  std::shared_ptr<DbStorage> db_;
  std::shared_ptr<std::thread> proposer_worker_;
  std::weak_ptr<Network> network_;
 
  const addr_t node_addr_;
  const secret_t node_sk_;
  const vrf_wrapper::vrf_sk_t vrf_sk_;
  const vrf_wrapper::vrf_pk_t vrf_pk_;
 
  const uint64_t kPbftGasLimit;
  const uint64_t kDagGasLimit;
 
  const HardforksConfig kHardforks;
  const uint64_t kValidatorMaxVote;
 
  LOG_OBJECTS_DEFINE
};
 
}  // namespace taraxa