70 lines
1.5 KiB
Python
70 lines
1.5 KiB
Python
|
import trio
|
||
|
|
import tractor
|
||
|
|
|
||
|
|
|
||
|
|
def sync_pause():
|
||
|
|
tractor.pause_from_sync()
|
||
|
|
|
||
|
|
|
||
|
|
@tractor.context
|
||
|
|
async def start_n_sync_pause(
|
||
|
|
ctx: tractor.Context,
|
||
|
|
):
|
||
|
|
# sync to requesting peer
|
||
|
|
await ctx.started()
|
||
|
|
|
||
|
|
actor: tractor.Actor = tractor.current_actor()
|
||
|
|
print(f'entering SYNC PAUSE in {actor.uid}')
|
||
|
|
sync_pause()
|
||
|
|
print(f'back from SYNC PAUSE in {actor.uid}')
|
||
|
|
|
||
|
|
|
||
|
|
async def main() -> None:
|
||
|
|
|
||
|
|
from tractor._rpc import maybe_import_gb
|
||
|
|
|
||
|
|
async with tractor.open_nursery(
|
||
|
|
debug_mode=True,
|
||
|
|
) as an:
|
||
|
|
|
||
|
|
# TODO: where to put this?
|
||
|
|
# => just inside `open_root_actor()` yah?
|
||
|
|
await maybe_import_gb()
|
||
|
|
|
||
|
|
p: tractor.Portal = await an.start_actor(
|
||
|
|
'subactor',
|
||
|
|
enable_modules=[__name__],
|
||
|
|
# infect_asyncio=True,
|
||
|
|
debug_mode=True,
|
||
|
|
loglevel='cancel',
|
||
|
|
)
|
||
|
|
|
||
|
|
# TODO: 3 sub-actor usage cases:
|
||
|
|
# -[ ] via a `.run_in_actor()` call
|
||
|
|
# -[ ] via a `.run()`
|
||
|
|
# -[ ] via a `.open_context()`
|
||
|
|
#
|
||
|
|
async with p.open_context(
|
||
|
|
start_n_sync_pause,
|
||
|
|
) as (ctx, first):
|
||
|
|
assert first is None
|
||
|
|
|
||
|
|
await tractor.pause()
|
||
|
|
sync_pause()
|
||
|
|
|
||
|
|
# TODO: make this work!!
|
||
|
|
await trio.to_thread.run_sync(
|
||
|
|
sync_pause,
|
||
|
|
abandon_on_cancel=False,
|
||
|
|
)
|
||
|
|
|
||
|
|
await ctx.cancel()
|
||
|
|
|
||
|
|
# TODO: case where we cancel from trio-side while asyncio task
|
||
|
|
# has debugger lock?
|
||
|
|
await p.cancel_actor()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
trio.run(main)
|