check for stopping the query in between dialing and querying

4c2cf464 · Adin Schmahmann · Steven Allen · dd0bbb2c · 4c2cf464
Commit 4c2cf464 authored Feb 04, 2020 by Adin Schmahmann Committed by Steven Allen Apr 03, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

query.go query.go +5 -0

No files found.
--- 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)