假设我有几个协程在循环中运行。如何使得如果其中一些失败而异常整个程序会因此异常而失败?因为现在asyncio甚至不打印来自协同程序的错误消息,除非我使用日志级别“DEBUG”。
from asyncio import get_event_loop, sleep
async def c(sleep_time=2, fail=False):
print('c', sleep_time, fail)
if fail:
raise Exception('fail')
while True:
print('doing stuff')
await sleep(sleep_time)
loop = get_event_loop()
loop.create_task(c(sleep_time=10, fail=False))
loop.create_task(c(fail=True))
loop.run_forever()
答案 0 :(得分:8)
优雅的方法是使用错误处理API。
https://docs.python.org/3/library/asyncio-eventloop.html#error-handling-api
示例:
import asyncio
async def run_division(a, b):
await asyncio.sleep(2)
return a / b
def custom_exception_handler(loop, context):
# first, handle with default handler
loop.default_exception_handler(context)
exception = context.get('exception')
if isinstance(exception, ZeroDivisionError):
print(context)
loop.stop()
loop = asyncio.get_event_loop()
# Set custom handler
loop.set_exception_handler(custom_exception_handler)
loop.create_task(run_division(1, 0))
loop.run_forever()
答案 1 :(得分:4)
以下是您可能想要用来制作解决方案的一些注意事项:
检索couroutine异常(或结果!)的最简单方法是await
。 asyncio.gather()
将从协同程序创建任务,并将所有这些任务包装在一个包含任务中,如果其中一个子任务失败,该任务将失败:
import asyncio
import random
async def coro(n):
print("Start", n)
await asyncio.sleep(random.uniform(0.2, 0.5))
if n % 4 == 0:
raise Exception('fail ({})'.format(n))
return "OK: {}".format(n)
async def main():
tasks = [coro(i) for i in range(10)]
await asyncio.gather(*tasks)
print("done")
loop = asyncio.get_event_loop()
try:
asyncio.ensure_future(main())
loop.run_forever()
finally:
loop.close()
然而,这不会关闭循环。要停止正在运行的循环,请使用loop.stop()
。请改用:
async def main():
tasks = [coro(i) for i in range(10)]
try:
await asyncio.gather(*tasks)
except Exception as e:
loop.stop()
raise
print("done")
当一些长时间运行的协同程序正在运行时停止循环可能不是你想要的。您可能希望首先通过事件发出一些信号来关闭:
import asyncio
import random
async def repeat(n):
print("start", n)
while not shutting_down.is_set():
print("repeat", n)
await asyncio.sleep(random.uniform(1, 3))
print("done", n)
async def main():
print("waiting 6 seconds..")
await asyncio.sleep(6)
print("shutting down")
shutting_down.set() # not a coroutine!
print("waiting")
await asyncio.wait(long_running)
print("done")
loop.stop()
loop = asyncio.get_event_loop()
shutting_down = asyncio.Event(loop=loop)
long_running = [loop.create_task(repeat(i + 1)) for i in range(5)]
try:
asyncio.ensure_future(main())
loop.run_forever()
finally:
loop.close()
如果您不想await
执行任务,可能需要使用asyncio.Event
(或asyncio.Queue
)来指示全局错误处理程序停止循环:< / p>
import asyncio
async def fail():
try:
print("doing stuff...")
await asyncio.sleep(0.2)
print("doing stuff...")
await asyncio.sleep(0.2)
print("doing stuff...")
raise Exception('fail')
except Exception as e:
error_event.payload = e
error_event.set()
raise # optional
async def error_handler():
await error_event.wait()
e = error_event.payload
print("Got:", e)
raise e
loop = asyncio.get_event_loop()
error_event = asyncio.Event()
try:
loop.create_task(fail())
loop.run_until_complete(error_handler())
finally:
loop.close()
(为简单起见,此处与run_until_complete()
一起使用,但也可与loop.stop()
一起使用)
答案 2 :(得分:1)
好的,我找到了不需要重写任何现有代码的解决方案。它可能看起来很丑陋,但我想我喜欢它。
因为我已经像这样抓住了KeyboardInterrupt
。
def main(task=None):
task = task or start()
loop = get_event_loop()
try:
task = loop.create_task(task)
future = ensure_future(task)
loop.run_until_complete(task)
except KeyboardInterrupt:
print('\nperforming cleanup...')
task.cancel()
loop.run_until_complete(future)
loop.close()
sys.exit()
如何从协程发送KeyboardInterrupt
给自己?我认为这会挂起应用程序,因为os.kill
会等待应用程序关闭,因为它等待的应用程序是同一个应用程序,它会造成一种死锁,但谢天谢地我错了。此代码实际上可以正常工作并在退出之前打印clean up
。
async def c(sleep_time=2, fail=False):
print('c', sleep_time, fail)
if fail:
await sleep(sleep_time)
os.kill(os.getpid(), signal.SIGINT)
while True:
print('doing stuff')
await sleep(sleep_time)
loop = get_event_loop()
loop.create_task(c(sleep_time=10, fail=False))
loop.create_task(c(fail=True))
try:
loop.run_forever()
except KeyboardInterrupt:
print('clean up')
loop.close()
sys.exit()