v2.hpp

#pragma once
 
#include <atomic>
#include <optional>
 
#include <userver/congestion_control/limiter.hpp>
#include <userver/congestion_control/sensor.hpp>
#include <userver/utils/periodic_task.hpp>
#include <userver/utils/statistics/fwd.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace congestion_control::v2 {
 
struct Stats {
    std::atomic<bool> is_enabled{false};
    std::atomic<bool> is_fake_mode{false};
    std::atomic<int64_t> current_limit{0};
    std::atomic<int64_t> enabled_epochs{0};
};
 
void DumpMetric(utils::statistics::Writer& writer, const Stats& stats);
 
class Controller {
public:
    struct Config {
        bool fake_mode{false};
        bool enabled{true};
    };
 
    Controller(const std::string& name, v2::Sensor& sensor, Limiter& limiter, Stats& stats, const Config& config);
 
    virtual ~Controller() = default;
 
    void Start();
    void Stop();
 
    void Step();
 
    const std::string& GetName() const;
 
    void SetEnabled(bool enabled);
 
protected:
    struct LimitWithDetails {
        Limit limit;
        std::optional<std::string> sensor_details;
    };
 
    virtual LimitWithDetails Update(const v2::Sensor::Data& data) = 0;
 
    // NOLINTNEXTLINE(misc-non-private-member-variables-in-classes)
    std::optional<size_t> current_limit_;
 
private:
    std::string_view LogFakeMode() const;
 
    const std::string name_;
    Sensor& sensor_;
    congestion_control::Limiter& limiter_;
    Stats& stats_;
    const Config config_;
    USERVER_NAMESPACE::utils::PeriodicTask periodic_;
    std::atomic<bool> enabled_{true};
};
 
}  // namespace congestion_control::v2
 
USERVER_NAMESPACE_END