Python 异步任务如何检测运行时间

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

这是一个创建于 1403 天前的主题，其中的信息可能已经有所发展或是发生改变。

    async def run_task(self, task_name, task_fun):
        self.logger.info(f'{task_name}')
        begin_time = int(time.time())
        self.task_state = False
        # t = threading.Thread(target=self.check_task, args=(begin_time, task_name))
        # t.start()
        # t.join()

		
        try:
            await getattr(self, task_fun)()
            await asyncio.sleep(1)
            self.logger.warning(f'{task_name} -> DONE.')
            self.task_state = True
        except Exception as e:
            self.logger.error(e)
            self.task_state = True
        finally:
            await self.close_page()
            await asyncio.sleep(1)

    def check_task(self, begin_time, task_name):
        while True:
            if self.task_state:
                return True

            now_time = int(time.time())
            n = now_time - begin_time
            if n > 120:
                self.logger.warning(f'{task_name} runtime {n}s.')
                return False
            time.sleep(1)

想要实现的是 task_fun 执行时间超过 N 秒直接退出 run_task
自己想了笨方法，写了个线程检测运行时间，但不知道怎么抛异常退出 task_fun
有没有更好的实现？

3 条回复 • 2021-01-21 14:47:44 +08:00

ysc3839

2021-01-20 17:29:55 +08:00 via Android

我不了解 Python 的 async，如果是 js 的话可以用 Promise.race，然后取消 task 运行。

Vegetable

2021-01-20 17:31:32 +08:00

想得倒是挺多，不过你没感觉这是一个很常见的需求吗？

https://docs.python.org/zh-cn/3/library/asyncio-task.html#timeouts

async def main():
# Wait for at most 1 second
try:
await asyncio.wait_for(eternity(), timeout=1.0)
except asyncio.TimeoutError:
print('timeout!')

asyncio.run(main())

linw1995

2021-01-21 14:47:44 +08:00

如果不是同步堵塞住的话，用 `asyncio.wait_for` 就好了。同步堵塞的话，建议修改这个函数，通过 `loop.run_in_executor` 用副线程去跑，这样就不会堵塞住了。同时在外部用 `asyncio.wait_for`，且通过在 coro 内部去 try-except 这个超时而抛出的 `asyncio.CancelError` ，再通过 theading.Event 等方式，把取消事件传递给副线程的函数