refreshing the routing table after the dht context is cancelled retur…

…ns an error instead of hanging
libp2p · Apr 3, 2020 · f3318e8 · f3318e8
1 parent c79d566
commit f3318e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dht_bootstrap.go b/dht_bootstrap.go
@@ -248,6 +248,10 @@ func (dht *IpfsDHT) Bootstrap(_ context.Context) error {
 // error and close. The channel is buffered and safe to ignore.
 func (dht *IpfsDHT) RefreshRoutingTable() <-chan error {
 	res := make(chan error, 1)
-	dht.triggerRtRefresh <- res
+	select {
+	case dht.triggerRtRefresh <- res:
+	case <-dht.ctx.Done():
+		res <- dht.ctx.Err()
+	}
 	return res
 }