ProcessMetrics.hpp

Go to the documentation of this file.

#pragma once
 
#include <prometheus/gauge.h>
#include <prometheus/registry.h>
 
#include <filesystem>
#include <thread>
 
#include <sys/times.h>

class ProcessMetrics {
  private:
    std::shared_ptr<std::atomic_flag> _checkFlag; 
 
    prometheus::Gauge *_pInitTime;              
    prometheus::Gauge *_pCurrentTime;        
    prometheus::Gauge *_pMemory;              
    prometheus::Gauge *_pPageFaults;          
    prometheus::Gauge *_pCpuUsage;              
    prometheus::Gauge *_pDiskRead;              
    prometheus::Gauge *_pDiskWrite;            
    prometheus::Gauge *_pThreadCount;        
    prometheus::Gauge *_pFileDescriptorCount; 
 
    size_t _oldReadBytes{0};  
    size_t _oldWriteBytes{0}; 
 
    clock_t _oldCpuTime{0}; 
    struct tms _oldCpu{
        .tms_utime = 0, .tms_stime = 0, .tms_cutime = 0, .tms_cstime = 0}; 
 
    std::unique_ptr<std::jthread> _thread; 

    static size_t countDirectoryEntries(const std::filesystem::path &path);
 
  protected:
    static long int getMemoryUsage();

    static long int getPageFaults();

    double getCpuUsage();

    std::pair<size_t, size_t> getDiskIO();

    static size_t getThreadCount();

    static size_t getFileDescriptorCount();

    void update();

    void threadRunner(const std::stop_token &stopToken) noexcept;
 
  public:
    ProcessMetrics(std::shared_ptr<std::atomic_flag> checkFlag, const std::shared_ptr<prometheus::Registry> &reg);

    ProcessMetrics(const ProcessMetrics & /*unused*/) = delete;

    ProcessMetrics(ProcessMetrics && /*unused*/) = delete;

    ProcessMetrics &operator=(ProcessMetrics /*unused*/) = delete;

    ProcessMetrics &operator=(ProcessMetrics && /*unused*/) = delete;

    ~ProcessMetrics();
};