fix(query): cancel the context when the query finishes

e6a4d6a3 · Steven Allen · 30ba05e2 · e6a4d6a3
Commit e6a4d6a3 authored Feb 27, 2020 by Steven Allen
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

query.go query.go +4 -0

No files found.
--- a/query.go
+++ b/query.go
@@ -125,6 +125,10 @@ func newQueryRunner(q *dhtQuery) *dhtQueryRunner {
 }

 func (r *dhtQueryRunner) Run(ctx context.Context, peers []peer.ID) (*dhtQueryResult, error) {
+	// Make sure to clean everything up when we return from this function.
+	ctx, cancel := context.WithCancel(ctx)
+	defer cancel()
+
 	r.log = logger
 	r.runCtx = ctx