Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 119547d288 | |||
| b58167f0de | |||
| 121bd007b3 |
+4
-3
@@ -1,17 +1,18 @@
|
|||||||
import os
|
import os
|
||||||
|
|
||||||
|
|
||||||
def try_parse(type, value: str):
|
def try_parse(type, value: str):
|
||||||
try:
|
try:
|
||||||
return type(value)
|
return type(value)
|
||||||
except Exception:
|
except Exception:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
USER_ID = try_parse(int, os.environ.get("USER_ID")) or 1
|
||||||
# MQTT config
|
# MQTT config
|
||||||
MQTT_BROKER_HOST = os.environ.get("MQTT_BROKER_HOST") or "mqtt"
|
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_BROKER_PORT = try_parse(int, os.environ.get("MQTT_BROKER_PORT")) or 1883
|
||||||
MQTT_TOPIC = os.environ.get("MQTT_TOPIC") or "agent"
|
MQTT_TOPIC = os.environ.get("MQTT_TOPIC") or "agent"
|
||||||
|
|
||||||
# Data-related config
|
# Delay for sending data to mqtt in seconds
|
||||||
USER_ID = try_parse(int, os.environ.get("USER_ID")) or 1
|
|
||||||
DELAY = try_parse(float, os.environ.get("DELAY")) or 1
|
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
|
# Prepare mqtt client
|
||||||
client = connect_mqtt(config.MQTT_BROKER_HOST, config.MQTT_BROKER_PORT)
|
client = connect_mqtt(config.MQTT_BROKER_HOST, config.MQTT_BROKER_PORT)
|
||||||
# Prepare datasource
|
# 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
|
# Infinity publish data
|
||||||
publish(client, config.MQTT_TOPIC, datasource)
|
publish(client, config.MQTT_TOPIC, datasource)
|
||||||
|
|
||||||
|
|||||||
+3
-20
@@ -14,7 +14,8 @@ services:
|
|||||||
mqtt_network:
|
mqtt_network:
|
||||||
|
|
||||||
|
|
||||||
agent1:
|
fake_agent:
|
||||||
|
container_name: agent
|
||||||
build:
|
build:
|
||||||
context: .
|
context: .
|
||||||
dockerfile: agent/Dockerfile
|
dockerfile: agent/Dockerfile
|
||||||
@@ -25,25 +26,7 @@ services:
|
|||||||
MQTT_BROKER_HOST: "mqtt"
|
MQTT_BROKER_HOST: "mqtt"
|
||||||
MQTT_BROKER_PORT: 1883
|
MQTT_BROKER_PORT: 1883
|
||||||
MQTT_TOPIC: "agent_data_topic"
|
MQTT_TOPIC: "agent_data_topic"
|
||||||
DELAY: 1.2
|
DELAY: 0.1
|
||||||
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
|
|
||||||
networks:
|
networks:
|
||||||
mqtt_network:
|
mqtt_network:
|
||||||
|
|
||||||
|
|||||||
+1
-1
@@ -45,7 +45,7 @@ async def websocket_endpoint(websocket: WebSocket, user_id: int):
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
# send already available data
|
# 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()
|
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]
|
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 sys
|
||||||
|
import os
|
||||||
|
|
||||||
print("Checking for dead containers...")
|
print("Checking for dead containers...")
|
||||||
|
|
||||||
@@ -14,6 +15,9 @@ for i in statuses:
|
|||||||
if not i[status_index:].startswith("Up "):
|
if not i[status_index:].startswith("Up "):
|
||||||
service_name = i[name_index:]
|
service_name = i[name_index:]
|
||||||
print(f"Crash detected in {service_name}")
|
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
|
exit_code = 1
|
||||||
|
|
||||||
sys.exit(exit_code)
|
sys.exit(exit_code)
|
||||||
|
|||||||
Reference in New Issue
Block a user