thread_integration.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 <functional>
#include <memory>
#include <future>
#include <chrono>
#include <vector>
 
namespace kcenon::network::integration {
 
class thread_pool_interface {
public:
    virtual ~thread_pool_interface() = default;
 
    virtual std::future<void> submit(std::function<void()> task) = 0;
 
    virtual std::future<void> submit_delayed(
        std::function<void()> task,
        std::chrono::milliseconds delay
    ) = 0;
 
    virtual size_t worker_count() const = 0;
 
    virtual bool is_running() const = 0;
 
    virtual size_t pending_tasks() const = 0;
};
 
class basic_thread_pool : public thread_pool_interface {
public:
    explicit basic_thread_pool(size_t num_threads = 0);
 
    ~basic_thread_pool();
 
    // thread_pool_interface implementation
    std::future<void> submit(std::function<void()> task) override;
    std::future<void> submit_delayed(
        std::function<void()> task,
        std::chrono::milliseconds delay
    ) override;
    size_t worker_count() const override;
    bool is_running() const override;
    size_t pending_tasks() const override;
 
    void stop(bool wait_for_tasks = true);
 
    size_t completed_tasks() const;
 
private:
    class impl;
 
    std::shared_ptr<impl> pimpl_;
};
 
class thread_integration_manager {
public:
    static thread_integration_manager& instance();
 
    void set_thread_pool(std::shared_ptr<thread_pool_interface> pool);
 
    std::shared_ptr<thread_pool_interface> get_thread_pool();
 
    std::future<void> submit_task(std::function<void()> task);
 
    std::future<void> submit_delayed_task(
        std::function<void()> task,
        std::chrono::milliseconds delay
    );
 
    struct metrics {
        size_t worker_threads = 0;
        size_t pending_tasks = 0;
        size_t completed_tasks = 0;
        bool is_running = false;
    };
 
    metrics get_metrics() const;
 
private:
    thread_integration_manager();
    ~thread_integration_manager();
 
    class impl;
 
    std::shared_ptr<impl> pimpl_;
};
 
} // namespace kcenon::network::integration
 
// Backward compatibility
namespace network_system {
    namespace integration = kcenon::network::integration;
}