Resource update & asyncio fix

correct bioyond config

prcxi example

fix append_resource

fix regularcontainer

fix cancel error

fix resource_get param

fix json dumps

support name change during materials change

enable slave mode

change uuid logger to trace level

correct remove_resource stats

disable slave connect websocket

adjust with_children param

modify devices to use correct executor (sleep, create_task)

support sleep and create_task in node

fix run async execution error

unilabos/ros/nodes/base_device_node.py

@@ -53,7 +53,7 @@ from unilabos.ros.nodes.resource_tracker import (
 )
 from unilabos.ros.x.rclpyx import get_event_loop
 from unilabos.ros.utils.driver_creator import WorkstationNodeCreator, PyLabRobotCreator, DeviceClassCreator
-from unilabos.utils.async_util import run_async_func
+from rclpy.task import Task, Future
 from unilabos.utils.import_manager import default_manager
 from unilabos.utils.log import info, debug, warning, error, critical, logger, trace
 from unilabos.utils.type_check import get_type_class, TypeEncoder, get_result_info_str
@@ -466,8 +466,9 @@ class BaseROS2DeviceNode(Node, Generic[T]):
                 contain_model = not isinstance(resource, Deck)
                 if isinstance(resource, ResourcePLR):
                     # resources.list()
-                    resources_tree = dict_to_tree(copy.deepcopy({r["id"]: r for r in resources}))
-                    plr_instance = resource_ulab_to_plr(resources_tree[0], contain_model)
+                    plr_instance = ResourceTreeSet.from_raw_list(resources).to_plr_resources()[0]
+                    # resources_tree = dict_to_tree(copy.deepcopy({r["id"]: r for r in resources}))
+                    # plr_instance = resource_ulab_to_plr(resources_tree[0], contain_model)
 
                     if isinstance(plr_instance, Plate):
                         empty_liquid_info_in = [(None, 0)] * plr_instance.num_items
@@ -555,6 +556,15 @@ class BaseROS2DeviceNode(Node, Generic[T]):
         rclpy.get_global_executor().add_node(self)
         self.lab_logger().debug(f"ROS节点初始化完成")
 
+    async def sleep(self, rel_time: float, callback_group=None):
+        if callback_group is None:
+            callback_group = self.callback_group
+        await ROS2DeviceNode.async_wait_for(self, rel_time, callback_group)
+
+    @classmethod
+    async def create_task(cls, func, trace_error=True, **kwargs) -> Task:
+        return ROS2DeviceNode.run_async_func(func, trace_error, **kwargs)
+
     async def update_resource(self, resources: List["ResourcePLR"]):
         r = SerialCommand.Request()
         tree_set = ResourceTreeSet.from_plr_resources(resources)
