modify: 调整项目结构，更新调度器

tests/test_stress.py

@@ -49,12 +49,14 @@ class TestStress:
         for i in range(TASK_NUM):  # 增加任务数量
             message_item = MessageItem(content={"num": i}, destination="example")
             messages.append(message_item)
-        
+
         cls.is_monitoring = True
         monitor_thread = threading.Thread(target=cls.monitor_resources, daemon=True)
         monitor_thread.start()
-        queue_scheduler = QueueScheduler(receive_thread_num=4, task_thread_num=8, shard_num=12)
-        
+        queue_scheduler = QueueScheduler(
+            receive_thread_num=4, task_thread_num=8, shard_num=12
+        )
+
         # threads = []
         # for message_item in messages:
         #     t = threading.Thread(target=queue_scheduler.send_message, args=(message_item, cls.test_callback))
@@ -62,26 +64,29 @@ class TestStress:
 
         # 添加性能计时
         start_time = time.perf_counter()
-        
-        queue_scheduler.start_queue_scheduler()
-        
+
+        queue_scheduler.start()
+
         for message_item in messages:  # 增加任务数量
             message_item: MessageItem
             queue_scheduler.send_message(message_item, cls._callback)
-        
+
         # # 多线程发送任务
         # for t in threads:
         #     t.start()
-        
+
         # # 动态等待任务完成
         start_wait = time.time()
-        while (queue_scheduler.queue_operation.get_queue_length() > 0 or cls.callback_counter < TASK_NUM):  # 30秒超时
+        while (
+            queue_scheduler.queue_operation.get_queue_length() > 0
+            or cls.callback_counter < TASK_NUM
+        ):  # 30秒超时
             time.sleep(0.5)
             print(f"当前队列长度：{queue_scheduler.queue_operation.get_queue_length()}")
             print(f"当前回调数量：{cls.callback_counter}")
-        
-        queue_scheduler.stop_queue_scheduler()
-        
+
+        queue_scheduler.stop()
+
         print(f"当前队列长度：{queue_scheduler.queue_operation.get_queue_length()}")
         print(f"测试函数：斐波那契数列")
         print(f"次数：{TASK_NUM}")
@@ -89,4 +94,3 @@ class TestStress:
 
         cls.is_monitoring = False
         monitor_thread.join(timeout=1)
-