Influxdata
/
influxdb
mirror of https://github.com/influxdata/influxdb.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

feat: disable prometheus protobuf (#2578) 

Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com>
pull/24376/head
Raphael Taylor-Davies 2021-09-20 10:16:04 +01:00 committed by GitHub
parent f62d0eab3c
commit e96aa49390
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 52 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Cargo.lock generated
View File

@ -3189,7 +3189,6 @@ dependencies = [
"lazy_static",
"memchr",
"parking_lot",
"protobuf",
"thiserror",
]
@ -3244,12 +3243,6 @@ dependencies = [
"prost",
]
[[package]]
name = "protobuf"
version = "2.25.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "23129d50f2c9355ced935fce8a08bd706ee2e7ce2b3b33bf61dace0e379ac63a"
[[package]]
name = "query"
version = "0.1.0"

2
metric_exporters/Cargo.toml
View File

@ -8,7 +8,7 @@ edition = "2018"
observability_deps = { path = "../observability_deps" }
metric = { path = "../metric" }
prometheus = "0.12"
prometheus = { version = "0.12", default-features = false }
[dev-dependencies] # In alphabetical order
test_helpers = { path = "../test_helpers" }

87
metric_exporters/src/lib.rs
View File

@ -69,7 +69,7 @@ impl<'a, W: Write> metric::Reporter for PrometheusTextEncoder<'a, W> {
}
};
let mut metric = MetricFamily::new();
let mut metric = MetricFamily::default();
metric.set_name(name);
metric.set_help(description.to_string());
metric.set_field_type(metric_type);
@ -82,75 +82,90 @@ impl<'a, W: Write> metric::Reporter for PrometheusTextEncoder<'a, W> {
let metrics = metrics.mut_metric();
let mut metric = Metric::new();
let mut metric = Metric::default();
for (name, value) in attributes.iter() {
let mut pair = LabelPair::new();
pair.set_name(name.to_string());
pair.set_value(value.to_string());
metric.mut_label().push(pair)
}
metric.set_label(
attributes
.iter()
.map(|(name, value)| {
let mut pair = LabelPair::default();
pair.set_name(name.to_string());
pair.set_value(value.to_string());
pair
})
.collect(),
);
match observation {
Observation::U64Counter(v) => {
let mut counter = Counter::new();
let mut counter = Counter::default();
counter.set_value(v as f64);
metric.set_counter(counter)
}
Observation::U64Gauge(v) => {
let mut gauge = Gauge::new();
let mut gauge = Gauge::default();
gauge.set_value(v as f64);
metric.set_gauge(gauge)
}
Observation::DurationCounter(v) => {
let mut counter = Counter::new();
let mut counter = Counter::default();
counter.set_value(v.as_secs_f64());
metric.set_counter(counter)
}
Observation::DurationGauge(v) => {
let mut gauge = Gauge::new();
let mut gauge = Gauge::default();
gauge.set_value(v.as_secs_f64());
metric.set_gauge(gauge)
}
Observation::U64Histogram(v) => {
let mut histogram = Histogram::new();
let mut histogram = Histogram::default();
let mut cumulative_count = 0;
for observation in v.buckets {
cumulative_count += observation.count;
histogram.set_bucket(
v.buckets
.into_iter()
.map(|observation| {
cumulative_count += observation.count;
let mut bucket = Bucket::new();
let le = match observation.le {
u64::MAX => f64::INFINITY,
v => v as f64,
};
let mut bucket = Bucket::default();
let le = match observation.le {
u64::MAX => f64::INFINITY,
v => v as f64,
};
bucket.set_upper_bound(le);
bucket.set_cumulative_count(cumulative_count);
histogram.mut_bucket().push(bucket)
}
bucket.set_upper_bound(le);
bucket.set_cumulative_count(cumulative_count);
bucket
})
.collect(),
);
histogram.set_sample_count(cumulative_count);
histogram.set_sample_sum(v.total as f64);
metric.set_histogram(histogram)
}
Observation::DurationHistogram(v) => {
let mut histogram = Histogram::new();
let mut histogram = Histogram::default();
let mut cumulative_count = 0;
for observation in v.buckets {
cumulative_count += observation.count;
histogram.set_bucket(
v.buckets
.into_iter()
.map(|observation| {
cumulative_count += observation.count;
let mut bucket = Bucket::new();
let le = match observation.le {
metric::DURATION_MAX => f64::INFINITY,
v => v.as_secs_f64(),
};
let mut bucket = Bucket::default();
let le = match observation.le {
metric::DURATION_MAX => f64::INFINITY,
v => v.as_secs_f64(),
};
bucket.set_upper_bound(le);
bucket.set_cumulative_count(cumulative_count);
histogram.mut_bucket().push(bucket)
}
bucket.set_upper_bound(le);
bucket.set_cumulative_count(cumulative_count);
bucket
})
.collect(),
);
histogram.set_sample_count(cumulative_count);
histogram.set_sample_sum(v.total.as_secs_f64());