@@ -629,6 +639,145 @@ class BaseROS2DeviceNode(Node, Generic[T]):
         - remove: 从资源树中移除资源
         """
         from pylabrobot.resources.resource import Resource as ResourcePLR
+
+        def _handle_add(
+            plr_resources: List[ResourcePLR], tree_set: ResourceTreeSet, additional_add_params: Dict[str, Any]
+        ) -> Dict[str, Any]:
+            """
+            处理资源添加操作的内部函数
+
+            Args:
+                plr_resources: PLR资源列表
+                tree_set: 资源树集合
+                additional_add_params: 额外的添加参数
+
+            Returns:
+                操作结果字典
+            """
+            for plr_resource, tree in zip(plr_resources, tree_set.trees):
+                self.resource_tracker.add_resource(plr_resource)
+                self.transfer_to_new_resource(plr_resource, tree, additional_add_params)
+
+            func = getattr(self.driver_instance, "resource_tree_add", None)
+            if callable(func):
+                func(plr_resources)
+
+            return {"success": True, "action": "add"}
+
+        def _handle_remove(resources_uuid: List[str]) -> Dict[str, Any]:
+            """
+            处理资源移除操作的内部函数
+
+            Args:
+                resources_uuid: 要移除的资源UUID列表
+
+            Returns:
+                操作结果字典，包含移除的资源列表
+            """
+            found_resources: List[List[Union[ResourcePLR, dict]]] = self.resource_tracker.figure_resource(
+                [{"uuid": uid} for uid in resources_uuid], try_mode=True
+            )
+            found_plr_resources = []
+            other_plr_resources = []
+
+            for found_resource in found_resources:
+                for resource in found_resource:
+                    if issubclass(resource.__class__, ResourcePLR):
+                        found_plr_resources.append(resource)
+                    else:
+                        other_plr_resources.append(resource)
+
+            # 调用driver的remove回调
+            func = getattr(self.driver_instance, "resource_tree_remove", None)
+            if callable(func):
+                func(found_plr_resources)
+
+            # 从parent卸载并从tracker移除
+            for plr_resource in found_plr_resources:
+                if plr_resource.parent is not None:
+                    plr_resource.parent.unassign_child_resource(plr_resource)
+                self.resource_tracker.remove_resource(plr_resource)
+                self.lab_logger().info(f"移除物料 {plr_resource} 及其子节点")
+
+            for other_plr_resource in other_plr_resources:
+                self.resource_tracker.remove_resource(other_plr_resource)
+                self.lab_logger().info(f"移除物料 {other_plr_resource} 及其子节点")
+
+            return {
+                "success": True,
+                "action": "remove",
+                # "removed_plr": found_plr_resources,
+                # "removed_other": other_plr_resources,
+            }
+
+        def _handle_update(
+            plr_resources: List[ResourcePLR], tree_set: ResourceTreeSet, additional_add_params: Dict[str, Any]
+        ) -> Dict[str, Any]:
+            """
+            处理资源更新操作的内部函数
+
+            Args:
+                plr_resources: PLR资源列表（包含新状态）
+                tree_set: 资源树集合
+                additional_add_params: 额外的参数
+
+            Returns:
+                操作结果字典
+            """
+            for plr_resource, tree in zip(plr_resources, tree_set.trees):
+                states = plr_resource.serialize_all_state()
+                original_instance: ResourcePLR = self.resource_tracker.figure_resource(
+                    {"uuid": tree.root_node.res_content.uuid}, try_mode=False
+                )
+
+                # Update操作中包含改名：需要先remove再add
+                if original_instance.name != plr_resource.name:
+                    old_name = original_instance.name
+                    new_name = plr_resource.name
+                    self.lab_logger().info(f"物料改名操作：{old_name} -> {new_name}")
+
+                    # 收集所有相关的uuid（包括子节点）
+                    _handle_remove([original_instance.unilabos_uuid])
+                    original_instance.name = new_name
+                    _handle_add([original_instance], tree_set, additional_add_params)
+
+                    self.lab_logger().info(f"物料改名完成：{old_name} -> {new_name}")
+
+                # 常规更新：不涉及改名
+                original_parent_resource = original_instance.parent
+                original_parent_resource_uuid = getattr(original_parent_resource, "unilabos_uuid", None)
+                target_parent_resource_uuid = tree.root_node.res_content.uuid_parent
+
+                self.lab_logger().info(
+                    f"物料{original_instance} 原始父节点{original_parent_resource_uuid} "
+                    f"目标父节点{target_parent_resource_uuid} 更新"
+                )
+
+                # 更新extra
+                if getattr(plr_resource, "unilabos_extra", None) is not None:
+                    original_instance.unilabos_extra = getattr(plr_resource, "unilabos_extra")  # type: ignore  # noqa: E501
+
+                # 如果父节点变化，需要重新挂载
+                if (
+                    original_parent_resource_uuid != target_parent_resource_uuid
+                    and original_parent_resource is not None
+                ):
+                    self.transfer_to_new_resource(original_instance, tree, additional_add_params)
+
+                # 加载状态
+                original_instance.load_all_state(states)
+                child_count = len(original_instance.get_all_children())
+                self.lab_logger().info(
+                    f"更新了资源属性 {plr_resource}[{tree.root_node.res_content.uuid}] " f"及其子节点 {child_count} 个"
+                )
+
+            # 调用driver的update回调
+            func = getattr(self.driver_instance, "resource_tree_update", None)
+            if callable(func):
+                func(plr_resources)
+
+            return {"success": True, "action": "update"}
+
         try:
             data = json.loads(req.command)
             results = []
@@ -647,7 +796,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
                     ].call_async(
                         SerialCommand.Request(
                             command=json.dumps(
-                                {"data": {"data": resources_uuid, "with_children": False}, "action": "get"}
+                                {"data": {"data": resources_uuid, "with_children": True if action == "add" else False}, "action": "get"}
                             )
                         )
                     )  # type: ignore
@@ -655,68 +804,20 @@ class BaseROS2DeviceNode(Node, Generic[T]):
                     tree_set = ResourceTreeSet.from_raw_list(raw_nodes)
                 try:
                     if action == "add":
-                        # 添加资源到资源跟踪器
+                        if tree_set is None:
+                            raise ValueError("tree_set不能为None")
                         plr_resources = tree_set.to_plr_resources()
-                        for plr_resource, tree in zip(plr_resources, tree_set.trees):
-                            self.resource_tracker.add_resource(plr_resource)
-                            self.transfer_to_new_resource(plr_resource, tree, additional_add_params)
-                        func = getattr(self.driver_instance, "resource_tree_add", None)
-                        if callable(func):
-                            func(plr_resources)
-                            results.append({"success": True, "action": "add"})
+                        result = _handle_add(plr_resources, tree_set, additional_add_params)
+                        results.append(result)
                     elif action == "update":
-                        # 更新资源
+                        if tree_set is None:
+                            raise ValueError("tree_set不能为None")
                         plr_resources = tree_set.to_plr_resources()
-                        for plr_resource, tree in zip(plr_resources, tree_set.trees):
-                            states = plr_resource.serialize_all_state()
-                            original_instance: ResourcePLR = self.resource_tracker.figure_resource(
-                                {"uuid": tree.root_node.res_content.uuid}, try_mode=False
-                            )
-                            original_parent_resource = original_instance.parent
-                            original_parent_resource_uuid = getattr(original_parent_resource, "unilabos_uuid", None)
-                            target_parent_resource_uuid = tree.root_node.res_content.uuid_parent
-                            self.lab_logger().info(
-                                f"物料{original_instance} 原始父节点{original_parent_resource_uuid} 目标父节点{target_parent_resource_uuid} 更新"
-                            )
-                            # todo: 对extra进行update
-                            if getattr(plr_resource, "unilabos_extra", None) is not None:
-                                original_instance.unilabos_extra = getattr(plr_resource, "unilabos_extra")
-                            if original_parent_resource_uuid != target_parent_resource_uuid and original_parent_resource is not None:
-                                self.transfer_to_new_resource(original_instance, tree, additional_add_params)
-                            original_instance.load_all_state(states)
-                            self.lab_logger().info(
-                                f"更新了资源属性 {plr_resource}[{tree.root_node.res_content.uuid}] 及其子节点 {len(original_instance.get_all_children())} 个"
-                            )
-
-                        func = getattr(self.driver_instance, "resource_tree_update", None)
-                        if callable(func):
-                            func(plr_resources)
-                            results.append({"success": True, "action": "update"})
+                        result = _handle_update(plr_resources, tree_set, additional_add_params)
+                        results.append(result)
                     elif action == "remove":
-                        # 移除资源
-                        found_resources: List[List[Union[ResourcePLR, dict]]] = self.resource_tracker.figure_resource(
-                            [{"uuid": uid} for uid in resources_uuid], try_mode=True
-                        )
-                        found_plr_resources = []
-                        other_plr_resources = []
-                        for found_resource in found_resources:
-                            for resource in found_resource:
-                                if issubclass(resource.__class__, ResourcePLR):
-                                    found_plr_resources.append(resource)
-                                else:
-                                    other_plr_resources.append(resource)
-                        func = getattr(self.driver_instance, "resource_tree_remove", None)
-                        if callable(func):
-                            func(found_plr_resources)
-                        for plr_resource in found_plr_resources:
-                            if plr_resource.parent is not None:
-                                plr_resource.parent.unassign_child_resource(plr_resource)
-                            self.resource_tracker.remove_resource(plr_resource)
-                            self.lab_logger().info(f"移除物料 {plr_resource} 及其子节点")
-                        for other_plr_resource in other_plr_resources:
-                            self.resource_tracker.remove_resource(other_plr_resource)
-                            self.lab_logger().info(f"移除物料 {other_plr_resource} 及其子节点")
-                        results.append({"success": True, "action": "remove"})
+                        result = _handle_remove(resources_uuid)
+                        results.append(result)
                 except Exception as e:
                     error_msg = f"Error processing {action} operation: {str(e)}"
                     self.lab_logger().error(f"[Resource Tree Update] {error_msg}")
@@ -725,7 +826,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
 
             # 返回处理结果
             result_json = {"results": results, "total": len(data)}
-            res.response = json.dumps(result_json, ensure_ascii=False)
+            res.response = json.dumps(result_json, ensure_ascii=False, cls=TypeEncoder)
             self.lab_logger().info(f"[Resource Tree Update] Completed processing {len(data)} operations")
 
         except json.JSONDecodeError as e:
@@ -995,9 +1096,14 @@ class BaseROS2DeviceNode(Node, Generic[T]):
 
                             # 通过资源跟踪器获取本地实例
                             final_resources = queried_resources if is_sequence else queried_resources[0]
-                            final_resources = self.resource_tracker.figure_resource({"name": final_resources.name}, try_mode=False) if not is_sequence else [
-                                self.resource_tracker.figure_resource({"name": res.name}, try_mode=False) for res in queried_resources
-                            ]
+                            final_resources = (
+                                self.resource_tracker.figure_resource({"name": final_resources.name}, try_mode=False)
+                                if not is_sequence
+                                else [
+                                    self.resource_tracker.figure_resource({"name": res.name}, try_mode=False)
+                                    for res in queried_resources
+                                ]
+                            )
                             action_kwargs[k] = final_resources
 
                         except Exception as e:
@@ -1218,6 +1324,7 @@ class BaseROS2DeviceNode(Node, Generic[T]):
             raise JsonCommandInitError(
                 f"执行动作时JSON缺少function_name或function_args: {ex}\n原JSON: {string}\n{traceback.format_exc()}"
             )
+
     def _convert_resource_sync(self, resource_data: Dict[str, Any]):
         """同步转换资源数据为实例"""
         # 创建资源查询请求
@@ -1385,18 +1492,27 @@ class ROS2DeviceNode:
     它不继承设备类，而是通过代理模式访问设备类的属性和方法。
     """
 
