From 3f03290e4fc2878d7c7e07a0d5138be23518dc69 Mon Sep 17 00:00:00 2001 From: chakcy <947105045@qq.com> Date: Tue, 12 Aug 2025 00:41:18 +0800 Subject: [PATCH] =?UTF-8?q?modify:=20=E7=9B=AE=E5=BD=95=E7=BB=93=E6=9E=84?= =?UTF-8?q?=E6=95=B4=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/queue_sqlite/cycle/listen_cycle.py | 0 src/queue_sqlite/mounter/__init__.py | 3 +- src/queue_sqlite/mounter/base_mounter.py | 42 ------------------- .../queue_operation/listen_operation.py | 1 - src/queue_sqlite/scheduler/task_scheduler.py | 2 +- 5 files changed, 3 insertions(+), 45 deletions(-) delete mode 100644 src/queue_sqlite/cycle/listen_cycle.py delete mode 100644 src/queue_sqlite/mounter/base_mounter.py 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) # 并行执行任务