Fix for ThreeMammals#15

biesigrr · Jul 18, 2018 · c46ec79 · c46ec79
1 parent 98e7d50
commit c46ec79
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/Rafty/Concensus/States/Leader.cs b/src/Rafty/Concensus/States/Leader.cs
@@ -192,6 +192,12 @@ private async Task SendAppendEntries(int electionTimerId)
                     var nextIndex = new NextIndex(p, await _log.LastLogIndex());
                     PeerStates.Add(new PeerState(p, matchIndex, nextIndex));
                 });
+
+                var peerStatesNotInPeers = PeerStates.Where(p => !peers.Select(x => x.Id).Contains(p.Peer.Id)).ToList();
+                foreach (var peerState in peerStatesNotInPeers)
+                {
+                    PeerStates.Remove(peerState);
+                }
             }
 
             var appendEntriesResponses = SetUpAppendingEntries();