This commit is contained in:
@ -148,6 +148,7 @@ class LongTask(BaseModel):
|
|||||||
datafile.to_handle[self.callback] = [
|
datafile.to_handle[self.callback] = [
|
||||||
t for t in datafile.to_handle[self.callback] if t.uuid != self.uuid
|
t for t in datafile.to_handle[self.callback] if t.uuid != self.uuid
|
||||||
]
|
]
|
||||||
|
logger.info(f"LongTask 工作完成 UUID={self.uuid} callback={self.callback}")
|
||||||
|
|
||||||
def clean(self):
|
def clean(self):
|
||||||
self._aio_task = None
|
self._aio_task = None
|
||||||
@ -257,6 +258,7 @@ async def create_longtask(
|
|||||||
deadline=deadline,
|
deadline=deadline,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
logger.info(f"创建了新的 LongTask UUID={task.uuid} CALLBACK={task.callback}")
|
||||||
await task.start()
|
await task.start()
|
||||||
|
|
||||||
async with longtask_data() as d:
|
async with longtask_data() as d:
|
||||||
@ -266,10 +268,17 @@ async def create_longtask(
|
|||||||
|
|
||||||
|
|
||||||
async def init_longtask():
|
async def init_longtask():
|
||||||
|
counter = 0
|
||||||
|
req: set[str] = set()
|
||||||
|
|
||||||
async with longtask_data() as data:
|
async with longtask_data() as data:
|
||||||
for v in data.to_handle.values():
|
for v in data.to_handle.values():
|
||||||
for t in v:
|
for t in v:
|
||||||
await t.start()
|
await t.start()
|
||||||
|
counter += 1
|
||||||
|
req.add(t.callback)
|
||||||
|
|
||||||
|
logger.info(f"LongTask 启动了任务 数量={counter} 期望的门类=[{','.join(req)}]")
|
||||||
|
|
||||||
|
|
||||||
DepLongTaskTarget = Annotated[LongTaskTarget, Depends(get_long_task_target)]
|
DepLongTaskTarget = Annotated[LongTaskTarget, Depends(get_long_task_target)]
|
||||||
|
|||||||
@ -4,13 +4,11 @@ import asyncio
|
|||||||
from loguru import logger
|
from loguru import logger
|
||||||
import nonebot
|
import nonebot
|
||||||
|
|
||||||
# from nonebot.adapters import Bot, Event
|
|
||||||
# from nonebot_plugin_alconna import UniMessage
|
# from nonebot_plugin_alconna import UniMessage
|
||||||
from konabot.common.longtask import (
|
from konabot.common.longtask import (
|
||||||
# DepLongTaskTarget,
|
# DepLongTaskTarget,
|
||||||
# LongTask,
|
# LongTask,
|
||||||
# create_longtask,
|
# create_longtask,
|
||||||
# get_long_task_target,
|
|
||||||
# handle_long_task,
|
# handle_long_task,
|
||||||
init_longtask,
|
init_longtask,
|
||||||
)
|
)
|
||||||
@ -45,6 +43,5 @@ async def _():
|
|||||||
# handler="test_callback_001",
|
# handler="test_callback_001",
|
||||||
# data={},
|
# data={},
|
||||||
# target=target,
|
# 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)
|
|
||||||
|
|||||||
Reference in New Issue
Block a user