proposed_blocks.hpp

Go to the documentation of this file.

#pragma once
 
#include <map>
#include <optional>
#include <shared_mutex>
 
#include "common/types.hpp"
#include "storage/storage.hpp"
 
namespace taraxa {
 
class PbftBlock;
class Vote;
 
class ProposedBlocks {
 public:
  ProposedBlocks(std::shared_ptr<DbStorage> db) : db_(db) {}
 
  bool pushProposedPbftBlock(const std::shared_ptr<PbftBlock>& proposed_block, bool save_to_db = true);
 
  void markBlockAsValid(const std::shared_ptr<PbftBlock>& proposed_block);
 
  std::optional<std::pair<std::shared_ptr<PbftBlock>, bool>> getPbftProposedBlock(PbftPeriod period,
                                                                                  const blk_hash_t& block_hash) const;
 
  bool isInProposedBlocks(PbftPeriod period, const blk_hash_t& block_hash) const;
 
  void cleanupProposedPbftBlocksByPeriod(PbftPeriod period);
 
  std::optional<std::string> checkOldBlocksPresence(PbftPeriod current_period) const;
 
  std::map<PbftPeriod, std::vector<std::shared_ptr<PbftBlock>>> getProposedBlocks() const;
 
 private:
  // <PBFT period, <block hash, [block, is_valid_flag]>>
  std::map<PbftPeriod, std::unordered_map<blk_hash_t, std::pair<std::shared_ptr<PbftBlock>, bool>>> proposed_blocks_;
  mutable std::shared_mutex proposed_blocks_mutex_;
  std::shared_ptr<DbStorage> db_;
};
 
}  // namespace taraxa