status-im
/
nim-libp2p
mirror of https://github.com/status-im/nim-libp2p.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

getMuxedStream to more appropriate getStream (#448)

This commit is contained in:
Dmitriy Ryajov 2020-11-24 00:37:45 -06:00 committed by GitHub
parent 69ae24dc8d
commit 351489bfa9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libp2p/connmanager.nim
View File

@ -367,7 +367,7 @@ proc storeMuxer*(c: ConnManager,
asyncSpawn c.peerStartup(muxer.connection)
proc getMuxedStream*(c: ConnManager,
proc getStream*(c: ConnManager,
peerId: PeerID,
dir: Direction): Future[Connection] {.async, gcsafe.} =
## get a muxed stream for the provided peer
@ -378,7 +378,7 @@ proc getMuxedStream*(c: ConnManager,
if not(isNil(muxer)):
return await muxer.newStream()
proc getMuxedStream*(c: ConnManager,
proc getStream*(c: ConnManager,
peerId: PeerID): Future[Connection] {.async, gcsafe.} =
## get a muxed stream for the passed peer from any connection
##
@ -387,7 +387,7 @@ proc getMuxedStream*(c: ConnManager,
if not(isNil(muxer)):
return await muxer.newStream()
proc getMuxedStream*(c: ConnManager,
proc getStream*(c: ConnManager,
conn: Connection): Future[Connection] {.async, gcsafe.} =
## get a muxed stream for the passed connection
##

4
libp2p/switch.nim
View File

@ -352,7 +352,7 @@ proc dial*(s: Switch,
peerId: PeerID,
protos: seq[string]): Future[Connection] {.async.} =
trace "Dialing (existing)", peerId, protos
let stream = await s.connManager.getMuxedStream(peerId)
let stream = await s.connManager.getStream(peerId)
if stream.isNil:
raise newException(DialFailedError, "Couldn't get muxed stream")
@ -370,7 +370,7 @@ proc dial*(s: Switch,
trace "Dialing (new)", peerId, protos
let conn = await s.internalConnect(peerId, addrs)
trace "Opening stream", conn
let stream = await s.connManager.getMuxedStream(conn)
let stream = await s.connManager.getStream(conn)
proc cleanup() {.async.} =
if not(isNil(stream)):

8
tests/testconnmngr.nim
View File

@ -87,7 +87,7 @@ suite "Connection Manager":
connMngr.storeMuxer(muxer)
check muxer in connMngr
let stream = await connMngr.getMuxedStream(peer.peerId)
let stream = await connMngr.getStream(peer.peerId)
check not(isNil(stream))
check stream.peerInfo == peer
@ -107,9 +107,9 @@ suite "Connection Manager":
connMngr.storeMuxer(muxer)
check muxer in connMngr
let stream1 = await connMngr.getMuxedStream(peer.peerId, Direction.In)
let stream1 = await connMngr.getStream(peer.peerId, Direction.In)
check not(isNil(stream1))
let stream2 = await connMngr.getMuxedStream(peer.peerId, Direction.Out)
let stream2 = await connMngr.getStream(peer.peerId, Direction.Out)
check isNil(stream2)
await connMngr.close()
@ -128,7 +128,7 @@ suite "Connection Manager":
connMngr.storeMuxer(muxer)
check muxer in connMngr
let stream = await connMngr.getMuxedStream(conn)
let stream = await connMngr.getStream(conn)
check not(isNil(stream))
await connMngr.close()