diff --git a/src/queue_sqlite/cycle/listen_cycle.py b/src/queue_sqlite/cycle/listen_cycle.py deleted file mode 100644 index e69de29..0000000 diff --git a/src/queue_sqlite/mounter/__init__.py b/src/queue_sqlite/mounter/__init__.py index 36b815d..04e5b36 100644 --- a/src/queue_sqlite/mounter/__init__.py +++ b/src/queue_sqlite/mounter/__init__.py @@ -1,3 +1,4 @@ from .task_mounter import TaskMounter +from .listen_mounter import ListenMounter -__all__ = ["TaskMounter"] \ No newline at end of file +__all__ = ["TaskMounter", "ListenMounter"] \ No newline at end of file diff --git a/src/queue_sqlite/mounter/base_mounter.py b/src/queue_sqlite/mounter/base_mounter.py deleted file mode 100644 index 803ad53..0000000 --- a/src/queue_sqlite/mounter/base_mounter.py +++ /dev/null @@ -1,42 +0,0 @@ -from typing import Callable, List - - -class BaseMounter: - @classmethod - def mount_(cls, function: Callable): - setattr(cls, function.__name__, function) - - @staticmethod - def mount(meta: dict = {}): - """带参数的装饰器""" - def decorator(function: Callable): - # 使用自定义名称或函数原名 - function.meta = meta # type: ignore - setattr(BaseMounter, function.__name__, function) - return function - return decorator - - @classmethod - def get_function(cls, name: str): - return getattr(cls, name, None) - - @classmethod - def get_task_list(cls) -> List[str]: - """获取所有挂载的任务函数名称列表""" - task_list = [] - - # 遍历类属性字典 - for attr_name, attr_value in vars(cls).items(): - # 过滤条件: - # 1. 必须是可调用对象(函数) - # 2. 不是类自带的特殊方法(非双下划线开头) - # 3. 不是类方法本身(如 mount_, get_task_list 等) - if ( - callable(attr_value) - and not attr_name.startswith("__") - and attr_name not in ["mount_", "mount", "get_function", "get_task_list"] - ): - task_list.append(attr_name) - - return task_list - \ No newline at end of file diff --git a/src/queue_sqlite/queue_operation/listen_operation.py b/src/queue_sqlite/queue_operation/listen_operation.py index 668d85f..984278c 100644 --- a/src/queue_sqlite/queue_operation/listen_operation.py +++ b/src/queue_sqlite/queue_operation/listen_operation.py @@ -1,6 +1,5 @@ from ..mounter.listen_mounter import ListenMounter import sqlite3 -import threading class ListenOperation: diff --git a/src/queue_sqlite/scheduler/task_scheduler.py b/src/queue_sqlite/scheduler/task_scheduler.py index d638b34..03d7fc4 100644 --- a/src/queue_sqlite/scheduler/task_scheduler.py +++ b/src/queue_sqlite/scheduler/task_scheduler.py @@ -10,7 +10,7 @@ import threading class TaskScheduler: - def __init__(self, queue_operation: QueueOperation, task_thread_num: int = 1): + def __init__(self, queue_operation: QueueOperation, task_thread_num: int = 2): self.task_thread_num = task_thread_num self.is_running = False self.executor = ThreadPoolExecutor(max_workers=task_thread_num) # 并行执行任务