From 317e57a8a8398bad01de1a08c7bc5b5c391d4ec3 Mon Sep 17 00:00:00 2001 From: Amit Gupta Date: Wed, 11 Feb 2015 18:37:02 -0800 Subject: [PATCH] rafthttp: Panic informatively when removing unknown peer ID --- rafthttp/transport.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/rafthttp/transport.go b/rafthttp/transport.go index d137cfe9a..dc3e25c0c 100644 --- a/rafthttp/transport.go +++ b/rafthttp/transport.go @@ -132,7 +132,11 @@ func (t *transport) AddPeer(id types.ID, urls []string) { func (t *transport) RemovePeer(id types.ID) { t.mu.Lock() defer t.mu.Unlock() - t.peers[id].Stop() + if peer, ok := t.peers[id]; ok { + peer.Stop() + } else { + log.Panicf("rafthttp: unexpected removal of unknown peer '%d'", id) + } delete(t.peers, id) delete(t.leaderStats.Followers, id.String()) }