run format check

Author: MoonBoi9001

graph/src/components/metrics/block_state.rs

@@ -201,11 +201,16 @@ impl BlockStateMetrics {
         }
     }
 
-    pub fn track_storage_size_change(&mut self, entity_type: &EntityType, entity: &Entity, is_removal: bool) {
+    pub fn track_storage_size_change(
+        &mut self,
+        entity_type: &EntityType,
+        entity: &Entity,
+        is_removal: bool,
+    ) {
         if ENV_VARS.enable_dips_metrics {
             let key = CounterKey::Entity(entity_type.clone(), entity.id());
             let size = entity.weight() as u64;
-            
+
             let storage = self.current_storage_size.entry(key).or_insert(0);
             if is_removal {
                 *storage = storage.saturating_sub(size);
@@ -215,7 +220,12 @@ impl BlockStateMetrics {
         }
     }
 
-    pub fn track_storage_size_change_batch(&mut self, entity_type: &EntityType, entities: &[Entity], is_removal: bool) {
+    pub fn track_storage_size_change_batch(
+        &mut self,
+        entity_type: &EntityType,
+        entities: &[Entity],
+        is_removal: bool,
+    ) {
         if ENV_VARS.enable_dips_metrics {
             for entity in entities {
                 self.track_storage_size_change(entity_type, entity, is_removal);

graph/src/components/store/write.rs

@@ -497,21 +497,24 @@ impl RowGroup {
     }
 
     pub fn track_metrics(&self, metrics: &mut BlockStateMetrics) {
-        // Track entity count changes
-        let changes: Vec<i32> = self.rows.iter()
+        let changes: Vec<i32> = self
+            .rows
+            .iter()
             .map(|row| row.entity_count_change())
             .collect();
-        metrics.track_entity_count_change_batch(&self.entity_type, &changes);
 
-        // Track storage changes and writes
-        let (writes, removals): (Vec<_>, Vec<_>) = self.rows.iter()
+        let (writes, removals): (Vec<_>, Vec<_>) = self
+            .rows
+            .iter()
             .filter_map(|row| match row {
-                EntityModification::Insert { data, .. } | 
-                EntityModification::Overwrite { data, .. } => Some((data, false)),
+                EntityModification::Insert { data, .. }
+                | EntityModification::Overwrite { data, .. } => Some((data, false)),
                 EntityModification::Remove { .. } => None,
             })
             .unzip();
 
+        metrics.track_entity_count_change_batch(&self.entity_type, &changes);
+
         if !writes.is_empty() {
             metrics.track_entity_write_batch(&self.entity_type, &writes);
             metrics.track_storage_size_change_batch(&self.entity_type, &writes, false);

runtime/wasm/src/host_exports.rs

@@ -350,8 +350,10 @@ impl HostExports {
 
         state.metrics.track_entity_write(&entity_type, &entity);
 
-        state.metrics.track_storage_size_change(&entity_type, &entity, false);
-        
+        state
+            .metrics
+            .track_storage_size_change(&entity_type, &entity, false);
+
         if !state.entity_cache.contains_key(&key) {
             state.metrics.track_entity_count_change(&entity_type, 1);
         }
@@ -395,8 +397,10 @@ impl HostExports {
         )?;
 
         if let Some(entity) = state.entity_cache.get(&key, GetScope::Store)? {
-            state.metrics.track_storage_size_change(&entity_type, &entity, true);
-            
+            state
+                .metrics
+                .track_storage_size_change(&entity_type, &entity, true);
+
             state.metrics.track_entity_count_change(&entity_type, -1);
         }