jyapayne/Nim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix for unnecessary poll() in some uses of async

Browse source
This commit is contained in:
Julian Fondren 2019-05-15 19:57:45 -05:00
commit 22b4cf192d
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lib/pure/asyncstreams.nim
View file

@ -79,7 +79,7 @@ proc read*[T](future: FutureStream[T]): Future[(bool, T)] =
## ``FutureStream``. ## ``FutureStream``.
var resFut = newFuture[(bool, T)]("FutureStream.take") var resFut = newFuture[(bool, T)]("FutureStream.take")
let savedCb = future.cb let savedCb = future.cb
future.callback = var newCb =
proc (fs: FutureStream[T]) = proc (fs: FutureStream[T]) =
# Exit early if `resFut` is already complete. (See #8994). # Exit early if `resFut` is already complete. (See #8994).
if resFut.finished: return if resFut.finished: return
@ -100,6 +100,11 @@ proc read*[T](future: FutureStream[T]): Future[(bool, T)] =
# If the saved callback isn't nil then let's call it. # If the saved callback isn't nil then let's call it.
if not savedCb.isNil: savedCb() if not savedCb.isNil: savedCb()
if future.queue.len > 0 or future.finished:
newCb(future)
else:
future.callback = newCb
return resFut return resFut
proc len*[T](future: FutureStream[T]): int = proc len*[T](future: FutureStream[T]): int =