1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
|
// Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
//! This examples shows how to use multiple and custom registries,
//! and how to perform registration across function boundaries.
use std::collections::HashMap;
use prometheus::{Encoder, IntCounter, Registry};
use lazy_static::lazy_static;
lazy_static! {
static ref DEFAULT_COUNTER: IntCounter = IntCounter::new("default", "generic counter").unwrap();
static ref CUSTOM_COUNTER: IntCounter = IntCounter::new("custom", "dedicated counter").unwrap();
}
fn main() {
// Register default metrics.
default_metrics(prometheus::default_registry());
// Register custom metrics to a custom registry.
let mut labels = HashMap::new();
labels.insert("mykey".to_string(), "myvalue".to_string());
let custom_registry = Registry::new_custom(Some("myprefix".to_string()), Some(labels)).unwrap();
custom_metrics(&custom_registry);
// Print metrics for the default registry.
let mut buffer = Vec::<u8>::new();
let encoder = prometheus::TextEncoder::new();
encoder.encode(&prometheus::gather(), &mut buffer).unwrap();
println!("## Default registry");
println!("{}", String::from_utf8(buffer.clone()).unwrap());
// Print metrics for the custom registry.
let mut buffer = Vec::<u8>::new();
let encoder = prometheus::TextEncoder::new();
encoder
.encode(&custom_registry.gather(), &mut buffer)
.unwrap();
println!("## Custom registry");
println!("{}", String::from_utf8(buffer.clone()).unwrap());
}
/// Default metrics, to be collected by the default registry.
fn default_metrics(registry: &Registry) {
registry
.register(Box::new(DEFAULT_COUNTER.clone()))
.unwrap();
DEFAULT_COUNTER.inc();
assert_eq!(DEFAULT_COUNTER.get(), 1);
}
/// Custom metrics, to be collected by a dedicated registry.
fn custom_metrics(registry: &Registry) {
registry.register(Box::new(CUSTOM_COUNTER.clone())).unwrap();
CUSTOM_COUNTER.inc_by(42);
assert_eq!(CUSTOM_COUNTER.get(), 42);
}
|