sensor.hpp

#pragma once
 
#include <cstdint>
 
#include <userver/congestion_control/sensor.hpp>
#include <userver/engine/task/task_processor_fwd.hpp>
#include <userver/rcu/rcu.hpp>
#include <userver/utils/statistics/recentperiod.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace server::congestion_control {
 
class RequestsSource {
public:
    virtual ~RequestsSource() = default;
 
    virtual std::uint64_t GetTotalRequests() const = 0;
};
 
class Sensor final : public USERVER_NAMESPACE::congestion_control::Sensor {
public:
    explicit Sensor(engine::TaskProcessor& tp);
 
    Data FetchCurrent() override;
 
    void RegisterRequestsSource(RequestsSource& source);
 
private:
    engine::TaskProcessor& tp_;
    rcu::Variable<std::vector<RequestsSource*>> requests_sources_;
 
    std::chrono::steady_clock::time_point last_fetch_tp_;
    std::uint64_t last_overloads_{0};
    std::uint64_t last_no_overloads_{0};
    std::uint64_t last_requests_{0};
};
 
}  // namespace server::congestion_control
 
USERVER_NAMESPACE_END