mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-03-12 04:36:29 -07:00
140 lines
4.3 KiB
C++
140 lines
4.3 KiB
C++
#pragma once
|
|
|
|
#include "prometheus/atomic_floating.h"
|
|
#include "prometheus/metric.h"
|
|
#include "prometheus/family.h"
|
|
|
|
#include "prometheus/builder.h"
|
|
|
|
#include <atomic>
|
|
#include <ctime>
|
|
|
|
namespace prometheus {
|
|
|
|
/// \brief A gauge metric to represent a value that can arbitrarily go up and
|
|
/// down.
|
|
///
|
|
/// The class represents the metric type gauge:
|
|
/// https://prometheus.io/docs/concepts/metric_types/#gauge
|
|
///
|
|
/// Gauges are typically used for measured values like temperatures or current
|
|
/// memory usage, but also "counts" that can go up and down, like the number of
|
|
/// running processes.
|
|
///
|
|
/// The class is thread-safe. No concurrent call to any API of this type causes
|
|
/// a data race.
|
|
#include <stdint.h>
|
|
|
|
#if UINTPTR_MAX == 0xffFFffFF
|
|
// 32-bit
|
|
template <typename Value_ = uint32_t>
|
|
#elif UINTPTR_MAX == 0xffFFffFFffFFffFF
|
|
// 64-bit
|
|
template <typename Value_ = uint64_t>
|
|
#else
|
|
#error Unknown platform - does not look either like 32-bit or 64-bit
|
|
#endif
|
|
class Gauge : public Metric {
|
|
|
|
std::atomic<Value_> value { 0 };
|
|
|
|
public:
|
|
|
|
using Value = Value_;
|
|
using Family = CustomFamily<Gauge<Value>>;
|
|
|
|
static const Metric::Type static_type = Metric::Type::Gauge;
|
|
|
|
|
|
Gauge() : Metric (static_type) {} ///< \brief Create a gauge that starts at 0.
|
|
Gauge(const Value value_) : Metric(static_type), value{ value_ } {} ///< \brief Create a gauge that starts at the given amount.
|
|
Gauge(const Gauge<Value_> &rhs) : Metric(rhs.static_type), value{rhs.value.load()} {}
|
|
|
|
// original API
|
|
|
|
void Increment() { ++value; } ///< \brief Increment the gauge by 1.
|
|
void Increment(const Value& val) { value += val; } ///< \brief Increment the gauge by the given amount.
|
|
|
|
void Decrement() { --value; } ///< \brief Decrement the gauge by 1.
|
|
void Decrement(const Value& val) { value -= val; } ///< \brief Decrement the gauge by the given amount.
|
|
|
|
void SetToCurrentTime() { ///< \brief Set the gauge to the current unixtime in seconds.
|
|
const time_t time = std::time(nullptr);
|
|
value = static_cast<Value>(time);
|
|
}
|
|
void Set(const Value& val) { value = val; } ///< \brief Set the gauge to the given value.
|
|
const Value Get() const { return value; } ///< \brief Get the current value of the gauge.
|
|
|
|
virtual ClientMetric Collect() const { ///< \brief Get the current value of the gauge. Collect is called by the Registry when collecting metrics.
|
|
ClientMetric metric;
|
|
metric.gauge.value = static_cast<double>(value);
|
|
return metric;
|
|
}
|
|
|
|
// new API
|
|
|
|
Gauge& operator ++() {
|
|
++value;
|
|
return *this;
|
|
}
|
|
|
|
Gauge& operator++ (int) {
|
|
++value;
|
|
return *this;
|
|
}
|
|
|
|
Gauge& operator --() {
|
|
--value;
|
|
return *this;
|
|
}
|
|
|
|
Gauge& operator-- (int) {
|
|
--value;
|
|
return *this;
|
|
}
|
|
|
|
Gauge& operator+=(const Value& val) {
|
|
value += val;
|
|
return *this;
|
|
}
|
|
|
|
Gauge& operator-=(const Value& val) {
|
|
value -= val;
|
|
return *this;
|
|
}
|
|
|
|
};
|
|
|
|
|
|
/// \brief Return a builder to configure and register a Gauge metric.
|
|
///
|
|
/// @copydetails Family<>::Family()
|
|
///
|
|
/// Example usage:
|
|
///
|
|
/// \code
|
|
/// auto registry = std::make_shared<Registry>();
|
|
/// auto& gauge_family = prometheus::BuildGauge()
|
|
/// .Name("some_name")
|
|
/// .Help("Additional description.")
|
|
/// .Labels({{"key", "value"}})
|
|
/// .Register(*registry);
|
|
///
|
|
/// ...
|
|
/// \endcode
|
|
///
|
|
/// \return An object of unspecified type T, i.e., an implementation detail
|
|
/// except that it has the following members:
|
|
///
|
|
/// - Name(const std::string&) to set the metric name,
|
|
/// - Help(const std::string&) to set an additional description.
|
|
/// - Label(const std::map<std::string, std::string>&) to assign a set of
|
|
/// key-value pairs (= labels) to the metric.
|
|
///
|
|
/// To finish the configuration of the Gauge metric register it with
|
|
/// Register(Registry&).
|
|
using BuildGauge = Builder<Gauge<double>>;
|
|
|
|
|
|
} // namespace prometheus
|