Merge branch 'main' into preallocate-attribute-mem

Author: lalitb

opentelemetry-sdk/src/metrics/internal/histogram.rs

@@ -43,28 +43,36 @@ impl<T: Number<T>> Buckets<T> {
     }
 }
 
-/// Summarizes a set of measurements with explicitly defined buckets.
-struct HistValues<T> {
-    record_sum: bool,
-    bounds: Vec<f64>,
+/// Summarizes a set of measurements as a histogram with explicitly defined
+/// buckets.
+pub(crate) struct Histogram<T> {
     values: Mutex<HashMap<AttributeSet, Buckets<T>>>,
+    bounds: Vec<f64>,
+    record_min_max: bool,
+    record_sum: bool,
+    start: Mutex<SystemTime>,
 }
 
-impl<T: Number<T>> HistValues<T> {
-    fn new(mut bounds: Vec<f64>, record_sum: bool) -> Self {
-        bounds.retain(|v| !v.is_nan());
-        bounds.sort_by(|a, b| a.partial_cmp(b).expect("NaNs filtered out"));
-
-        HistValues {
-            record_sum,
-            bounds,
+impl<T: Number<T>> Histogram<T> {
+    pub(crate) fn new(boundaries: Vec<f64>, record_min_max: bool, record_sum: bool) -> Self {
+        let mut histogram = Histogram {
             values: Mutex::new(Default::default()),
-        }
+            bounds: boundaries,
+            record_min_max,
+            record_sum,
+            start: Mutex::new(SystemTime::now()),
+        };
+
+        histogram.bounds.retain(|v| !v.is_nan());
+        histogram
+            .bounds
+            .sort_by(|a, b| a.partial_cmp(b).expect("NaNs filtered out"));
+
+        histogram
     }
-}
 
-impl<T: Number<T>> HistValues<T> {
-    fn measure(&self, measurement: T, attrs: AttributeSet) {
+    pub(crate) fn measure(&self, measurement: T, attrs: &[KeyValue]) {
+        let attrs: AttributeSet = attrs.into();
         let f = measurement.into_float();
 
         // This search will return an index in the range `[0, bounds.len()]`, where
@@ -109,35 +117,12 @@ impl<T: Number<T>> HistValues<T> {
             b.sum(measurement)
         }
     }
-}
-
-/// Summarizes a set of measurements as a histogram with explicitly defined
-/// buckets.
-pub(crate) struct Histogram<T> {
-    hist_values: HistValues<T>,
-    record_min_max: bool,
-    start: Mutex<SystemTime>,
-}
-
-impl<T: Number<T>> Histogram<T> {
-    pub(crate) fn new(boundaries: Vec<f64>, record_min_max: bool, record_sum: bool) -> Self {
-        Histogram {
-            hist_values: HistValues::new(boundaries, record_sum),
-            record_min_max,
-            start: Mutex::new(SystemTime::now()),
-        }
-    }
-
-    pub(crate) fn measure(&self, measurement: T, attrs: &[KeyValue]) {
-        let attrs: AttributeSet = attrs.into();
-        self.hist_values.measure(measurement, attrs)
-    }
 
     pub(crate) fn delta(
         &self,
         dest: Option<&mut dyn Aggregation>,
     ) -> (usize, Option<Box<dyn Aggregation>>) {
-        let mut values = match self.hist_values.values.lock() {
+        let mut values = match self.values.lock() {
             Ok(guard) if !guard.is_empty() => guard,
             _ => return (0, None),
         };
@@ -174,9 +159,9 @@ impl<T: Number<T>> Histogram<T> {
                 start_time: start,
                 time: t,
                 count: b.count,
-                bounds: self.hist_values.bounds.clone(),
+                bounds: self.bounds.clone(),
                 bucket_counts: b.counts.clone(),
-                sum: if self.hist_values.record_sum {
+                sum: if self.record_sum {
                     b.total
                 } else {
                     T::default()
@@ -207,7 +192,7 @@ impl<T: Number<T>> Histogram<T> {
         &self,
         dest: Option<&mut dyn Aggregation>,
     ) -> (usize, Option<Box<dyn Aggregation>>) {
-        let values = match self.hist_values.values.lock() {
+        let values = match self.values.lock() {
             Ok(guard) if !guard.is_empty() => guard,
             _ => return (0, None),
         };
@@ -248,9 +233,9 @@ impl<T: Number<T>> Histogram<T> {
                 start_time: start,
                 time: t,
                 count: b.count,
-                bounds: self.hist_values.bounds.clone(),
+                bounds: self.bounds.clone(),
                 bucket_counts: b.counts.clone(),
-                sum: if self.hist_values.record_sum {
+                sum: if self.record_sum {
                     b.total
                 } else {
                     T::default()