metrics_service.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2024, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include <kcenon/thread/metrics/thread_pool_metrics.h>
#include <kcenon/thread/metrics/enhanced_metrics.h>
 
#include <atomic>
#include <chrono>
#include <memory>
#include <mutex>
 
namespace kcenon::thread::metrics {
 
class metrics_service {
public:
    metrics_service();
 
    ~metrics_service() = default;
 
    // Non-copyable, non-movable for thread safety
    metrics_service(const metrics_service&) = delete;
    metrics_service& operator=(const metrics_service&) = delete;
    metrics_service(metrics_service&&) = delete;
    metrics_service& operator=(metrics_service&&) = delete;
 
    // =========================================================================
    // Recording Methods (called by thread_pool and thread_worker)
    // =========================================================================
 
    void record_submission(std::size_t count = 1);
 
    void record_enqueue(std::size_t count = 1);
 
    void record_enqueue_with_latency(std::chrono::nanoseconds latency, std::size_t count = 1);
 
    void record_execution(std::uint64_t duration_ns, bool success);
 
    void record_execution_with_wait_time(
        std::chrono::nanoseconds duration,
        std::chrono::nanoseconds wait_time,
        bool success);
 
    void record_idle_time(std::uint64_t duration_ns);
 
    void record_queue_depth(std::size_t depth);
 
    void record_worker_state(
        std::size_t worker_id,
        bool busy,
        std::uint64_t duration_ns = 0);
 
    // =========================================================================
    // Enhanced Metrics Control
    // =========================================================================
 
    void set_enhanced_metrics_enabled(bool enabled, std::size_t worker_count = 0);
 
    [[nodiscard]] bool is_enhanced_metrics_enabled() const;
 
    void update_worker_count(std::size_t count);
 
    void set_active_workers(std::size_t count);
 
    // =========================================================================
    // Query Methods
    // =========================================================================
 
    [[nodiscard]] const ThreadPoolMetrics& basic_metrics() const noexcept;
 
    [[nodiscard]] const EnhancedThreadPoolMetrics& enhanced_metrics() const;
 
    [[nodiscard]] EnhancedSnapshot enhanced_snapshot() const;
 
    [[nodiscard]] std::shared_ptr<ThreadPoolMetrics> get_basic_metrics() const noexcept;
 
    // =========================================================================
    // Management Methods
    // =========================================================================
 
    void reset();
 
private:
    std::shared_ptr<ThreadPoolMetrics> basic_metrics_;
 
    std::shared_ptr<EnhancedThreadPoolMetrics> enhanced_metrics_;
 
    std::atomic<bool> enhanced_enabled_{false};
 
    mutable std::mutex init_mutex_;
};
 
} // namespace kcenon::thread::metrics