Merge pull request OptimatistOpenSource#31 from Thaumy/into-fixed

Add `into_fixed` fn for groups
Thaumy · Mar 21, 2024 · 3a5c531 · 3a5c531
2 parents 78284a2 + cf8ac51
commit 3a5c531
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/perf_event/counting/group/mod.rs b/src/perf_event/counting/group/mod.rs
@@ -81,10 +81,14 @@ impl CounterGroup {
 
     pub fn enable(self) -> io::Result<FixedCounterGroup> {
         self.inner_mut().enable()?;
-        FixedCounterGroup::new(self.inner).wrap_ok()
+        self.into_fixed()
     }
 
     pub fn stat(&mut self) -> io::Result<CounterGroupStat> {
         self.inner_mut().stat()
     }
+
+    pub fn into_fixed(self) -> io::Result<FixedCounterGroup> {
+        FixedCounterGroup::new(self.inner).wrap_ok()
+    }
 }
diff --git a/src/perf_event/sampling/group/mod.rs b/src/perf_event/sampling/group/mod.rs
@@ -74,7 +74,7 @@ impl SamplerGroup {
 
     pub fn enable(self) -> io::Result<FixedSamplerGroup> {
         self.inner().enable()?;
-        FixedSamplerGroup::new(self.inner).wrap_ok()
+        self.into_fixed()
     }
 
     pub fn next_record(&self, guard: &SamplerGuard) -> Option<Record> {
@@ -84,4 +84,8 @@ impl SamplerGroup {
     pub fn stat(&mut self) -> io::Result<SamplerGroupStat> {
         self.inner_mut().stat()
     }
+
+    pub fn into_fixed(self) -> io::Result<FixedSamplerGroup> {
+        FixedSamplerGroup::new(self.inner).wrap_ok()
+    }
 }