forked from goodboy/tractor
1
0
Fork 0

Terminate async gen example caller to avoid (benign) errors in console output

debugger_test_tweaks
Tyler Goodlet 2021-08-02 21:49:15 -04:00
parent 7431e8ea01
commit ace1b1312c
1 changed files with 21 additions and 19 deletions

View File

@ -1,23 +1,20 @@
from typing import AsyncIterator
from itertools import repeat from itertools import repeat
import trio import trio
import tractor import tractor
tractor.log.get_console_log("INFO")
async def stream_forever() -> AsyncIterator[int]:
async def stream_forever():
for i in repeat("I can see these little future bubble things"): for i in repeat("I can see these little future bubble things"):
# each yielded value is sent over the ``Channel`` to the # each yielded value is sent over the ``Channel`` to the parent actor
# parent actor
yield i yield i
await trio.sleep(0.01) await trio.sleep(0.01)
async def main(): async def main():
# stream for at most 1 seconds
with trio.move_on_after(1) as cancel_scope:
async with tractor.open_nursery() as n: async with tractor.open_nursery() as n:
portal = await n.start_actor( portal = await n.start_actor(
@ -28,12 +25,17 @@ async def main():
# this async for loop streams values from the above # this async for loop streams values from the above
# async generator running in a separate process # async generator running in a separate process
async with portal.open_stream_from(stream_forever) as stream: async with portal.open_stream_from(stream_forever) as stream:
count = 0
async for letter in stream: async for letter in stream:
print(letter) print(letter)
count += 1
# we support trio's cancellation system if count > 50:
assert cancel_scope.cancelled_caught break
assert n.cancelled
print('stream terminated')
await portal.cancel_actor()
if __name__ == '__main__': if __name__ == '__main__':