mirror of
https://github.com/nim-lang/Nim.git
synced 2025-12-30 18:02:05 +00:00
7ff17aff5f
* Fixes #8994. FutureStream read procedure data loss no longer occurs.
* Optimises the fix for #8994.
(cherry picked from commit 21ecf64d24)
73 lines
1.3 KiB
Nim
73 lines
1.3 KiB
Nim
discard """
|
|
file: "tfuturestream.nim"
|
|
exitcode: 0
|
|
output: '''
|
|
0
|
|
1
|
|
2
|
|
3
|
|
4
|
|
5
|
|
Done
|
|
Finished
|
|
'''
|
|
"""
|
|
import asyncdispatch
|
|
|
|
var fs = newFutureStream[int]()
|
|
|
|
proc alpha() {.async.} =
|
|
for i in 0 .. 5:
|
|
await fs.write(i)
|
|
await sleepAsync(1000)
|
|
|
|
echo("Done")
|
|
fs.complete()
|
|
|
|
proc beta() {.async.} =
|
|
while not fs.finished:
|
|
let (hasValue, value) = await fs.read()
|
|
if hasValue:
|
|
echo(value)
|
|
|
|
echo("Finished")
|
|
|
|
asyncCheck alpha()
|
|
waitFor beta()
|
|
|
|
template ensureCallbacksAreScheduled =
|
|
# callbacks are called directly if the dispatcher is not running
|
|
discard getGlobalDispatcher()
|
|
|
|
proc testCompletion() {.async.} =
|
|
ensureCallbacksAreScheduled
|
|
|
|
var stream = newFutureStream[string]()
|
|
|
|
for i in 1..5:
|
|
await stream.write($i)
|
|
|
|
var readFuture = stream.readAll()
|
|
stream.complete()
|
|
yield readFuture
|
|
let data = readFuture.read()
|
|
doAssert(data.len == 5, "actual data len = " & $data.len)
|
|
|
|
waitFor testCompletion()
|
|
|
|
# TODO: Something like this should work eventually.
|
|
# proc delta(): FutureStream[string] {.async.} =
|
|
# for i in 0 .. 5:
|
|
# await sleepAsync(1000)
|
|
# result.put($i)
|
|
|
|
# return ""
|
|
|
|
# proc omega() {.async.} =
|
|
# let fut = delta()
|
|
# while not fut.finished():
|
|
# echo(await fs.takeAsync())
|
|
|
|
# echo("Finished")
|
|
|
|
# waitFor omega() |