[profiling] Fuzz profiler API

profiling/src/api.rs

@@ -128,6 +128,7 @@ pub struct Sample<'a> {
 }
 
 #[derive(Debug)]
+#[cfg_attr(test, derive(bolero_generator::TypeGenerator))]
 pub enum UpscalingInfo {
     Poisson {
         // sum_value_offset and count_value_offset are offsets in the profile values type array

profiling/src/internal/endpoint_stats.rs

@@ -19,7 +19,8 @@ impl From<HashMap<String, i64>> for ProfiledEndpointsStats {
 
 impl ProfiledEndpointsStats {
     pub fn add_endpoint_count(&mut self, endpoint_name: String, value: i64) {
-        *self.count.entry(endpoint_name).or_insert(0) += value;
+        let entry = self.count.entry(endpoint_name).or_insert(0);
+        *entry = entry.saturating_add(value);
     }
 
     pub fn is_empty(&self) -> bool {

profiling/src/internal/mod.rs

@@ -23,7 +23,6 @@ pub use label::*;
 pub use location::*;
 pub use mapping::*;
 pub use observation::*;
-pub use owned_types::*;
 pub use profile::*;
 pub use sample::*;
 pub use stack_trace::*;

profiling/src/internal/owned_types/mod.rs

@@ -0,0 +1,44 @@
+// Copyright 2021-Present Datadog, Inc. https://www.datadoghq.com/
+// SPDX-License-Identifier: Apache-2.0
+
+use crate::api;
+
+#[cfg_attr(test, derive(bolero_generator::TypeGenerator))]
+#[derive(Clone, Debug)]
+pub struct ValueType {
+    pub typ: Box<str>,
+    pub unit: Box<str>,
+}
+
+impl<'a> From<&'a api::ValueType<'a>> for ValueType {
+    #[inline]
+    fn from(value_type: &'a api::ValueType<'a>) -> Self {
+        Self {
+            typ: Box::from(value_type.r#type),
+            unit: Box::from(value_type.unit),
+        }
+    }
+}
+
+impl<'a> From<&'a ValueType> for api::ValueType<'a> {
+    fn from(value: &'a ValueType) -> Self {
+        Self::new(&value.typ, &value.unit)
+    }
+}
+
+#[cfg_attr(test, derive(bolero_generator::TypeGenerator))]
+#[derive(Clone, Debug)]
+pub struct Period {
+    pub typ: ValueType,
+    pub value: i64,
+}
+
+impl<'a> From<&'a api::Period<'a>> for Period {
+    #[inline]
+    fn from(period: &'a api::Period<'a>) -> Self {
+        Self {
+            typ: ValueType::from(&period.r#type),
+            value: period.value,
+        }
+    }
+}