整理代码

sync_state_machine/datasource/api/datasource.py

@@ -65,22 +65,10 @@ class ApiDataSource(BaseDataSource):
         """关闭数据源（关闭 HTTP 客户端）"""
         await self.client.close()
     
-    def add_handler(self, handler: NodeHandler) -> None:
-        """
-        添加业务 Handler
-        
-        Args:
-            handler: API Handler 实例
-        """
+    def _prepare_handler_for_registration(self, handler: NodeHandler) -> None:
         handler.set_api_client(self.client)
         handler.set_poll_mode(self.poll_mode)
-        
-        # 调用基类添加方法
-        super().add_handler(handler)
-
-        # 如果 collection 已设置，确保新 handler 拿到引用
-        if self._collection is not None:
-            handler.set_collection(self._collection)  # type: ignore[arg-type]
+        super()._prepare_handler_for_registration(handler)
 
     @staticmethod
     def _fmt_trace_value(value, max_len: int = 600) -> str: