CorruptionDetector.hpp

#pragma once
 
#include "data/LedgerCacheInterface.hpp"
#include "etl/SystemState.hpp"
#include "util/log/Logger.hpp"
 
#include <functional>
 
namespace etl {

class CorruptionDetector {
    std::reference_wrapper<SystemState> state_;
    std::reference_wrapper<data::LedgerCacheInterface> cache_;
 
    util::Logger log_{"ETL"};
 
public:
    CorruptionDetector(SystemState& state, data::LedgerCacheInterface& cache)
        : state_{std::ref(state)}, cache_{std::ref(cache)}
    {
    }

    void
    onCorruptionDetected()
    {
        if (not state_.get().isCorruptionDetected) {
            state_.get().isCorruptionDetected = true;
 
            LOG(
                log_.error()
            ) << "Disabling the cache to avoid corrupting the DB further. Please investigate.";
            cache_.get().setDisabled();
        }
    }
};
 
}  // namespace etl