io_context_thread_manager.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 <asio.hpp>
#include <functional>
#include <future>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
 
#include "kcenon/network/integration/thread_integration.h"
 
namespace kcenon::network::integration {
 
class io_context_thread_manager {
public:
    static io_context_thread_manager& instance();
 
    std::future<void> run_io_context(
        std::shared_ptr<asio::io_context> io_context,
        const std::string& component_name = ""
    );
 
    void stop_io_context(std::shared_ptr<asio::io_context> io_context);
 
    void stop_all();
 
    void wait_all();
 
    size_t active_count() const;
 
    bool is_active(std::shared_ptr<asio::io_context> io_context) const;
 
    void set_thread_pool(std::shared_ptr<thread_pool_interface> pool);
 
    struct metrics {
        size_t active_contexts = 0;     
        size_t total_started = 0;       
        size_t total_completed = 0;     
    };
 
    metrics get_metrics() const;
 
private:
    io_context_thread_manager();
    ~io_context_thread_manager();
 
    // Non-copyable, non-movable
    io_context_thread_manager(const io_context_thread_manager&) = delete;
    io_context_thread_manager& operator=(const io_context_thread_manager&) = delete;
    io_context_thread_manager(io_context_thread_manager&&) = delete;
    io_context_thread_manager& operator=(io_context_thread_manager&&) = delete;
 
    class impl;
 
    std::shared_ptr<impl> pimpl_;
};
 
} // namespace kcenon::network::integration
 
// Backward compatibility - use the existing namespace alias from thread_integration.h
// The alias `namespace kcenon::network::integration = kcenon::network::integration;` is
// already defined in thread_integration.h, which is included before this header.