properly handle empty messages

2019-10-02 15:47:07 -06:00 · 2019-10-02 15:47:07 -06:00 · cfd31c5251
parent 60b5b2dc2e
commit cfd31c5251
1 changed files with 6 additions and 1 deletions
--- a/libp2p/muxers/mplex/mplex.nim
+++ b/libp2p/muxers/mplex/mplex.nim
@ -55,7 +55,12 @@ method handle*(m: Mplex) {.async, gcsafe.} =
  trace "starting mplex main loop"
  try:
    while not m.connection.closed:
-      let (id, msgType, data) = (await m.connection.readMsg()).get()
+      let msg = await m.connection.readMsg()
+      if msg.isNone:
+        await sleepAsync(10.millis)
+        continue
+
+      let (id, msgType, data) = msg.get()
      let initiator = bool(ord(msgType) and 1)
      var channel: LPChannel
      if MessageType(msgType) != MessageType.New: