diff --git a/src/collector/otlp/service.rs b/src/collector/otlp/service.rs index ffd7846..19c9bab 100644 --- a/src/collector/otlp/service.rs +++ b/src/collector/otlp/service.rs @@ -370,6 +370,7 @@ fn process_sample( let kind = match (sample_type_type, sample_type_unit) { ("samples", "count") => SampleKind::OnCPU, ("events", "nanoseconds") => SampleKind::OffCPU, + ("events", "count") => SampleKind::UProbe, _ => SampleKind::Unknown, }; diff --git a/src/storage/tables/traceevents.rs b/src/storage/tables/traceevents.rs index 428aa59..4076480 100644 --- a/src/storage/tables/traceevents.rs +++ b/src/storage/tables/traceevents.rs @@ -31,6 +31,7 @@ pub enum SampleKind { Mixed, OnCPU, OffCPU, + UProbe, // _MaxKind should always be the last entry // in this enum. _MaxKind, @@ -45,6 +46,7 @@ impl TryFrom for SampleKind { 1 => Ok(SampleKind::Mixed), 2 => Ok(SampleKind::OnCPU), 3 => Ok(SampleKind::OffCPU), + 4 => Ok(SampleKind::UProbe), _ => Err(()), } } diff --git a/src/ui/app.rs b/src/ui/app.rs index ca85e5e..ca60e96 100644 --- a/src/ui/app.rs +++ b/src/ui/app.rs @@ -268,6 +268,7 @@ impl DevfilerUi { ui.selectable_value(&mut self.kind, SampleKind::Mixed, "Mixed"); ui.selectable_value(&mut self.kind, SampleKind::OnCPU, "On CPU"); ui.selectable_value(&mut self.kind, SampleKind::OffCPU, "Off CPU"); + ui.selectable_value(&mut self.kind, SampleKind::UProbe, "UProbe"); }); ui.label("Sample kind:");