container_plugin.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 <atomic>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
#include <vector>
 
#include "../../collectors/plugin_metric_collector.h"
 
namespace kcenon {
namespace monitoring {
 
// Forward declarations
class container_collector;
 
namespace plugins {
 
enum class container_runtime {
    auto_detect,  
    docker,       
    containerd,   
    podman,       
    cri_o         
};
 
struct container_plugin_config {
    container_runtime runtime = container_runtime::auto_detect;
 
    bool enable_docker = true;
 
    bool enable_kubernetes = false;
 
    bool enable_cgroup = true;
 
    std::string docker_socket = "/var/run/docker.sock";
 
    std::string kubeconfig_path;
 
    std::string namespace_filter;
 
    bool collect_network_metrics = true;
 
    bool collect_blkio_metrics = true;
 
    bool collect_pid_metrics = true;
};
 
class container_plugin : public metric_collector_plugin {
   public:
    static std::unique_ptr<container_plugin> create(const container_plugin_config& config = {});
 
    ~container_plugin() override;
 
    // Disable copy
    container_plugin(const container_plugin&) = delete;
    container_plugin& operator=(const container_plugin&) = delete;
 
    // metric_collector_plugin interface
    bool initialize(const std::unordered_map<std::string, std::string>& config) override;
    std::vector<metric> collect() override;
    std::string get_name() const override;
    std::vector<std::string> get_metric_types() const override;
    bool is_healthy() const override;
    std::unordered_map<std::string, double> get_statistics() const override;
 
    static bool is_running_in_container();
 
    static bool is_kubernetes_environment();
 
    static container_runtime detect_runtime();
 
    bool is_docker_available() const;
 
    bool is_kubernetes_available() const;
 
    bool is_cgroup_available() const;
 
    container_plugin_config get_config() const;
 
   private:
    explicit container_plugin(const container_plugin_config& config);
 
    // Internal collectors
    std::unique_ptr<container_collector> container_collector_;
 
    // Configuration
    container_plugin_config config_;
    bool initialized_{false};
 
    // Statistics
    mutable std::mutex stats_mutex_;
    std::atomic<size_t> total_collections_{0};
    std::atomic<size_t> collection_errors_{0};
    std::atomic<size_t> containers_found_{0};
 
    // Helper methods
    void initialize_collectors();
};
 
}  // namespace plugins
}  // namespace monitoring
}  // namespace kcenon