ABC学习网

Python使用signal定时结束AsyncIOScheduler任务的问题

热度:9℃发布时间:2023-12-29 12:11:07

在使用aiohttp结合apscheduler的AsyncIOScheduler模拟定点并发的时候遇到两个问题

在调度器scheduler.start()后,程序直接退出(在Jupiter中任务可以正常启动) 如何在指定时间调用scheduler.shutdown()? (因为程序直接退出了)

原调试代码如下:

from datetime import datetime, timedeltaimport aiohttpfrom apscheduler.schedulers.asyncio import AsyncIOSchedulerfrom apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStoreasync def get(session): url = ’https://httpbin.org/get?a=1’ async with session.get(url) as res: print(’get’, res.status) return await res.text()async def post(session): url = ’https://httpbin.org/post?b=2’ async with session.post(url) as res: print(’post’, res.status) return await res.text()async def main(): async with aiohttp.ClientSession() as session: await get(session) await post(session)if __name__ == ’__main__’: jobstores = {’default’: SQLAlchemyJobStore(url=’sqlite:///jobs.sqlite’)} scheduler = AsyncIOScheduler(jobstores=jobstores) for i in range(10): # 添加10个任务 job = scheduler.add_job(main, ’date’, run_date=datetime.now() + timedelta(seconds=10)) scheduler.start()

Google后发现AsyncIOScheduler的使用需要在scheduler启动后,需要自己调用asyncio.get_event_loop().run_forever()来启动协程任务。
但是一旦run_forever()则就会阻塞至死。除非有KeyboardInterrupt, SystemExit等异常或者强杀来停止其运行。
此时想到使用Python的signal来定时发送信号,修改后程序如下,可以正常延迟停止(感觉有点像模拟Go的defer)。

# -*- coding: utf-8 -*-"""@Time : 2021/7/23@Auth : hanzhichao@Desc:"""from datetime import datetime, timedeltaimport signalimport asyncioimport aiohttpfrom apscheduler.schedulers.asyncio import AsyncIOSchedulerfrom apscheduler.jobstores.sqlalchemy import SQLAlchemyJobStoreasync def get(session): url = ’https://httpbin.org/get?a=1’ async with session.get(url) as res: print(’get’, res.status) return await res.text()async def post(session): url = ’https://httpbin.org/post?b=2’ async with session.post(url) as res: print(’post’, res.status) return await res.text()async def main(): async with aiohttp.ClientSession() as session: await get(session) await post(session)if __name__ == ’__main__’: jobstores = {’default’: SQLAlchemyJobStore(url=’sqlite:///jobs.sqlite’)} scheduler = AsyncIOScheduler(jobstores=jobstores) for i in range(10): # 添加10个任务 job = scheduler.add_job(main, ’date’, run_date=datetime.now() + timedelta(seconds=10)) scheduler.start() signal.alarm(20) # 20秒后终止程序 asyncio.get_event_loop().run_forever() # 永远运行

到此这篇关于Python使用signal定时结束AsyncIOScheduler任务的文章就介绍到这了,更多相关Python定时结束AsyncIOScheduler任务内容请搜索ABC学习网以前的文章或继续浏览下面的相关文章希望大家以后多多支持ABC学习网!

(window.slotbydup = window.slotbydup || []).push({id: "u6915441",container: "_5rmj5io5v3i",async: true});
网友评论
评论
发 布

更多编程教程
  • 编程教程推荐
更多+