bugfix: Handle nil DB on get peers

dominant-strategies · Nov 18, 2024 · ca8fb99 · ca8fb99
1 parent e598bb0
commit ca8fb99
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/p2p/node/peerManager/peerManager.go b/p2p/node/peerManager/peerManager.go
@@ -510,15 +510,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 nil
 }
 
 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 nil
 }
 
 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 nil
 }
 
 func (pm *BasicPeerManager) AdjustPeerQuality(peer p2p.PeerID, topic string, adjFn func(int) int) {