forked from open-telemetry/opentelemetry-rust
-
Notifications
You must be signed in to change notification settings - Fork 1
/
metrics.rs
76 lines (68 loc) · 2.24 KB
/
metrics.rs
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
use crate::metrics::{self, Meter, MeterProvider};
use std::sync::{Arc, RwLock};
lazy_static::lazy_static! {
/// The global `Meter` provider singleton.
static ref GLOBAL_METER_PROVIDER: RwLock<GlobalMeterProvider> = RwLock::new(GlobalMeterProvider::new(metrics::noop::NoopMeterProvider::new()));
}
/// Represents the globally configured [`MeterProvider`] instance for this
/// application.
#[derive(Debug, Clone)]
pub struct GlobalMeterProvider {
provider: Arc<dyn MeterProvider + Send + Sync>,
}
impl MeterProvider for GlobalMeterProvider {
fn meter(
&self,
name: &'static str,
version: Option<&'static str>,
schema_url: Option<&'static str>,
) -> Meter {
self.provider.meter(name, version, schema_url)
}
}
impl GlobalMeterProvider {
/// Create a new global meter provider
pub fn new<P>(provider: P) -> Self
where
P: MeterProvider + Send + Sync + 'static,
{
GlobalMeterProvider {
provider: Arc::new(provider),
}
}
}
/// Sets the given [`MeterProvider`] instance as the current global meter
/// provider.
pub fn set_meter_provider<P>(new_provider: P)
where
P: metrics::MeterProvider + Send + Sync + 'static,
{
let mut global_provider = GLOBAL_METER_PROVIDER
.write()
.expect("GLOBAL_METER_PROVIDER RwLock poisoned");
*global_provider = GlobalMeterProvider::new(new_provider);
}
/// Returns an instance of the currently configured global [`MeterProvider`]
/// through [`GlobalMeterProvider`].
pub fn meter_provider() -> GlobalMeterProvider {
GLOBAL_METER_PROVIDER
.read()
.expect("GLOBAL_METER_PROVIDER RwLock poisoned")
.clone()
}
/// Creates a named [`Meter`] via the configured [`GlobalMeterProvider`].
///
/// If the name is an empty string, the provider will use a default name.
///
/// This is a more convenient way of expressing `global::meter_provider().meter(name)`.
pub fn meter(name: &'static str) -> Meter {
meter_provider().meter(name, None, None)
}
/// Creates a [`Meter`] with the name and version.
pub fn meter_with_version(
name: &'static str,
version: &'static str,
schema_url: &'static str,
) -> Meter {
meter_provider().meter(name, Some(version), Some(schema_url))
}