use crate::metrics::{
sync_instrument::{SyncBoundInstrument, SyncInstrument},
Descriptor, InstrumentKind, Measurement, Meter, Number, NumberKind, Result, Unit,
};
use crate::KeyValue;
use std::marker;
#[derive(Clone, Debug)]
pub struct ValueRecorder<T>(SyncInstrument<T>);
impl<T> ValueRecorder<T>
where
T: Into<Number>,
{
pub fn bind(&self, attributes: &[KeyValue]) -> BoundValueRecorder<T> {
let bound_instrument = self.0.bind(attributes);
BoundValueRecorder { bound_instrument }
}
pub fn record(&self, value: T, attributes: &[KeyValue]) {
self.0.direct_record(value.into(), attributes)
}
pub fn measurement(&self, value: T) -> Measurement {
Measurement::new(value.into(), self.0.instrument().clone())
}
}
#[derive(Clone, Debug)]
pub struct BoundValueRecorder<T> {
bound_instrument: SyncBoundInstrument<T>,
}
impl<T> BoundValueRecorder<T>
where
T: Into<Number>,
{
pub fn record(&self, value: T) {
self.bound_instrument.direct_record(value.into())
}
}
#[derive(Debug)]
pub struct ValueRecorderBuilder<'a, T> {
meter: &'a Meter,
descriptor: Descriptor,
_marker: marker::PhantomData<T>,
}
impl<'a, T> ValueRecorderBuilder<'a, T> {
pub(crate) fn new(meter: &'a Meter, name: String, number_kind: NumberKind) -> Self {
ValueRecorderBuilder {
meter,
descriptor: Descriptor::new(
name,
meter.instrumentation_library().name,
meter.instrumentation_library().version,
InstrumentKind::ValueRecorder,
number_kind,
),
_marker: marker::PhantomData,
}
}
pub fn with_description<S: Into<String>>(mut self, description: S) -> Self {
self.descriptor.set_description(description.into());
self
}
pub fn with_unit(mut self, unit: Unit) -> Self {
self.descriptor.config.unit = Some(unit);
self
}
pub fn try_init(self) -> Result<ValueRecorder<T>> {
let instrument = self.meter.new_sync_instrument(self.descriptor)?;
Ok(ValueRecorder(SyncInstrument::new(instrument)))
}
pub fn init(self) -> ValueRecorder<T> {
ValueRecorder(SyncInstrument::new(
self.meter.new_sync_instrument(self.descriptor).unwrap(),
))
}
}