mirror of
https://github.com/deepmodeling/Uni-Lab-OS
synced 2026-03-24 04:10:33 +00:00
c001f6a151
fast registry load minor fix on skill & registry stripe ros2 schema desc add create-device-skill new registry system backwards to yaml remove not exist resource new registry sys exp. support with add device add ai conventions correct raise create resource error ret info fix revert ret info fix fix prcxi check add create_resource schema re signal host ready event add websocket connection timeout and improve reconnection logic add open_timeout parameter to websocket connection add TimeoutError and InvalidStatus exception handling implement exponential backoff for reconnection attempts simplify reconnection logic flow add gzip change pose extra to any add isFlapY
73 lines
3.2 KiB
Python
73 lines
3.2 KiB
Python
import time
|
||
from typing import Any, Dict, Optional, Tuple
|
||
|
||
from unilabos.utils.log import logger
|
||
from unilabos.utils.tools import normalize_json as _normalize_device
|
||
|
||
|
||
def register_devices_and_resources(lab_registry, gather_only=False) -> Optional[Tuple[Dict[str, Any], Dict[str, Any]]]:
|
||
"""
|
||
注册设备和资源到服务器（仅支持HTTP）
|
||
"""
|
||
|
||
from unilabos.app.web.client import http_client
|
||
|
||
logger.info("[UniLab Register] 开始注册设备和资源...")
|
||
|
||
devices_to_register = {}
|
||
for device_info in lab_registry.obtain_registry_device_info():
|
||
devices_to_register[device_info["id"]] = _normalize_device(device_info)
|
||
logger.trace(f"[UniLab Register] 收集设备: {device_info['id']}")
|
||
|
||
resources_to_register = {}
|
||
for resource_info in lab_registry.obtain_registry_resource_info():
|
||
resources_to_register[resource_info["id"]] = resource_info
|
||
logger.trace(f"[UniLab Register] 收集资源: {resource_info['id']}")
|
||
|
||
if gather_only:
|
||
return devices_to_register, resources_to_register
|
||
|
||
if devices_to_register:
|
||
try:
|
||
start_time = time.time()
|
||
response = http_client.resource_registry(
|
||
{"resources": list(devices_to_register.values())},
|
||
tag="device_registry",
|
||
)
|
||
cost_time = time.time() - start_time
|
||
res_data = response.json() if response.status_code == 200 else {}
|
||
skipped = res_data.get("data", {}).get("skipped", False)
|
||
if skipped:
|
||
logger.info(
|
||
f"[UniLab Register] 设备注册跳过（内容未变化）"
|
||
f" {len(devices_to_register)} 个 {cost_time:.3f}s"
|
||
)
|
||
elif response.status_code in [200, 201]:
|
||
logger.info(f"[UniLab Register] 成功注册 {len(devices_to_register)} 个设备 {cost_time:.3f}s")
|
||
else:
|
||
logger.error(f"[UniLab Register] 设备注册失败: {response.status_code}, {response.text} {cost_time:.3f}s")
|
||
except Exception as e:
|
||
logger.error(f"[UniLab Register] 设备注册异常: {e}")
|
||
|
||
if resources_to_register:
|
||
try:
|
||
start_time = time.time()
|
||
response = http_client.resource_registry(
|
||
{"resources": list(resources_to_register.values())},
|
||
tag="resource_registry",
|
||
)
|
||
cost_time = time.time() - start_time
|
||
res_data = response.json() if response.status_code == 200 else {}
|
||
skipped = res_data.get("data", {}).get("skipped", False)
|
||
if skipped:
|
||
logger.info(
|
||
f"[UniLab Register] 资源注册跳过（内容未变化）"
|
||
f" {len(resources_to_register)} 个 {cost_time:.3f}s"
|
||
)
|
||
elif response.status_code in [200, 201]:
|
||
logger.info(f"[UniLab Register] 成功注册 {len(resources_to_register)} 个资源 {cost_time:.3f}s")
|
||
else:
|
||
logger.error(f"[UniLab Register] 资源注册失败: {response.status_code}, {response.text} {cost_time:.3f}s")
|
||
except Exception as e:
|
||
logger.error(f"[UniLab Register] 资源注册异常: {e}")
|