handler.h

#pragma once
 
#include <memory>
#include <mutex>
#include <vector>
 
#include "CivetServer.h"
#include "prometheus/collectable.h"
#include "prometheus/counter.h"
#include "prometheus/family.h"
#include "prometheus/registry.h"
#include "prometheus/summary.h"
 
namespace prometheus::detail {
class MetricsHandler : public CivetHandler {
 public:
  explicit MetricsHandler(Registry& registry);
 
  void RegisterCollectable(const std::weak_ptr<Collectable>& collectable);
  void RemoveCollectable(const std::weak_ptr<Collectable>& collectable);
 
  bool handleGet(CivetServer* server, struct mg_connection* conn) override;
 
 private:
  static void CleanupStalePointers(
      std::vector<std::weak_ptr<Collectable>>& collectables);
 
  std::mutex collectables_mutex_;
  std::vector<std::weak_ptr<Collectable>> collectables_;
  Family<Counter>& bytes_transferred_family_;
  Counter& bytes_transferred_;
  Family<Counter>& num_scrapes_family_;
  Counter& num_scrapes_;
  Family<Summary>& request_latencies_family_;
  Summary& request_latencies_;
};
}  // namespace prometheus::detail