소스 검색

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

Ian Goldberg 1 년 전
부모
커밋
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()
+    }
 }