bugfix: Handle nil DB on get peers

dominant-strategies · Nov 19, 2024 · 82e490f · 82e490f
1 parent a8086bf
commit 82e490f
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/p2p/node/peerManager/peerManager.go b/p2p/node/peerManager/peerManager.go
@@ -511,15 +511,24 @@ func (pm *BasicPeerManager) queryDHT(topic *pubsubManager.Topic, peerList map[p2
 }
 
 func (pm *BasicPeerManager) getBestPeers(topic string) map[p2p.PeerID]struct{} {
-	return pm.getPeersHelper(pm.peerDBs[topic][Best], c_minBestPeersFromDb)
+	if db, ok := pm.peerDBs[topic]; ok {
+		return pm.getPeersHelper(db[Best], c_minBestPeersFromDb)
+	}
+	return make(map[peer.ID]struct{})
 }
 
 func (pm *BasicPeerManager) getResponsivePeers(topic string) map[p2p.PeerID]struct{} {
-	return pm.getPeersHelper(pm.peerDBs[topic][Responsive], c_minResponsivePeersFromDb)
+	if db, ok := pm.peerDBs[topic]; ok {
+		return pm.getPeersHelper(db[Responsive], c_minBestPeersFromDb)
+	}
+	return make(map[peer.ID]struct{})
 }
 
 func (pm *BasicPeerManager) getLastResortPeers(topic string) map[p2p.PeerID]struct{} {
-	return pm.getPeersHelper(pm.peerDBs[topic][LastResort], c_minLastResortPeersFromDb)
+	if db, ok := pm.peerDBs[topic]; ok {
+		return pm.getPeersHelper(db[LastResort], c_minBestPeersFromDb)
+	}
+	return make(map[peer.ID]struct{})
 }
 
 func (pm *BasicPeerManager) AdjustPeerQuality(peer p2p.PeerID, topic string, adjFn func(int) int) {