From 5158d96eaf88e1b98f4929913369a32ffa9478b0 Mon Sep 17 00:00:00 2001
From: Dmitriy Ryajov <dryajov@gmail.com>
Date: Wed, 13 May 2020 11:36:10 -0600
Subject: [PATCH] close connection on chronos close

---
 libp2p/stream/chronosstream.nim | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/libp2p/stream/chronosstream.nim b/libp2p/stream/chronosstream.nim
index b3af6b682..b9baa1968 100644
--- a/libp2p/stream/chronosstream.nim
+++ b/libp2p/stream/chronosstream.nim
@@ -16,6 +16,12 @@ logScope:
 type ChronosStream* = ref object of LPStream
     client: StreamTransport
 
+proc onTransportClose(s: ChronosStream,
+                      client: StreamTransport) {.async.} =
+  await client.join()
+  trace "Transport closed, closing connection"
+  await s.close()
+
 proc newChronosStream*(client: StreamTransport): ChronosStream =
   new result
   result.client = client