Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-14 20:48:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Add support for callbacks for gauges/counters, restore process stat metrics

Browse source
This commit is contained in:
Tim Wojtulewicz 2024-03-14 15:08:08 -07:00
parent e93e4cc26d
commit 1cad305e58
8 changed files with 103 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/telemetry/Manager.h
View file

@ -122,10 +122,11 @@ public:
template<class ValueType = int64_t>
std::shared_ptr<Counter<ValueType>> CounterInstance(std::string_view prefix, std::string_view name,
Span<const LabelView> labels, std::string_view helptext,
std::string_view unit = "", bool is_sum = false) {
std::string_view unit = "", bool is_sum = false,
prometheus::CollectCallbackPtr callback = nullptr) {
return WithLabelNames(labels, [&, this](auto labelNames) {
auto family = CounterFamily<ValueType>(prefix, name, labelNames, helptext, unit, is_sum);
return family->GetOrAdd(labels);
return family->GetOrAdd(labels, callback);
});
}
@ -134,9 +135,10 @@ public:
std::shared_ptr<Counter<ValueType>> CounterInstance(std::string_view prefix, std::string_view name,
std::initializer_list<LabelView> labels,
std::string_view helptext, std::string_view unit = "",
bool is_sum = false) {
bool is_sum = false,
prometheus::CollectCallbackPtr callback = nullptr) {
auto lbl_span = Span{labels.begin(), labels.size()};
return CounterInstance<ValueType>(prefix, name, lbl_span, helptext, unit, is_sum);
return CounterInstance<ValueType>(prefix, name, lbl_span, helptext, unit, is_sum, callback);
}
/**
@ -199,10 +201,11 @@ public:
template<class ValueType = int64_t>
std::shared_ptr<Gauge<ValueType>> GaugeInstance(std::string_view prefix, std::string_view name,
Span<const LabelView> labels, std::string_view helptext,
std::string_view unit = "", bool is_sum = false) {
std::string_view unit = "", bool is_sum = false,
prometheus::CollectCallbackPtr callback = nullptr) {
return WithLabelNames(labels, [&, this](auto labelNames) {
auto family = GaugeFamily<ValueType>(prefix, name, labelNames, helptext, unit, is_sum);
return family->GetOrAdd(labels);
return family->GetOrAdd(labels, callback);
});
}
@ -210,9 +213,10 @@ public:
template<class ValueType = int64_t>
std::shared_ptr<Gauge<ValueType>> GaugeInstance(std::string_view prefix, std::string_view name,
std::initializer_list<LabelView> labels, std::string_view helptext,
std::string_view unit = "", bool is_sum = false) {
std::string_view unit = "", bool is_sum = false,
prometheus::CollectCallbackPtr callback = nullptr) {
auto lbl_span = Span{labels.begin(), labels.size()};
return GaugeInstance<ValueType>(prefix, name, lbl_span, helptext, unit, is_sum);
return GaugeInstance<ValueType>(prefix, name, lbl_span, helptext, unit, is_sum, callback);
}
// Forces the compiler to use the type `Span<const T>` instead of trying to