Add PerformanceMetricsStorageAdapter::reset()

LDeakin · Oct 31, 2024 · 952f543 · 952f543
1 parent 68b9b29
commit 952f543
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/zarrs_storage/CHANGELOG.md b/zarrs_storage/CHANGELOG.md
@@ -9,6 +9,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
  - Add `ByteRange::new` and `From` for `RangeBounds<u64>`
+ - Add `PerformanceMetricsStorageAdapter::reset()`
 
 ### Changed
  - Bump `unsafe_cell_slice` to 0.2.0

diff --git a/zarrs_storage/src/storage_adapter/performance_metrics.rs b/zarrs_storage/src/storage_adapter/performance_metrics.rs
@@ -42,6 +42,14 @@ impl<TStorage: ?Sized> PerformanceMetricsStorageAdapter<TStorage> {
         }
     }
 
+    /// Reset the performance metrics.
+    pub fn reset(&self) {
+        self.bytes_read.store(0, Ordering::Relaxed);
+        self.bytes_written.store(0, Ordering::Relaxed);
+        self.reads.store(0, Ordering::Relaxed);
+        self.writes.store(0, Ordering::Relaxed);
+    }
+
     /// Returns the number of bytes read.
     pub fn bytes_read(&self) -> usize {
         self.bytes_read.load(Ordering::Relaxed)