Merge pull request #219 from libp2p/fix/dht-query-hang

fix a potential DHT query hang

Merge pull request #219 from libp2p/fix/dht-query-hang
fix a potential DHT query hang
627b5f96 · Steven Allen · GitHub · c3adc437 · eb3e7fe9 · 627b5f96
Unverified Commit 627b5f96 authored Dec 27, 2018 by Steven Allen Committed by GitHub Dec 27, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

query.go query.go +2 -0

No files found.
--- a/query.go
+++ b/query.go
@@ -209,6 +209,8 @@ func (r *dhtQueryRunner) spawnWorkers(proc process.Process) {
 			select {
 			case p, more := <-r.peersToQuery.DeqChan:
 				if !more {
+					// Put this back so we can finish any outstanding queries.
+					r.rateLimit <- struct{}{}
 					return // channel closed.
 				}