From 7728124b5ee1a448916ff598b8acb1ebe7cfdad7 Mon Sep 17 00:00:00 2001
From: Jeromy <why@ipfs.io>
Date: Thu, 3 Nov 2016 23:44:42 -0700
Subject: [PATCH] respect contexts when returning diagnostics responses

License: MIT
Signed-off-by: Jeromy <why@ipfs.io>
---
 diagnostics/diag.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/diagnostics/diag.go b/diagnostics/diag.go
index da76a6fb0..4022b8590 100644
--- 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)
 	}