ProcessMetrics.hpp

Go to the documentation of this file.

#pragma once
 
#include <prometheus/gauge.h>
#include <prometheus/registry.h>
 
#include <thread>
 
#include <sys/times.h>
 
class ProcessMetrics {
  private:
    std::atomic_flag _shouldStop{false};          
    std::unique_ptr<std::thread> _thread;         
    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 {
        0, 0, 0, 0
    }; 
 
    static size_t countDirectoryEntries(const std::string &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() 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();
};