Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 119547d288 | |||
| b58167f0de | |||
| 121bd007b3 |
+4
-3
@@ -1,17 +1,18 @@
|
||||
import os
|
||||
|
||||
|
||||
def try_parse(type, value: str):
|
||||
try:
|
||||
return type(value)
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
|
||||
USER_ID = try_parse(int, os.environ.get("USER_ID")) or 1
|
||||
# MQTT config
|
||||
MQTT_BROKER_HOST = os.environ.get("MQTT_BROKER_HOST") or "mqtt"
|
||||
MQTT_BROKER_PORT = try_parse(int, os.environ.get("MQTT_BROKER_PORT")) or 1883
|
||||
MQTT_TOPIC = os.environ.get("MQTT_TOPIC") or "agent"
|
||||
|
||||
# Data-related config
|
||||
USER_ID = try_parse(int, os.environ.get("USER_ID")) or 1
|
||||
# Delay for sending data to mqtt in seconds
|
||||
DELAY = try_parse(float, os.environ.get("DELAY")) or 1
|
||||
GPS_SOURCE = os.environ.get("GPS_SOURCE") or "data/gps.csv"
|
||||
|
||||
+1
-1
@@ -42,7 +42,7 @@ def run():
|
||||
# Prepare mqtt client
|
||||
client = connect_mqtt(config.MQTT_BROKER_HOST, config.MQTT_BROKER_PORT)
|
||||
# Prepare datasource
|
||||
datasource = FileDatasource(16384.0, "data/accelerometer.csv", config.GPS_SOURCE, "data/parking.csv")
|
||||
datasource = FileDatasource(16384.0, "data/accelerometer.csv", "data/gps.csv", "data/parking.csv")
|
||||
# Infinity publish data
|
||||
publish(client, config.MQTT_TOPIC, datasource)
|
||||
|
||||
|
||||
+3
-20
@@ -14,7 +14,8 @@ services:
|
||||
mqtt_network:
|
||||
|
||||
|
||||
agent1:
|
||||
fake_agent:
|
||||
container_name: agent
|
||||
build:
|
||||
context: .
|
||||
dockerfile: agent/Dockerfile
|
||||
@@ -25,25 +26,7 @@ services:
|
||||
MQTT_BROKER_HOST: "mqtt"
|
||||
MQTT_BROKER_PORT: 1883
|
||||
MQTT_TOPIC: "agent_data_topic"
|
||||
DELAY: 1.2
|
||||
USER_ID: 2
|
||||
networks:
|
||||
mqtt_network:
|
||||
|
||||
agent2:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: agent/Dockerfile
|
||||
depends_on:
|
||||
- mqtt
|
||||
environment:
|
||||
PYTHONUNBUFFERED: 1
|
||||
MQTT_BROKER_HOST: "mqtt"
|
||||
MQTT_BROKER_PORT: 1883
|
||||
MQTT_TOPIC: "agent_data_topic"
|
||||
GPS_SOURCE: "data/route2.csv"
|
||||
DELAY: 1.0
|
||||
USER_ID: 3
|
||||
DELAY: 0.1
|
||||
networks:
|
||||
mqtt_network:
|
||||
|
||||
|
||||
+1
-1
@@ -45,7 +45,7 @@ async def websocket_endpoint(websocket: WebSocket, user_id: int):
|
||||
|
||||
try:
|
||||
# send already available data
|
||||
r = processed_agent_data.select()
|
||||
r = processed_agent_data.select().order_by(processed_agent_data.c.timestamp)
|
||||
stored_data = SessionLocal().execute(r).fetchall()
|
||||
|
||||
jsonable_data = [{c.name: getattr(i, c.name) for c in processed_agent_data.columns} for i in stored_data]
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
import sys
|
||||
import os
|
||||
|
||||
print("Checking for dead containers...")
|
||||
|
||||
@@ -14,6 +15,9 @@ for i in statuses:
|
||||
if not i[status_index:].startswith("Up "):
|
||||
service_name = i[name_index:]
|
||||
print(f"Crash detected in {service_name}")
|
||||
print(f"docker logs for the container:\n")
|
||||
os.system(f"docker logs {i.split(' ')[0]}")
|
||||
print()
|
||||
exit_code = 1
|
||||
|
||||
sys.exit(exit_code)
|
||||
|
||||
Reference in New Issue
Block a user