archive/blender-cloud-addon
Archived
1
3
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Removed time dependency in asyncio loop kicking.

Browse Source
This commit is contained in:
Sybren A. Stüvel 2016-03-22 13:59:50 +01:00
parent 6b84dcf282
commit c5f66ec0d9
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
blender_cloud/async_loop.py
View File

@ -27,20 +27,28 @@ def setup_asyncio_executor():
def kick_async_loop(*args): def kick_async_loop(*args):
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
# We always need to do one more 'kick' to handle task-done callbacks.
stop_after_this_kick = False
if loop.is_closed(): if loop.is_closed():
log.warning('loop closed, stopping') log.warning('loop closed, stopping immediately.')
stop_async_loop() stop_async_loop()
return return
all_tasks = asyncio.Task.all_tasks() all_tasks = asyncio.Task.all_tasks()
if not len(all_tasks): if not len(all_tasks):
log.debug('no more scheduled tasks, stopping') log.debug('no more scheduled tasks, stopping after this kick.')
stop_async_loop() stop_after_this_kick = True
return
elif all(task.done() for task in all_tasks):
log.debug('all %i tasks are done, fetching results and stopping after this kick.',
len(all_tasks))
stop_after_this_kick = True
if all(task.done() for task in all_tasks):
log.info('all %i tasks are done, fetching results and stopping.', len(all_tasks))
for task_idx, task in enumerate(all_tasks): for task_idx, task in enumerate(all_tasks):
if not task.done():
continue
# noinspection PyBroadException # noinspection PyBroadException
try: try:
res = task.result() res = task.result()
@ -51,16 +59,12 @@ def kick_async_loop(*args):
except Exception: except Exception:
print('{}: resulted in exception'.format(task)) print('{}: resulted in exception'.format(task))
traceback.print_exc() traceback.print_exc()
loop.stop()
loop.run_forever()
if stop_after_this_kick:
stop_async_loop() stop_async_loop()
return
# Perform a single async loop step
def stop_loop(future):
future.set_result('done')
future = asyncio.Future()
loop.call_later(0.005, stop_loop, future)
loop.run_until_complete(future)
def async_loop_handler() -> callable: def async_loop_handler() -> callable: