status-im/go-waku
2
0
Fork 0
mirror of https://github.com/status-im/go-waku.git synced 2025-01-27 05:56:07 +00:00
Code Issues Projects Releases Wiki Activity

fix: do not block ctx.Done() on peer discovery

Browse Source
This commit is contained in:
Richard Ramos 2023-05-09 17:25:58 -04:00 committed by RichΛrd
parent e14c55c263
commit fa61e58d3b
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
waku/v2/discv5/discover.go
View File

@ -294,8 +294,13 @@ func (d *DiscoveryV5) iterate(ctx context.Context) error {
}
if len(peerAddrs) != 0 {
d.peerConnector.PeerChannel() <- peerAddrs[0]
select {
case d.peerConnector.PeerChannel() <- peerAddrs[0]:
case <-ctx.Done():
return nil
}
}
select {
case <-ctx.Done():
return nil

6
waku/v2/rendezvous/rendezvous.go
View File

@ -124,7 +124,11 @@ func (r *Rendezvous) discover(ctx context.Context) {
server.Unlock()
for _, addr := range addrInfo {
r.peerConnector.PeerChannel() <- addr
select {
case r.peerConnector.PeerChannel() <- addr:
case <-ctx.Done():
return
}
}
} else {
// TODO: change log level to DEBUG in go-libp2p-rendezvous@v0.4.1/svc.go:234 discover query