153 lines
4.9 KiB
Python
153 lines
4.9 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
MQTT Connection Test Example
|
|
IP: 192.168.189.97
|
|
Port: 1883
|
|
"""
|
|
|
|
import paho.mqtt.client as mqtt
|
|
import time
|
|
import json
|
|
from datetime import datetime
|
|
|
|
|
|
class MQTTService:
|
|
def __init__(self, host="192.168.189.97", port=1883, keepalive=60):
|
|
self.host = host
|
|
self.port = port
|
|
self.keepalive = keepalive
|
|
self.client = mqtt.Client()
|
|
|
|
# Set callback functions
|
|
self.client.on_connect = self.on_connect
|
|
self.client.on_message = self.on_message
|
|
self.client.on_disconnect = self.on_disconnect
|
|
self.client.on_publish = self.on_publish
|
|
|
|
def on_connect(self, client, userdata, flags, rc):
|
|
"""Connection callback"""
|
|
if rc == 0:
|
|
print(f"Successfully connected to MQTT server {self.host}:{self.port}")
|
|
# Subscribe to topics after successful connection
|
|
self.client.subscribe("vwed/task/status")
|
|
self.client.subscribe("vwed/device/+/data")
|
|
print("Subscribed to topics: vwed/task/status, vwed/device/+/data")
|
|
else:
|
|
print(f"Connection failed, error code: {rc}")
|
|
|
|
def on_message(self, client, userdata, msg):
|
|
"""Message received callback"""
|
|
try:
|
|
topic = msg.topic
|
|
payload = msg.payload.decode('utf-8')
|
|
print(f"Received message - Topic: {topic}")
|
|
print(f"Payload: {payload}")
|
|
print(f"Time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
|
print("-" * 50)
|
|
except Exception as e:
|
|
print(f"Error processing message: {e}")
|
|
|
|
def on_disconnect(self, client, userdata, rc):
|
|
"""Disconnect callback"""
|
|
print(f"Disconnected from MQTT server, code: {rc}")
|
|
|
|
def on_publish(self, client, userdata, mid):
|
|
"""Message publish callback"""
|
|
print(f"Message published successfully, message ID: {mid}")
|
|
|
|
def connect(self):
|
|
"""Connect to MQTT server"""
|
|
try:
|
|
print(f"Connecting to MQTT server {self.host}:{self.port}...")
|
|
self.client.connect(self.host, self.port, self.keepalive)
|
|
return True
|
|
except Exception as e:
|
|
print(f"Connection failed: {e}")
|
|
return False
|
|
|
|
def disconnect(self):
|
|
"""Disconnect from server"""
|
|
self.client.disconnect()
|
|
|
|
def publish(self, topic, payload, qos=0, retain=False):
|
|
"""Publish message"""
|
|
try:
|
|
if isinstance(payload, dict):
|
|
payload = json.dumps(payload, ensure_ascii=False)
|
|
|
|
result = self.client.publish(topic, payload, qos, retain)
|
|
if result.rc == mqtt.MQTT_ERR_SUCCESS:
|
|
print(f"Published message to topic {topic}: {payload}")
|
|
return True
|
|
else:
|
|
print(f"Failed to publish message, error code: {result.rc}")
|
|
return False
|
|
except Exception as e:
|
|
print(f"Error publishing message: {e}")
|
|
return False
|
|
|
|
def subscribe(self, topic, qos=0):
|
|
"""Subscribe to topic"""
|
|
try:
|
|
result = self.client.subscribe(topic, qos)
|
|
if result[0] == mqtt.MQTT_ERR_SUCCESS:
|
|
print(f"Successfully subscribed to topic: {topic}")
|
|
return True
|
|
else:
|
|
print(f"Failed to subscribe to topic, error code: {result[0]}")
|
|
return False
|
|
except Exception as e:
|
|
print(f"Error subscribing to topic: {e}")
|
|
return False
|
|
|
|
def start_loop(self):
|
|
"""Start message loop"""
|
|
self.client.loop_start()
|
|
|
|
def stop_loop(self):
|
|
"""Stop message loop"""
|
|
self.client.loop_stop()
|
|
|
|
|
|
def test_mqtt_connection():
|
|
"""Test MQTT connection"""
|
|
print("=== MQTT Connection Test ===")
|
|
|
|
# Create MQTT service instance
|
|
mqtt_service = MQTTService()
|
|
|
|
# Connect to server
|
|
if mqtt_service.connect():
|
|
# Start message loop
|
|
mqtt_service.start_loop()
|
|
|
|
# Wait for connection to establish
|
|
time.sleep(2)
|
|
|
|
# Publish test messages
|
|
test_data = {
|
|
"device_id": "robot_001",
|
|
"status": "running",
|
|
"battery": 85,
|
|
"position": {"x": 100, "y": 200},
|
|
"timestamp": datetime.now().isoformat()
|
|
}
|
|
|
|
mqtt_service.publish("vwed/device/robot_001/data", test_data)
|
|
mqtt_service.publish("vwed/task/status", "Task is running...")
|
|
|
|
# Keep connection alive and listen for messages
|
|
print("Keeping connection alive for 30 seconds, listening for messages...")
|
|
time.sleep(30)
|
|
|
|
# Stop loop and disconnect
|
|
mqtt_service.stop_loop()
|
|
mqtt_service.disconnect()
|
|
print("Test completed")
|
|
else:
|
|
print("Unable to connect to MQTT server")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_mqtt_connection() |