autoscaling_pool_policy.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 "pool_policy.h"
#include <kcenon/thread/scaling/autoscaler.h>
#include <kcenon/thread/scaling/autoscaling_policy.h>
 
#include <atomic>
#include <memory>
#include <string>
 
namespace kcenon::thread
{
    class autoscaling_pool_policy : public pool_policy
    {
    public:
        explicit autoscaling_pool_policy(thread_pool& pool, const autoscaling_policy& config = {});
 
        explicit autoscaling_pool_policy(std::shared_ptr<autoscaler> scaler);
 
        ~autoscaling_pool_policy() override;
 
        // Non-copyable
        autoscaling_pool_policy(const autoscaling_pool_policy&) = delete;
        autoscaling_pool_policy& operator=(const autoscaling_pool_policy&) = delete;
 
        // Non-movable (std::atomic is not movable)
        autoscaling_pool_policy(autoscaling_pool_policy&&) = delete;
        autoscaling_pool_policy& operator=(autoscaling_pool_policy&&) = delete;
 
        // ============================================
        // pool_policy Interface
        // ============================================
 
        auto on_enqueue(job& j) -> common::VoidResult override;
 
        void on_job_start(job& j) override;
 
        void on_job_complete(job& j, bool success, const std::exception* error = nullptr) override;
 
        [[nodiscard]] auto get_name() const -> std::string override;
 
        [[nodiscard]] auto is_enabled() const -> bool override;
 
        void set_enabled(bool enabled) override;
 
        // ============================================
        // Autoscaling Specific Methods
        // ============================================
 
        void start();
 
        void stop();
 
        [[nodiscard]] auto is_active() const -> bool;
 
        [[nodiscard]] auto get_autoscaler() const -> std::shared_ptr<autoscaler>;
 
        [[nodiscard]] auto get_stats() const -> autoscaling_stats;
 
        void set_policy(const autoscaling_policy& config);
 
        [[nodiscard]] auto get_policy() const -> const autoscaling_policy&;
 
        [[nodiscard]] auto evaluate_now() -> scaling_decision;
 
        auto scale_to(std::size_t target_workers) -> common::VoidResult;
 
    private:
        std::shared_ptr<autoscaler> autoscaler_;
        std::atomic<bool> enabled_{true};
    };
 
} // namespace kcenon::thread