Statistics.h

/*
This program has been developed by students from the bachelor Computer Science at
Utrecht University within the Software Project course.
© Copyright Utrecht University (Department of Information and Computing Sciences)
*/
 
#pragma once
 
#include <prometheus/counter.h>
#include <prometheus/exposer.h>
#include <prometheus/registry.h>
#include <queue>
 
#define SYNCHRONIZE_DELAY 10000000 // 10 seconds.
 
enum StatFamily
{
    reqCount,
    methCount,
    langCount,
    jobCount,
    reqTime,
    vulnCount,
    recProj,
    recVuln
};
 
class Statistics
{
  public:
    virtual void Initialize();
 
    void addRecentProject(std::string url);
 
    void addRecentVulnerability(std::string vulnCode);
 
    void synchronize(std::string file);
 
    void writeToFile(std::string file);
 
    void readFromFile(std::string file);
 
    // The families to store the statistics in.
    prometheus::Family<prometheus::Counter> *requestCounter;
    prometheus::Family<prometheus::Counter> *methodCounter;
    prometheus::Family<prometheus::Counter> *languageCounter;
    prometheus::Family<prometheus::Counter> *jobCounter;
    prometheus::Family<prometheus::Gauge> *latestRequest;
    prometheus::Family<prometheus::Counter> *vulnCounter;
    prometheus::Family<prometheus::Gauge> *recentProjects;
    prometheus::Family<prometheus::Gauge> *recentVulns;
 
    // The ip of this node for identifying the node in the statistics.
    std::string myIP;
 
    // A boolean to store whether there has been a new request.
    bool newRequest;
 
  protected:
    // The exposer for the statistics.
    prometheus::Exposer *exposer;
 
    // The registry that stores the families.
    std::shared_ptr<prometheus::Registry> registry;
 
    std::queue<prometheus::Gauge *> recentProjectsQueue;
    std::queue<prometheus::Gauge *> recentVulnsQueue;
};