prometheus-cpp/lib/registry.cc

#include "prometheus/registry.h"

namespace prometheus {

Family<Counter>& Registry::AddCounter(
    const std::string& name, const std::string& help,
    const std::map<std::string, std::string>& labels) {
  std::lock_guard<std::mutex> lock{mutex_};
  auto counter_family = new Family<Counter>(name, help, labels);
  collectables_.push_back(std::unique_ptr<Collectable>{counter_family});
  return *counter_family;
}

Family<Gauge>& Registry::AddGauge(
    const std::string& name, const std::string& help,
    const std::map<std::string, std::string>& labels) {
  std::lock_guard<std::mutex> lock{mutex_};
  auto gauge_family = new Family<Gauge>(name, help, labels);
  collectables_.push_back(std::unique_ptr<Collectable>{gauge_family});
  return *gauge_family;
}

Family<Histogram>& Registry::AddHistogram(
    const std::string& name, const std::string& help,
    const std::map<std::string, std::string>& labels) {
  std::lock_guard<std::mutex> lock{mutex_};
  auto histogram_family = new Family<Histogram>(name, help, labels);
  collectables_.push_back(std::unique_ptr<Collectable>{histogram_family});
  return *histogram_family;
}

Family<Summary>& Registry::AddSummary(
    const std::string& name, const std::string& help,
    const std::map<std::string, std::string>& labels) {
  std::lock_guard<std::mutex> lock{mutex_};
  auto histogram_family = new Family<Summary>(name, help, labels);
  collectables_.push_back(std::unique_ptr<Collectable>{histogram_family});
  return *histogram_family;
}

std::vector<io::prometheus::client::MetricFamily> Registry::Collect() {
  std::lock_guard<std::mutex> lock{mutex_};
  auto results = std::vector<io::prometheus::client::MetricFamily>{};
  for (auto&& collectable : collectables_) {
    auto metrics = collectable->Collect();
    results.insert(results.end(), metrics.begin(), metrics.end());
  }

  return results;
}
}