添加进程及生命周期管理器,添加轻雪框架支持

This commit is contained in:
远野千束 2024-07-24 02:54:59 +08:00
parent c137f2f916
commit 041a219151

View File

@ -105,6 +105,34 @@ $$$$$$$$/ $$$$$$/ $$/ $$$$$$$$/ $$/ $$$$$$/ $$/ $$/ $$$$$$/
else: else:
should_exit = True should_exit = True
@staticmethod
def _run_coroutine(*coro: Coroutine):
"""
运行协程
Args:
coro:
Returns:
"""
# 检测是否有现有的事件循环
new_loop = False
try:
loop = asyncio.get_event_loop()
except RuntimeError:
new_loop = True
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
if new_loop:
for c in coro:
loop.run_until_complete(c)
loop.close()
else:
for c in coro:
loop.create_task(c)
@property @property
def status(self) -> int: def status(self) -> int:
""" """
@ -123,9 +151,9 @@ $$$$$$$$/ $$$$$$/ $$/ $$$$$$$$/ $$/ $$$$$$/ $$/ $$/ $$$$$$/
logger.info("Stopping LiteyukiBot...") logger.info("Stopping LiteyukiBot...")
logger.debug("Running before_restart functions...") logger.debug("Running before_restart functions...")
asyncio.run(self.lifespan.before_restart()) self._run_coroutine(self.lifespan.before_restart())
logger.debug("Running before_shutdown functions...") logger.debug("Running before_shutdown functions...")
asyncio.run(self.lifespan.before_shutdown()) self._run_coroutine(self.lifespan.before_shutdown())
ProcessingManager.restart() ProcessingManager.restart()
self.running = False self.running = False