124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
MQTT设备处理服务测试
|
||
测试集成的MQTT服务功能
|
||
"""
|
||
|
||
import asyncio
|
||
import sys
|
||
import os
|
||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from services.online_script.device_handler_service import get_device_service, DeviceType
|
||
from services.online_script.mqtt_service import MQTTService
|
||
from datetime import datetime
|
||
import json
|
||
import time
|
||
|
||
|
||
def test_device_handler(message):
|
||
"""测试设备消息处理器"""
|
||
print(f"[设备处理器] 收到消息:")
|
||
print(f" 设备ID: {message.device_id}")
|
||
print(f" 设备类型: {message.device_type.value}")
|
||
print(f" 主题: {message.topic}")
|
||
print(f" 载荷: {json.dumps(message.payload, ensure_ascii=False, indent=2)}")
|
||
print(f" 时间戳: {datetime.fromtimestamp(message.timestamp).strftime('%Y-%m-%d %H:%M:%S')}")
|
||
print("-" * 50)
|
||
|
||
# 返回转发配置
|
||
return {
|
||
"forward": True,
|
||
"payload": {
|
||
"processed_by": "vwed_device_handler",
|
||
"original_data": message.payload,
|
||
"processed_time": datetime.now().isoformat()
|
||
}
|
||
}
|
||
|
||
|
||
async def test_mqtt_device_service():
|
||
"""测试MQTT设备处理服务"""
|
||
print("=== MQTT设备处理服务测试 ===")
|
||
|
||
# 获取设备服务实例
|
||
device_service = get_device_service()
|
||
|
||
try:
|
||
# 启动服务
|
||
print("启动设备处理服务...")
|
||
await device_service.start_service()
|
||
|
||
# 等待服务启动
|
||
await asyncio.sleep(2)
|
||
|
||
# 注册设备处理器
|
||
print("注册设备处理器...")
|
||
device_service.register_and_run(
|
||
device_id="test_robot_001",
|
||
device_type=DeviceType.ROBOT,
|
||
listen_topics=["vwed/device/test_robot_001/data", "vwed/task/status"],
|
||
forward_topics=["vwed/processed/test_robot_001"],
|
||
handler=test_device_handler,
|
||
script_id="test_script_001",
|
||
description="测试机器人设备处理器"
|
||
)
|
||
|
||
# 等待注册完成
|
||
await asyncio.sleep(1)
|
||
|
||
# 发送测试消息
|
||
print("发送测试消息...")
|
||
|
||
test_messages = [
|
||
{
|
||
"topic": "vwed/device/test_robot_001/data",
|
||
"payload": {
|
||
"robot_id": "test_robot_001",
|
||
"status": "running",
|
||
"battery": 75,
|
||
"position": {"x": 150, "y": 300},
|
||
"timestamp": datetime.now().isoformat()
|
||
}
|
||
},
|
||
{
|
||
"topic": "vwed/task/status",
|
||
"payload": {
|
||
"task_id": "task_001",
|
||
"status": "in_progress",
|
||
"progress": 60,
|
||
"timestamp": datetime.now().isoformat()
|
||
}
|
||
}
|
||
]
|
||
|
||
for msg in test_messages:
|
||
await device_service.publish_message(msg["topic"], msg["payload"])
|
||
print(f"已发布消息到主题: {msg['topic']}")
|
||
await asyncio.sleep(1)
|
||
|
||
# 保持运行一段时间以接收消息
|
||
print("保持服务运行30秒，等待消息处理...")
|
||
await asyncio.sleep(30)
|
||
|
||
# 查看统计信息
|
||
print("\n=== 设备处理统计 ===")
|
||
stats = device_service.get_running_handlers()
|
||
print(json.dumps(stats, ensure_ascii=False, indent=2))
|
||
|
||
except Exception as e:
|
||
print(f"测试过程中出错: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
|
||
finally:
|
||
# 停止服务
|
||
print("\n停止设备处理服务...")
|
||
await device_service.stop_service()
|
||
print("测试完成")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 运行测试
|
||
asyncio.run(test_mqtt_device_service()) |