close connection on chronos close

2020-05-13 11:36:10 -06:00 · 2020-05-13 11:36:10 -06:00 · 5158d96eaf
parent 46daed9a38
commit 5158d96eaf
1 changed files with 6 additions and 0 deletions
--- 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