codex-storage
/
nim-libp2p
mirror of https://github.com/codex-storage/nim-libp2p.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use loop on closed flag

This commit is contained in:
Dmitriy Ryajov 2019-08-24 13:15:45 -06:00
parent 402067ceae
commit f24c866a30
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libp2p/multistreamselect.nim
View File

@ -64,7 +64,7 @@ proc handle*(m: MultisteamSelect, conn: Connection) {.async.} =
if not (await m.select(conn)): if not (await m.select(conn)):
return return
block handleLoop: while not conn.closed:
var ms = cast[string](await conn.readLp()) var ms = cast[string](await conn.readLp())
ms.removeSuffix("\n") ms.removeSuffix("\n")
if ms.len() <= 0: if ms.len() <= 0:
@ -73,16 +73,14 @@ proc handle*(m: MultisteamSelect, conn: Connection) {.async.} =
case ms: case ms:
of "ls": of "ls":
for h in m.handlers: for h in m.handlers:
await conn.writeLp(h.proto) await conn.writeLp(h.proto & "\n")
break handleLoop
else: else:
for h in m.handlers: for h in m.handlers:
if (not isNil(h.match) and h.match(ms)) or ms == h.proto: if (not isNil(h.match) and h.match(ms)) or ms == h.proto:
await h.handler(conn, ms) await h.handler(conn, ms)
break return
else: else:
await conn.write(Na) await conn.write(Na)
break handleLoop
proc addHandler*(m: MultisteamSelect, proc addHandler*(m: MultisteamSelect,
proto: string, proto: string,