mirror of
https://github.com/status-im/nim-libp2p.git
synced 2025-02-28 04:20:39 +00:00
moved util funcs to utils module
This commit is contained in:
parent
c61b9dff61
commit
7a49b6d962
@ -7,7 +7,6 @@
|
|||||||
## This file may not be copied, modified, or distributed except according to
|
## This file may not be copied, modified, or distributed except according to
|
||||||
## those terms.
|
## those terms.
|
||||||
|
|
||||||
import macros
|
|
||||||
import chronos
|
import chronos
|
||||||
|
|
||||||
# TODO: Rework without methods
|
# TODO: Rework without methods
|
||||||
@ -20,7 +19,6 @@ type
|
|||||||
|
|
||||||
Stream*[T] = ref object of RootObj
|
Stream*[T] = ref object of RootObj
|
||||||
name*: string
|
name*: string
|
||||||
closed*: bool
|
|
||||||
eof*: bool
|
eof*: bool
|
||||||
sourceImpl*: proc (s: Stream[T]): Source[T] {.gcsafe.}
|
sourceImpl*: proc (s: Stream[T]): Source[T] {.gcsafe.}
|
||||||
sinkImpl*: proc(s: Stream[T]): Sink[T] {.gcsafe.}
|
sinkImpl*: proc(s: Stream[T]): Sink[T] {.gcsafe.}
|
||||||
@ -44,42 +42,5 @@ proc atEof*[T](s: Stream[T]): bool =
|
|||||||
s.eof
|
s.eof
|
||||||
|
|
||||||
proc close*[T](s: Stream[T]) {.async.} =
|
proc close*[T](s: Stream[T]) {.async.} =
|
||||||
s.closed = true
|
# close is called externally
|
||||||
|
s.eof = true
|
||||||
template toFuture*[T](v: T): Future[T] =
|
|
||||||
var fut = newFuture[T]()
|
|
||||||
fut.complete(v)
|
|
||||||
fut
|
|
||||||
|
|
||||||
proc toThrough*[T](s: Stream[T]): Through[T] =
|
|
||||||
proc sinkit(i: Source[T]) {.async.} =
|
|
||||||
await s.sink()(i)
|
|
||||||
|
|
||||||
return proc(i: Source[T]): Source[T] =
|
|
||||||
asyncCheck sinkit(i)
|
|
||||||
s.source()
|
|
||||||
|
|
||||||
template pipe*[T](s: Stream[T] | Source[T],
|
|
||||||
t: varargs[Through[T]]): Source[T] =
|
|
||||||
var pipeline: Source[T]
|
|
||||||
when s is Source[T]:
|
|
||||||
pipeline = s
|
|
||||||
elif s is Stream[T]:
|
|
||||||
pipeline = s.source
|
|
||||||
|
|
||||||
for i in t:
|
|
||||||
pipeline = i(pipeline)
|
|
||||||
|
|
||||||
pipeline
|
|
||||||
|
|
||||||
template pipe*[T](s: Stream[T] | Source[T],
|
|
||||||
t: varargs[Through[T]],
|
|
||||||
x: Stream[T] | Sink[T]): Future[void] =
|
|
||||||
var pipeline = pipe(s, t)
|
|
||||||
var terminal: Future[void]
|
|
||||||
when x is Stream[T]:
|
|
||||||
terminal = x.sink()(pipeline)
|
|
||||||
elif x is Sink[T]:
|
|
||||||
terminal = x(pipeline)
|
|
||||||
|
|
||||||
terminal
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user