reputation_repository_impl.hpp

Go to the documentation of this file.

#ifndef KAGOME_NETWORK_REPUTATIONREPOSITORYIMPL
#define KAGOME_NETWORK_REPUTATIONREPOSITORYIMPL

#include "network/reputation_repository.hpp"

#include <memory>
#include <unordered_map>

#include <libp2p/basic/scheduler.hpp>

#include "log/logger.hpp"

namespace kagome::network {

  class ReputationRepositoryImpl
      : public ReputationRepository,
        public std::enable_shared_from_this<ReputationRepositoryImpl> {
   public:
    ReputationRepositoryImpl(
        std::shared_ptr<libp2p::basic::Scheduler> scheduler);

    Reputation reputation(const PeerId &peer_id) const override;

    Reputation change(const PeerId &peer_id, ReputationChange diff) override;

    Reputation changeForATime(const PeerId &peer_id,
                              ReputationChange diff,
                              std::chrono::seconds duration) override;

   private:
    void tick();

    std::shared_ptr<libp2p::basic::Scheduler> scheduler_;
    std::unordered_map<PeerId, Reputation> reputation_table_;

    libp2p::basic::Scheduler::Handle tick_handler_;

    log::Logger log_;
  };

}  // namespace kagome::network

#endif  // KAGOME_NETWORK_REPUTATIONREPOSITORYIMPL