Trigger prune for removed peers (#361)

This commit is contained in:
petricadaipegsp 2024-11-17 13:32:54 +01:00 committed by GitHub
parent f511149b36
commit ceba9ff200
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -633,6 +633,19 @@ loop:
func (bs *BlossomSubRouter) RemovePeer(p peer.ID) {
log.Debugf("PEERDOWN: Remove disconnected peer %s", p)
masks := make([][]byte, 0)
bs.meshMx.Lock()
for bitmask, peers := range bs.mesh {
if _, ok := peers[p]; !ok {
continue
}
masks = append(masks, []byte(bitmask))
}
bs.meshMx.Unlock()
for _, bitmask := range masks {
log.Debugf("PEERDOWN: Pruning peer %s from bitmask %s", p, bitmask)
bs.tracer.Prune(p, bitmask)
}
bs.tracer.RemovePeer(p)
delete(bs.peers, p)
bs.meshMx.Lock()