mirror of https://github.com/vacp2p/nim-libp2p.git
73 lines
2.2 KiB
Nim
73 lines
2.2 KiB
Nim
## Nim-LibP2P
|
|
## Copyright (c) 2018 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
|
|
import chronos, chronicles
|
|
import ../protocols/protocol,
|
|
../connection
|
|
|
|
logScope:
|
|
topic = "Muxer"
|
|
|
|
type
|
|
StreamHandler* = proc(conn: Connection): Future[void] {.gcsafe.}
|
|
MuxerHandler* = proc(muxer: Muxer): Future[void] {.gcsafe.}
|
|
|
|
Muxer* = ref object of RootObj
|
|
streamHandler*: StreamHandler
|
|
connection*: Connection
|
|
|
|
MuxerCreator* = proc(conn: Connection): Muxer {.gcsafe, closure.}
|
|
# this wraps a creator proc that knows how to make muxers
|
|
MuxerProvider* = ref object of LPProtocol
|
|
newMuxer*: MuxerCreator
|
|
streamHandler*: StreamHandler
|
|
muxerHandler*: MuxerHandler
|
|
|
|
method newStream*(m: Muxer, name: string = ""): Future[Connection] {.base, async, gcsafe.} = discard
|
|
method close*(m: Muxer) {.base, async, gcsafe.} = discard
|
|
method handle*(m: Muxer): Future[void] {.base, async, gcsafe.} = discard
|
|
method `streamHandler=`*(m: Muxer, handler: StreamHandler) {.base, gcsafe.} =
|
|
m.streamHandler = handler
|
|
|
|
proc newMuxerProvider*(creator: MuxerCreator, codec: string): MuxerProvider {.gcsafe.} =
|
|
new result
|
|
result.newMuxer = creator
|
|
result.codec = codec
|
|
result.init()
|
|
|
|
method `streamHandler=`*(m: MuxerProvider, handler: StreamHandler) {.base, gcsafe.} =
|
|
## new stream (channels) handler
|
|
##
|
|
## triggered every time there is a new
|
|
## stream (channel) oppened over a muxed
|
|
## connection
|
|
##
|
|
m.streamHandler = handler
|
|
|
|
method `muxerHandler=`*(m: MuxerProvider, handler: MuxerHandler) {.base, gcsafe.} =
|
|
## new muxer (muxed connections) handler
|
|
##
|
|
## triggered every time there is a new muxed
|
|
## connection created
|
|
##
|
|
m.muxerHandler = handler
|
|
|
|
method init(c: MuxerProvider) =
|
|
proc handler(conn: Connection, proto: string) {.async, gcsafe, closure.} =
|
|
let muxer = c.newMuxer(conn)
|
|
if not isNil(c.muxerHandler):
|
|
asyncCheck c.muxerHandler(muxer)
|
|
|
|
if not isNil(c.streamHandler):
|
|
muxer.streamHandler = c.streamHandler
|
|
|
|
await muxer.handle()
|
|
|
|
c.handler = handler
|