浏览代码

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()
+    }
 }