central_collector.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2021-2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include <kcenon/monitoring/core/thread_local_buffer.h>
#include <kcenon/monitoring/core/performance_types.h>
#include <kcenon/monitoring/core/result_types.h>
#include <kcenon/monitoring/core/error_codes.h>
#include <shared_mutex>
#include <unordered_map>
#include <memory>
#include <atomic>
 
namespace kcenon { namespace monitoring {
 
class central_collector {
public:
    static constexpr size_t DEFAULT_MAX_PROFILES = 10000;
 
    explicit central_collector(size_t max_profiles = DEFAULT_MAX_PROFILES);
 
    void receive_batch(const std::vector<metric_sample>& samples);
 
    common::Result<performance_profile> get_profile(const std::string& operation_name) const;
 
    std::unordered_map<std::string, performance_profile> get_all_profiles() const;
 
    void clear();
 
    size_t get_operation_count() const;
 
    size_t get_total_sample_count() const {
        return total_samples_.load(std::memory_order_relaxed);
    }
 
    struct stats {
        size_t operation_count;     
        size_t total_samples;       
        size_t batches_received;    
        size_t lru_evictions;       
    };
 
    stats get_stats() const;
 
private:
    struct profile_data {
        performance_profile profile;
        std::atomic<std::chrono::steady_clock::rep> last_access_time;
        std::mutex mutex;  // Per-profile lock for sample aggregation
 
        profile_data() : last_access_time(0) {}
    };
 
    void process_sample(const metric_sample& sample);
 
    void evict_lru();
 
    mutable std::shared_mutex profiles_mutex_;
    std::unordered_map<std::string, std::unique_ptr<profile_data>> profiles_;
    size_t max_profiles_;
 
    // Statistics (atomic for thread-safe updates)
    std::atomic<size_t> total_samples_{0};
    std::atomic<size_t> batches_received_{0};
    std::atomic<size_t> lru_evictions_{0};
};
 
}} // namespace kcenon::monitoring