提升 LongTask 的健壮性

konabot/common/longtask.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from contextlib import asynccontextmanager
 import datetime
 import json
@@ -89,6 +91,7 @@ class LongTaskTarget(BaseModel):
                 )
                 return False
             await bot.send_message(self.channel_id, cast(Any, await msg.export()))
+            return True
         if self.platform == "discord":
             if not isinstance(bot, DCBot):
                 logger.warning(
@@ -104,6 +107,7 @@ class LongTaskTarget(BaseModel):
                 ),
                 tts=False,
             )
+            return True
         logger.warning(f"没有一个平台是期望的平台 PLATFORM={self.platform}")
         return False
 
@@ -111,7 +115,7 @@ class LongTaskTarget(BaseModel):
 class LongTask(BaseModel):
     uuid: str
     data_json: str
-    target: "LongTaskTarget"
+    target: LongTaskTarget
     callback: str
     deadline: datetime.datetime
     canceled: bool = False
@@ -143,11 +147,25 @@ class LongTask(BaseModel):
                 datafile.unhandled.setdefault(self.callback, []).append(self)
 
             return
-        await hdl(self)
+        success = False
+        try:
+            await hdl(self)
+            success = True
+        except Exception as e:
+            logger.exception(e)
         async with longtask_data() as datafile:
             datafile.to_handle[self.callback] = [
                 t for t in datafile.to_handle[self.callback] if t.uuid != self.uuid
             ]
+            if not success:
+                datafile.unhandled.setdefault(self.callback, []).append(self)
+                logger.info(
+                    f"LongTask 执行失败 UUID={self.uuid} callback={self.callback}"
+                )
+            else:
+                logger.info(
+                    f"LongTask 工作完成 UUID={self.uuid} callback={self.callback}"
+                )
 
     def clean(self):
         self._aio_task = None
@@ -257,6 +275,7 @@ async def create_longtask(
         deadline=deadline,
     )
 
+    logger.info(f"创建了新的 LongTask UUID={task.uuid} CALLBACK={task.callback}")
     await task.start()
 
     async with longtask_data() as d:
@@ -266,10 +285,17 @@ async def create_longtask(
 
 
 async def init_longtask():
+    counter = 0
+    req: set[str] = set()
+
     async with longtask_data() as data:
         for v in data.to_handle.values():
             for t in v:
                 await t.start()
+                counter += 1
+                req.add(t.callback)
+
+        logger.info(f"LongTask 启动了任务 数量={counter} 期望的门类=[{','.join(req)}]")
 
 
 DepLongTaskTarget = Annotated[LongTaskTarget, Depends(get_long_task_target)]

konabot/plugins/longtask_core.py

@@ -4,13 +4,11 @@ import asyncio
 from loguru import logger
 import nonebot
 
-# from nonebot.adapters import Bot, Event
 # from nonebot_plugin_alconna import UniMessage
 from konabot.common.longtask import (
     # DepLongTaskTarget,
     # LongTask,
     # create_longtask,
-    # get_long_task_target,
     # handle_long_task,
     init_longtask,
 )
@@ -45,6 +43,5 @@ async def _():
 #         handler="test_callback_001",
 #         data={},
 #         target=target,
-#         deadline=datetime.datetime.now() + datetime.timedelta(seconds=2),
+#         deadline=datetime.datetime.now() + datetime.timedelta(seconds=20),
 #     )
-# await target.send_message(UniMessage().text("Hello, world!"), at=True)