Weights.hpp

#pragma once
 
#include "util/StringHash.hpp"
#include "util/config/ConfigDefinition.hpp"
#include "web/dosguard/WeightsInterface.hpp"
 
#include <boost/json/object.hpp>
 
#include <cstddef>
#include <functional>
#include <optional>
#include <string>
#include <unordered_map>
 
namespace web::dosguard {

class Weights : public WeightsInterface {
public:
    struct Entry {
        size_t weight;
        std::optional<size_t> weightLedgerCurrent;
        std::optional<size_t> weightLedgerValidated;
    };
 
private:
    size_t defaultWeight_;
    std::unordered_map<std::string, Entry, util::StringHash, std::equal_to<>> weights_;
 
public:
    Weights(size_t defaultWeight, std::unordered_map<std::string, Entry> weights);

    static Weights
    make(util::config::ClioConfigDefinition const& config);

    size_t
    requestWeight(boost::json::object const& request) const override;
};
 
}  // namespace web::dosguard