118 lines
3.5 KiB
Python
118 lines
3.5 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.mqtt_service import MQTTService
|
||
from datetime import datetime
|
||
import json
|
||
import time
|
||
|
||
|
||
def test_message_handler(topic: str, payload: str):
|
||
"""测试消息处理器"""
|
||
print(f"[消息处理器] 收到消息:")
|
||
print(f" 主题: {topic}")
|
||
print(f" 内容: {payload}")
|
||
print(f" 时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
||
print("-" * 50)
|
||
|
||
|
||
def test_mqtt_service():
|
||
"""测试独立MQTT服务"""
|
||
print("=== MQTT服务独立测试 ===")
|
||
|
||
# 创建MQTT服务
|
||
mqtt_service = MQTTService()
|
||
|
||
try:
|
||
# 连接到服务器
|
||
print("连接MQTT服务器...")
|
||
if mqtt_service.connect():
|
||
print("MQTT连接成功")
|
||
|
||
# 订阅主题并添加处理器
|
||
test_topics = [
|
||
"vwed/test/message",
|
||
"vwed/device/+/status",
|
||
"vwed/task/status"
|
||
]
|
||
|
||
for topic in test_topics:
|
||
mqtt_service.subscribe(topic, test_message_handler)
|
||
print(f"已订阅主题: {topic}")
|
||
|
||
# 等待订阅完成
|
||
time.sleep(2)
|
||
|
||
# 发布测试消息
|
||
print("\n发布测试消息...")
|
||
test_messages = [
|
||
{
|
||
"topic": "vwed/test/message",
|
||
"payload": {
|
||
"message": "Hello MQTT Service!",
|
||
"timestamp": datetime.now().isoformat(),
|
||
"test_id": "test_001"
|
||
}
|
||
},
|
||
{
|
||
"topic": "vwed/device/robot_001/status",
|
||
"payload": {
|
||
"device_id": "robot_001",
|
||
"status": "online",
|
||
"battery": 85,
|
||
"position": {"x": 100, "y": 200}
|
||
}
|
||
},
|
||
{
|
||
"topic": "vwed/task/status",
|
||
"payload": {
|
||
"task_id": "task_001",
|
||
"status": "running",
|
||
"progress": 75
|
||
}
|
||
}
|
||
]
|
||
|
||
for msg in test_messages:
|
||
success = mqtt_service.publish(msg["topic"], msg["payload"])
|
||
if success:
|
||
print(f"✓ 成功发布消息到: {msg['topic']}")
|
||
else:
|
||
print(f"✗ 发布消息失败: {msg['topic']}")
|
||
time.sleep(1)
|
||
|
||
# 保持连接，监听消息
|
||
print("\n保持连接30秒，监听消息...")
|
||
time.sleep(30)
|
||
|
||
# 显示连接信息
|
||
print("\n=== 连接信息 ===")
|
||
info = mqtt_service.get_connection_info()
|
||
print(json.dumps(info, ensure_ascii=False, indent=2))
|
||
|
||
else:
|
||
print("MQTT连接失败")
|
||
|
||
except Exception as e:
|
||
print(f"测试过程中出错: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
|
||
finally:
|
||
# 断开连接
|
||
print("\n断开MQTT连接...")
|
||
mqtt_service.disconnect()
|
||
print("测试完成")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
test_mqtt_service() |