respect contexts when returning diagnostics responses

License: MIT Signed-off-by: Jeromy <why@ipfs.io>

respect contexts when returning diagnostics responses
License: MIT Signed-off-by: Jeromy <why@ipfs.io>
7728124b · Jeromy · 1ca2d428 · 7728124b
Commit 7728124b authored Nov 03, 2016 by Jeromy
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

diagnostics/diag.go diagnostics/diag.go +5 -1

No files found.
--- a/diagnostics/diag.go
+++ b/diagnostics/diag.go
@@ -191,7 +191,11 @@ func (d *Diagnostics) getDiagnosticFromPeers(ctx context.Context, peers map[peer
 				return
 			}
 			for d := range out {
-				respdata <- d
+				select {
+				case respdata <- d:
+				case <-ctx.Done():
+					return
+				}
 			}
 		}(p)
 	}