-    # 类变量，用于循环管理
-    _loop = None
-    _loop_running = False
-    _loop_thread = None
+    @classmethod
+    def run_async_func(cls, func, trace_error=True, **kwargs) -> Task:
+        def _handle_future_exception(fut):
+            try:
+                fut.result()
+            except Exception as e:
+                error(f"异步任务 {func.__name__} 报错了")
+                error(traceback.format_exc())
+
+        future = rclpy.get_global_executor().create_task(func(**kwargs))
+        if trace_error:
+            future.add_done_callback(_handle_future_exception)
+        return future
 
     @classmethod
-    def get_loop(cls):
-        return cls._loop
-
-    @classmethod
-    def run_async_func(cls, func, trace_error=True, **kwargs):
-        return run_async_func(func, loop=cls._loop, trace_error=trace_error, **kwargs)
+    async def async_wait_for(cls, node: Node, wait_time: float, callback_group=None):
+        future = Future()
+        timer = node.create_timer(wait_time, lambda : future.set_result(None), callback_group=callback_group, clock=node.get_clock())
+        await future
+        timer.cancel()
+        node.destroy_timer(timer)
 
     @property
     def driver_instance(self):
@@ -1436,11 +1552,6 @@ class ROS2DeviceNode:
             print_publish: 是否打印发布信息
             driver_is_ros:
         """
-        # 在初始化时检查循环状态
-        if ROS2DeviceNode._loop_running and ROS2DeviceNode._loop_thread is not None:
-            pass
-        elif ROS2DeviceNode._loop_thread is None:
-            self._start_loop()
 
         # 保存设备类是否支持异步上下文
         self._has_async_context = hasattr(driver_class, "__aenter__") and hasattr(driver_class, "__aexit__")
@@ -1529,17 +1640,6 @@ class ROS2DeviceNode:
             except Exception as e:
                 self._ros_node.lab_logger().error(f"设备后初始化失败: {e}")
 
-    def _start_loop(self):
-        def run_event_loop():
-            loop = asyncio.new_event_loop()
-            ROS2DeviceNode._loop = loop
-            asyncio.set_event_loop(loop)
-            loop.run_forever()
-
-        ROS2DeviceNode._loop_thread = threading.Thread(target=run_event_loop, daemon=True, name="ROS2DeviceNodeLoop")
-        ROS2DeviceNode._loop_thread.start()
-        logger.info(f"循环线程已启动")
-
 
 class DeviceInfoType(TypedDict):
     id: str