pool_moderator_impl.hpp

Go to the documentation of this file.

#ifndef KAGOME_POOL_MODERATOR_IMPL_HPP
#define KAGOME_POOL_MODERATOR_IMPL_HPP

#include "transaction_pool/pool_moderator.hpp"

#include <map>

#include "clock/clock.hpp"

namespace kagome::transaction_pool {

  class PoolModeratorImpl : public PoolModerator {
    using Map = std::map<common::Hash256, clock::SystemClock::TimePoint>;

   public:
    static constexpr size_t kDefaultExpectedSize = 2048;

    static constexpr clock::SystemClock::Duration kDefaultBanFor =
        std::chrono::minutes(30);

    struct Params {
      clock::SystemClock::Duration ban_for = kDefaultBanFor;
      size_t expected_size = kDefaultExpectedSize;
    };

    PoolModeratorImpl(std::shared_ptr<clock::SystemClock> clock,
                      Params parameters);

    ~PoolModeratorImpl() override = default;

    void ban(const common::Hash256 &tx_hash) override;

    bool banIfStale(primitives::BlockNumber current_block,
                    const Transaction &tx) override;

    bool isBanned(const common::Hash256 &tx_hash) const override;

    void updateBan() override;

    size_t bannedNum() const override;

   private:
    std::shared_ptr<clock::SystemClock> clock_;
    Params params_;
    Map banned_until_;
  };

}  // namespace kagome::transaction_pool

#endif  // KAGOME_POOL_MODERATOR_IMPL_HPP