upd

Signed-off-by: Akira Hayakawa <ruby.wktk@gmail.com>

lol2/src/process/peer_svc/mod.rs

@@ -32,6 +32,7 @@ struct ThreadHandles {
 }
 
 pub struct Inner {
+    membership: spin::RwLock<HashSet<NodeId>>,
     peer_contexts: spin::RwLock<HashMap<NodeId, PeerContexts>>,
     peer_threads: spin::Mutex<HashMap<NodeId, ThreadHandles>>,
 
@@ -44,6 +45,7 @@ pub struct PeerSvc(pub Arc<Inner>);
 impl PeerSvc {
     pub fn new(command_log: Ref<CommandLog>, driver: RaftDriver) -> Self {
         let inner = Inner {
+            membership: HashSet::new().into(),
             peer_contexts: HashMap::new().into(),
             peer_threads: HashMap::new().into(),
             command_log,
@@ -151,7 +153,9 @@ impl PeerSvc {
         for id in remove_peers {
             self.remove_peer(id);
         }
+
         info!("membership changed -> {:?}", config);
+        *self.membership.write() = config;
 
         self.command_log
             .membership_pointer
@@ -161,11 +165,7 @@ impl PeerSvc {
     }
 
     pub fn read_membership(&self) -> HashSet<NodeId> {
-        let mut out = HashSet::new();
-        for (id, _) in self.peer_contexts.read().iter() {
-            out.insert(id.clone());
-        }
-        out
+        self.membership.read().clone()
     }
 
     pub async fn find_new_commit_index(&self) -> Result<Index> {