Add RemovePeer().

log.go

@@ -205,9 +205,6 @@ func (l *Log) ContainsEntry(index uint64, term uint64) bool {
 // Retrieves a list of entries after a given index. This function also returns
 // the term of the index provided.
 func (l *Log) GetEntriesAfter(index uint64) ([]*LogEntry, uint64) {
-	l.mutex.Lock()
-	defer l.mutex.Unlock()
-
 	// Return an error if the index doesn't exist.
 	if index > uint64(len(l.entries)) {
 		panic(fmt.Sprintf("raft.Log: Index is beyond end of log: %v", index))

server.go

@@ -739,10 +739,17 @@ func (s *Server) RemovePeer(name string) error {
 	}
 	// Return error if peer doesn't exist.
 	peer := s.peers[name]
-	if peer != nil {
+	if peer == nil {
 		return fmt.Errorf("raft: Peer not found: %s", name)
 	}
 
+	// Flush entries to the peer first.
+	if s.state == Leader {
+		if _, _, err := peer.internalFlush(); err != nil {
+			warn("raft: Unable to notify peer of removal: %v", err)
+		}
+	}
+
 	// Stop peer and remove it.
 	peer.stop()
 	delete(s.peers, name)