Переглянути джерело

Be able to report the current size of the LoadBalancer's message store

Ian Goldberg 11 місяців тому
батько
коміт
b8d48d7a77
2 змінених файлів з 10 додано та 0 видалено
  1. 6 0
      sparta/src/load_balancer.rs
  2. 4 0
      sparta/src/omap.rs

+ 6 - 0
sparta/src/load_balancer.rs

@@ -355,4 +355,10 @@ impl LoadBalancer {
 
         responses.drain(0..num_requests).map(|r| r.0).collect()
     }
+
+    pub fn size(&self) -> usize {
+        self.submaps
+            .iter()
+            .fold(0usize, |acc, submap| acc + submap.size())
+    }
 }

+ 4 - 0
sparta/src/omap.rs

@@ -130,4 +130,8 @@ impl ObliviousMap {
         self.message_store.truncate(final_size);
         response
     }
+
+    pub fn size(&self) -> usize {
+        self.message_store.len()
+    }
 }