add debug assert to sure no case to call get_size recursively

initia-labs · Nov 1, 2024 · 0a30558 · 0a30558
1 parent f8dd0ef
commit 0a30558
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/crates/storage/src/allocator.rs b/crates/storage/src/allocator.rs
@@ -31,12 +31,14 @@ static GLOBAL: SizeCounterAllocator = SizeCounterAllocator;
 
 #[inline]
 fn start_metering() {
+    debug_assert!(!METERING.with(|metering| metering.get()));
     SIZE.with(|size| size.set(0));
     METERING.with(|metering| metering.set(true));
 }
 
 #[inline]
 fn finish_metering() -> usize {
+    debug_assert!(METERING.with(|metering| metering.get()));
     METERING.with(|metering| metering.set(false));
     SIZE.with(|size| size.get())
 }