metrics_group.hpp

Go to the documentation of this file.

#pragma once
 
#include <prometheus/gauge.h>
#include <prometheus/histogram.h>
#include <prometheus/registry.h>
 
namespace taraxa::metrics {
 
#define ADD_GAUGE_METRIC(method, name, description)                                                  \
  void method(double v) {                                                                            \
    static auto& label = addMetric<prometheus::Gauge>(group_name + "_" + name, description).Add({}); \
    label.Set(v);                                                                                    \
  }
 
#define ADD_HISTOGRAM_METRIC(method, name, description, buckets)                                           \
  void method(double v, std::map<std::string, std::string> labels) {                                       \
    static auto& label = addMetric<prometheus::Histogram>(group_name + "_" + name, description);           \
    label.Add(labels, prometheus::Histogram::BucketBoundaries{buckets.begin(), buckets.end()}).Observe(v); \
  }
 
#define ADD_UPDATER_METHOD(method)                               \
  void method##Updater(MetricGetter getter) {                    \
    updaters_.push_back([this, getter]() { method(getter()); }); \
  }
 
#define ADD_GAUGE_METRIC_WITH_UPDATER(method, name, description) \
  ADD_GAUGE_METRIC(method, name, description)                    \
  ADD_UPDATER_METHOD(method)
 
class MetricsGroup {
 public:
  using MetricGetter = std::function<double()>;
  using MetricUpdater = std::function<void()>;
  MetricsGroup(std::shared_ptr<prometheus::Registry> registry) : registry_(std::move(registry)) {}
  virtual ~MetricsGroup() = default;
 
  template <class Type>
  prometheus::Family<Type>& addMetric(const std::string& name, const std::string& help) {
    return prometheus::detail::Builder<Type>().Name(name).Help(help).Register(*registry_);
  }
  void updateData() {
    for (auto& update : updaters_) {
      update();
    }
  }
 
 protected:
  std::shared_ptr<prometheus::Registry> registry_;
  std::vector<MetricUpdater> updaters_;
};
 
using SharedMetricsGroup = std::shared_ptr<MetricsGroup>;
}  // namespace taraxa::metrics