61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import logging
|
|
from app.adapters.agent_mqtt_adapter import AgentMQTTAdapter
|
|
from app.adapters.hub_http_adapter import HubHttpAdapter
|
|
from app.adapters.hub_mqtt_adapter import HubMqttAdapter
|
|
from config import (
|
|
MQTT_BROKER_HOST,
|
|
MQTT_BROKER_PORT,
|
|
MQTT_TOPIC,
|
|
HUB_URL,
|
|
HUB_MQTT_BROKER_HOST,
|
|
HUB_MQTT_BROKER_PORT,
|
|
HUB_MQTT_TOPIC,
|
|
HUB_CONNECTION_TYPE,
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
# Configure logging settings
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="[%(asctime)s] [%(levelname)s] [%(module)s] %(message)s",
|
|
handlers=[
|
|
logging.StreamHandler(),
|
|
logging.FileHandler("app.log"),
|
|
],
|
|
)
|
|
|
|
# Logic to select the adapter based on configuration (SCRUM-93 & SCRUM-94)
|
|
# This allows easy switching between HTTP and MQTT protocols
|
|
if HUB_CONNECTION_TYPE.lower() == "http":
|
|
logging.info("Initializing HubHttpAdapter (SCRUM-93 integration)")
|
|
hub_adapter = HubHttpAdapter(
|
|
api_base_url=HUB_URL,
|
|
)
|
|
else:
|
|
logging.info("Initializing HubMqttAdapter (SCRUM-94 integration)")
|
|
hub_adapter = HubMqttAdapter(
|
|
broker=HUB_MQTT_BROKER_HOST,
|
|
port=HUB_MQTT_BROKER_PORT,
|
|
topic=HUB_MQTT_TOPIC,
|
|
)
|
|
|
|
# Create an instance of the AgentMQTTAdapter using the selected hub adapter
|
|
# This adapter acts as a bridge between the Agent and the Hub
|
|
agent_adapter = AgentMQTTAdapter(
|
|
broker_host=MQTT_BROKER_HOST,
|
|
broker_port=MQTT_BROKER_PORT,
|
|
topic=MQTT_TOPIC,
|
|
hub_gateway=hub_adapter,
|
|
)
|
|
|
|
try:
|
|
logging.info(f"Connecting to MQTT broker at {MQTT_BROKER_HOST}:{MQTT_BROKER_PORT}")
|
|
agent_adapter.connect()
|
|
|
|
logging.info("Broker connection success. Waiting for data...")
|
|
agent_adapter.loop_forever()
|
|
except KeyboardInterrupt:
|
|
logging.info("Interrupt signal received. Shutting down...")
|
|
agent_adapter.disconnect()
|
|
logging.info("Disconnected from MQTT broker.")
|