check for stopping the query in between dialing and querying

libp2p · Mar 4, 2020 · 66d3bb4 · 66d3bb4
1 parent 29f8bb3
commit 66d3bb4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/query.go b/query.go
@@ -280,6 +280,11 @@ func (q *qu) queryPeer(ctx context.Context, p peer.ID) *queryResult {
 		return &queryResult{}
 	}
 
+	if q.stopFn(q.localPeers) {
+		q.cancel()
+		return &queryResult{}
+	}
+
 	newPeers, err := q.queryFn(queryCtx, p)
 	if err != nil {
 		q.localPeers.Remove(p)