prevent already closed errors

2019-09-03 19:41:28 -06:00 · 2019-09-03 19:41:28 -06:00 · cc26d4e040
parent 8549ce76bb
commit cc26d4e040
1 changed files with 9 additions and 5 deletions
--- a/libp2p/stream/chronosstream.nim
+++ b/libp2p/stream/chronosstream.nim
@ -55,10 +55,14 @@ method write*(s: ChronosStream, msg: seq[byte], msglen = -1) {.async, gcsafe.} =
  await s.writer.write(msg, msglen)

 method close*(s: ChronosStream) {.async, gcsafe.} =
-  await s.reader.closeWait()
+  if not s.closed:
+    if not s.reader.closed:
+      await s.reader.closeWait()

-  await s.writer.finish()
-  await s.writer.closeWait()
+    await s.writer.finish()

-  await s.client.closeWait()
-  s.closed = true
+    if not s.writer.closed:
+      await s.writer.closeWait()
+
+    await s.client.closeWait()
+    s.closed = true