consistency_keeper.hpp

Go to the documentation of this file.

#ifndef KAGOME_CONSENSUS_BABE_CONSISTENCYKEEPER
#define KAGOME_CONSENSUS_BABE_CONSISTENCYKEEPER

#include "primitives/common.hpp"

namespace kagome::consensus::babe {

  class ConsistencyKeeper {
   public:
    class Guard final {
     public:
      Guard(ConsistencyKeeper &keeper, primitives::BlockInfo block)
          : keeper_(keeper), block_(block){};
      ~Guard() {
        rollback();
      }

      void commit() {
        if (block_.has_value()) {
          keeper_.commit(block_.value());
          block_.reset();
        }
      }
      void rollback() {
        if (block_.has_value()) {
          keeper_.rollback(block_.value());
          block_.reset();
        }
      }

     private:
      ConsistencyKeeper &keeper_;
      std::optional<primitives::BlockInfo> block_;
    };

    virtual ~ConsistencyKeeper() = default;

    virtual Guard start(primitives::BlockInfo block) = 0;

   protected:
    virtual void commit(primitives::BlockInfo block) = 0;
    virtual void rollback(primitives::BlockInfo block) = 0;
  };

}  // namespace kagome::consensus::babe

#endif  // KAGOME_CONSENSUS_BABE_CONSISTENCYKEEPER