mirror of https://github.com/vacp2p/nim-libp2p.git
60 lines
1.8 KiB
Nim
60 lines
1.8 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, peerinfo, multiaddress
|
|
|
|
const DefaultReadSize = 1024
|
|
|
|
type
|
|
Connection* = ref object of RootObj
|
|
reader: AsyncStreamReader
|
|
writer: AsyncStreamWriter
|
|
server: StreamServer
|
|
client: StreamTransport
|
|
isOpen*: bool
|
|
|
|
proc newConnection*(server: StreamServer,
|
|
client: StreamTransport): Connection =
|
|
## create a new Connection for the specified async stream reader/writer
|
|
new result
|
|
result.isOpen = false
|
|
result.server = server
|
|
result.client = client
|
|
|
|
result.reader = newAsyncStreamReader(client)
|
|
result.writer = newAsyncStreamWriter(client)
|
|
|
|
method read* (c: Connection, size: int = DefaultReadSize): Future[seq[byte]] {.base, async, gcsafe.} =
|
|
## read DefaultReadSize (1024) bytes or `size` bytes if specified
|
|
result = await c.reader.read(size)
|
|
|
|
method write* (c: Connection, data: pointer, size: int): Future[void] {.base, async.} =
|
|
## write bytes pointed to by `data` up to `size` size
|
|
discard c.writer.write(data, size)
|
|
|
|
method close* (c: Connection): Future[void] {.base, async.} =
|
|
## close connection
|
|
await c.reader.closeWait()
|
|
|
|
await c.writer.finish()
|
|
await c.writer.closeWait()
|
|
|
|
await c.client.closeWait()
|
|
c.server.stop()
|
|
c.server.close()
|
|
|
|
method getPeerInfo* (c: Connection): Future[PeerInfo] {.base, async.} =
|
|
## get up to date peer info
|
|
## TODO: implement PeerInfo refresh over identify
|
|
discard
|
|
|
|
method getObservedAddrs(c: Connection): Future[seq[MultiAddress]] {.base, async.} =
|
|
## get resolved multiaddresses for the connection
|
|
discard
|