bug 1704106 - Expose timespan's set_raw to Rust consumers

CHANGELOG.md

@@ -4,6 +4,7 @@
 
 * RLB
   * Provide an internal-use-only API to pass in raw samples for timing distributions ([#1561](https://github.com/mozilla/glean/pull/1561)).
+  * Expose Timespan's `set_raw` to Rust ([#1578](https://github.com/mozilla/glean/pull/1578)).
 * Android
   * BUGFIX: `TimespanMetricType.measure` and `TimingDistributionMetricType.measure` won't get inlined anymore ([#1560](https://github.com/mozilla/glean/pull/1560)).
     This avoids a potential bug where a `return` used inside the closure would end up not measuring the time.

glean-core/rlb/src/private/timespan.rs

@@ -3,6 +3,7 @@
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
 use std::sync::{Arc, RwLock};
+use std::time::Duration;
 
 use inherent::inherent;
 
@@ -68,6 +69,16 @@ impl glean_core::traits::Timespan for TimespanMetric {
         });
     }
 
+    fn set_raw(&self, elapsed: Duration) {
+        let metric = Arc::clone(&self.0);
+        crate::launch_with_glean(move |glean| {
+            let inner = metric
+                .write()
+                .expect("Lock poisoned for timespan metric on set_raw.");
+            inner.set_raw(glean, elapsed)
+        });
+    }
+
     fn test_get_value<'a, S: Into<Option<&'a str>>>(&self, ping_name: S) -> Option<u64> {
         crate::block_on_dispatcher();
 

glean-core/src/traits/timespan.rs

@@ -3,6 +3,7 @@
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
 use crate::ErrorType;
+use std::time::Duration;
 
 /// A description for the [`TimespanMetric`](crate::metrics::TimespanMetric) type.
 ///
@@ -28,6 +29,17 @@ pub trait Timespan {
     /// if no [`start`](Timespan::start) was called.
     fn cancel(&self);
 
+    /// Explicitly sets the timespan value.
+    ///
+    /// This API should only be used if your library or application requires recording
+    /// spans of time in a way that cannot make use of
+    /// [`start`](Timespan::start)/[`stop`](Timespan::stop)/[`cancel`](Timespan::cancel).
+    ///
+    /// # Arguments
+    ///
+    /// * `elapsed` - The elapsed time to record.
+    fn set_raw(&self, elapsed: Duration);
+
     /// **Exported for test purposes.**
     ///
     /// Gets the currently stored value as an integer.