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

fix a potential DHT query hang
ariescodescream · Dec 27, 2018 · 627b5f9 · 627b5f9
2 parents c3adc43 + eb3e7fe
commit 627b5f9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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.
 				}