fix: PeerManager signalAvailabiity() race

980ca8d4 · Dirk McCormick · 25318da3 · 980ca8d4
Commit 980ca8d4 authored Jun 10, 2020 by Dirk McCormick
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

internal/peermanager/peermanager.go internal/peermanager/peermanager.go +3 -0

No files found.
--- a/internal/peermanager/peermanager.go
+++ b/internal/peermanager/peermanager.go
@@ -231,6 +231,9 @@ func (pm *PeerManager) UnregisterSession(ses uint64) {
 // signalAvailability is called when a peer's connectivity changes.
 // It informs interested sessions.
 func (pm *PeerManager) signalAvailability(p peer.ID, isConnected bool) {
+	pm.psLk.Lock()
+	defer pm.psLk.Unlock()
+
 	sesIds, ok := pm.peerSessions[p]
 	if !ok {
 		return