mirror of https://github.com/vacp2p/nim-libp2p.git
Fix subclass calls to handleDisconnect (#314)
* Fix subclass calls to handleDisconnect * add peer ID to nil peer debug message
This commit is contained in:
parent
5c986cf657
commit
9bbe5e4841
|
@ -36,7 +36,7 @@ method subscribeTopic*(f: FloodSub,
|
||||||
|
|
||||||
let peer = f.peers.getOrDefault(peerId)
|
let peer = f.peers.getOrDefault(peerId)
|
||||||
if peer == nil:
|
if peer == nil:
|
||||||
debug "subscribeTopic on a nil peer!"
|
debug "subscribeTopic on a nil peer!", peer = peerId
|
||||||
return
|
return
|
||||||
|
|
||||||
if topic notin f.floodsub:
|
if topic notin f.floodsub:
|
||||||
|
@ -53,12 +53,15 @@ method subscribeTopic*(f: FloodSub,
|
||||||
|
|
||||||
method handleDisconnect*(f: FloodSub, peer: PubSubPeer) =
|
method handleDisconnect*(f: FloodSub, peer: PubSubPeer) =
|
||||||
## handle peer disconnects
|
## handle peer disconnects
|
||||||
|
##
|
||||||
|
|
||||||
|
procCall PubSub(f).handleDisconnect(peer)
|
||||||
|
|
||||||
|
if not(isNil(peer)) and peer.peerInfo notin f.conns:
|
||||||
for t in toSeq(f.floodsub.keys):
|
for t in toSeq(f.floodsub.keys):
|
||||||
if t in f.floodsub:
|
if t in f.floodsub:
|
||||||
f.floodsub[t].excl(peer)
|
f.floodsub[t].excl(peer)
|
||||||
|
|
||||||
procCall PubSub(f).handleDisconnect(peer)
|
|
||||||
|
|
||||||
method rpcHandler*(f: FloodSub,
|
method rpcHandler*(f: FloodSub,
|
||||||
peer: PubSubPeer,
|
peer: PubSubPeer,
|
||||||
rpcMsgs: seq[RPCMsg]) {.async.} =
|
rpcMsgs: seq[RPCMsg]) {.async.} =
|
||||||
|
|
|
@ -245,7 +245,11 @@ proc heartbeat(g: GossipSub) {.async.} =
|
||||||
|
|
||||||
method handleDisconnect*(g: GossipSub, peer: PubSubPeer) =
|
method handleDisconnect*(g: GossipSub, peer: PubSubPeer) =
|
||||||
## handle peer disconnects
|
## handle peer disconnects
|
||||||
|
##
|
||||||
|
|
||||||
procCall FloodSub(g).handleDisconnect(peer)
|
procCall FloodSub(g).handleDisconnect(peer)
|
||||||
|
|
||||||
|
if not(isNil(peer)) and peer.peerInfo notin g.conns:
|
||||||
for t in toSeq(g.gossipsub.keys):
|
for t in toSeq(g.gossipsub.keys):
|
||||||
g.gossipsub.removePeer(t, peer)
|
g.gossipsub.removePeer(t, peer)
|
||||||
|
|
||||||
|
@ -284,7 +288,7 @@ method subscribeTopic*(g: GossipSub,
|
||||||
|
|
||||||
let peer = g.peers.getOrDefault(peerId)
|
let peer = g.peers.getOrDefault(peerId)
|
||||||
if peer == nil:
|
if peer == nil:
|
||||||
debug "subscribeTopic on a nil peer!"
|
# floodsub method logs a debug line already
|
||||||
return
|
return
|
||||||
|
|
||||||
if subscribe:
|
if subscribe:
|
||||||
|
|
|
@ -63,6 +63,7 @@ type
|
||||||
method handleDisconnect*(p: PubSub, peer: PubSubPeer) {.base.} =
|
method handleDisconnect*(p: PubSub, peer: PubSubPeer) {.base.} =
|
||||||
## handle peer disconnects
|
## handle peer disconnects
|
||||||
##
|
##
|
||||||
|
|
||||||
if not(isNil(peer)) and peer.peerInfo notin p.conns:
|
if not(isNil(peer)) and peer.peerInfo notin p.conns:
|
||||||
trace "deleting peer", peer = peer.id
|
trace "deleting peer", peer = peer.id
|
||||||
peer.onConnect.fire() # Make sure all pending sends are unblocked
|
peer.onConnect.fire() # Make sure all pending sends are unblocked
|
||||||
|
|
Loading…
Reference in New Issue