Compare commits

..

1 Commits

Author SHA1 Message Date
e2456166b4 [P] Integrate additional store testing into CI pipeline
Some checks failed
Component testing / Hub testing (push) Successful in 27s
Component testing / Store testing (push) Failing after 1m23s
Component testing / Integration smoke testing (push) Has been skipped
2026-03-27 16:34:12 +02:00
8 changed files with 26 additions and 218 deletions

View File

@@ -7,13 +7,10 @@ jobs:
name: Reset docker state name: Reset docker state
steps: steps:
- name: Stop all containers - name: Stop all containers
if: ${{always()}}
run: docker stop $(docker ps -a | cut -d " " -f 1 | tail -n +2) run: docker stop $(docker ps -a | cut -d " " -f 1 | tail -n +2)
- name: Remove all containers - name: Remove all containers
if: ${{always()}}
run: docker rm $(docker ps -a | cut -d " " -f 1 | tail -n +2) run: docker rm $(docker ps -a | cut -d " " -f 1 | tail -n +2)
- name: Remove extra volumes - name: Remove extra volumes
if: ${{always()}}
run: docker volume rm road_vision_postgres_data road_vision_pgadmin-data run: docker volume rm road_vision_postgres_data road_vision_pgadmin-data

View File

@@ -38,7 +38,7 @@ jobs:
- name: Run Store tests - name: Run Store tests
working-directory: IoT-Systems working-directory: IoT-Systems
run: docker run --network host --rm -it local/store/${{gitea.sha}} run: docker run --rm -it local/store/${{gitea.sha}}
- name: Clean up containers - name: Clean up containers
if: ${{always()}} if: ${{always()}}

View File

@@ -9,7 +9,6 @@ from config import STORE_HOST, STORE_PORT
# Pydantic models # Pydantic models
class ProcessedAgentData(BaseModel): class ProcessedAgentData(BaseModel):
id: int
road_state: str road_state: str
user_id: int user_id: int
x: float x: float
@@ -33,12 +32,11 @@ class ProcessedAgentData(BaseModel):
class Datasource: class Datasource:
def __init__(self): def __init__(self, user_id: int):
self.websocket: Connection | None = None
self.index = 0 self.index = 0
self.user_id = user_id
self.connection_status = None self.connection_status = None
self._new_points = [] self._new_points = []
self._active_markers = []
asyncio.ensure_future(self.connect_to_server()) asyncio.ensure_future(self.connect_to_server())
def get_new_points(self): def get_new_points(self):
@@ -48,12 +46,11 @@ class Datasource:
return points return points
async def connect_to_server(self): async def connect_to_server(self):
uri = f"ws://{STORE_HOST}:{STORE_PORT}/ws" uri = f"ws://{STORE_HOST}:{STORE_PORT}/ws/{self.user_id}"
while True: while True:
Logger.debug("CONNECT TO SERVER") Logger.debug("CONNECT TO SERVER")
async with websockets.connect(uri) as websocket: async with websockets.connect(uri) as websocket:
self.connection_status = "Connected" self.connection_status = "Connected"
self.websocket = websocket
try: try:
while True: while True:
data = await websocket.recv() data = await websocket.recv()
@@ -63,26 +60,6 @@ class Datasource:
self.connection_status = "Disconnected" self.connection_status = "Disconnected"
Logger.debug("SERVER DISCONNECT") Logger.debug("SERVER DISCONNECT")
def update_db_record_visibility(self, record_id):
if self.websocket:
data = json.dumps({"id": record_id})
asyncio.ensure_future(self.websocket.send(data))
def map_lat_lon_to_processed_agent_data(self, lat: float, lon: float) -> ProcessedAgentData | None:
distances = tuple((abs(lon - marker.latitude) ** 2 + abs(lat - marker.longitude) ** 2) ** 0.5 for marker in
self._active_markers)
if len(distances) == 0:
return None
min_distance = min(distances)
marker = self._active_markers[distances.index(min_distance)]
if min_distance < 0.005:
return marker
else:
return None
def handle_received_data(self, data): def handle_received_data(self, data):
# Update your UI or perform actions with received data here # Update your UI or perform actions with received data here
Logger.debug(f"Received data: {data}") Logger.debug(f"Received data: {data}")
@@ -93,9 +70,6 @@ class Datasource:
], ],
key=lambda v: v.timestamp, key=lambda v: v.timestamp,
) )
self._active_markers += [i for i in processed_agent_data_list if i.road_state != 'normal']
new_points = [ new_points = [
( (
processed_agent_data.longitude, processed_agent_data.longitude,

View File

@@ -15,31 +15,18 @@ line_layer_colors = [
[1, 0, 1, 1], [1, 0, 1, 1],
] ]
def get_lat_lon(point: dict[str, float] | list[float] | tuple[float, float]) -> tuple[float, float] | None:
if isinstance(point, dict):
lat = point.get("lat")
lon = point.get("lon")
else:
lat, lon = point
if lat is None or lon is None:
return None
return lat, lon
class MapViewApp(App): class MapViewApp(App):
def __init__(self, **kwargs): def __init__(self, **kwargs):
super().__init__(**kwargs) super().__init__(**kwargs)
self.mapview: MapView | None = None self.mapview = None
self.datasource = Datasource() self.datasource = Datasource(user_id=1)
self.line_layers = dict() self.line_layers = dict()
self.car_markers = dict() self.car_markers = dict()
# додати необхідні змінні # додати необхідні змінні
self.bump_markers: list[MapMarker] = [] self.bump_markers = []
self.pothole_markers: list[MapMarker] = [] self.pothole_markers = []
def on_start(self): def on_start(self):
""" """
@@ -63,7 +50,7 @@ class MapViewApp(App):
# Оновлює лінію маршрута # Оновлює лінію маршрута
if user_id not in self.line_layers: if user_id not in self.line_layers:
self.line_layers[user_id] = LineMapLayer(color=line_layer_colors[user_id % len(line_layer_colors)]) self.line_layers[user_id] = LineMapLayer(color = line_layer_colors[user_id % len(line_layer_colors)])
self.mapview.add_layer(self.line_layers[user_id]) self.mapview.add_layer(self.line_layers[user_id])
self.line_layers[user_id].add_point((lat, lon)) self.line_layers[user_id].add_point((lat, lon))
@@ -104,16 +91,13 @@ class MapViewApp(App):
if user_id == config.TRACK_ID: if user_id == config.TRACK_ID:
self.mapview.center_on(lat, lon) self.mapview.center_on(lat, lon)
def map_lat_lon_to_marker(self, lat: float, lon: float) -> MapMarker | None:
flt = filter(lambda marker: lon == marker.lat and lat == marker.lon, self.pothole_markers + self.bump_markers)
try:
return next(flt)
except StopIteration:
return None
def set_pothole_marker(self, point): def set_pothole_marker(self, point):
lat, lon = get_lat_lon(point) if isinstance(point, dict):
lat = point.get("lat")
lon = point.get("lon")
else:
lat, lon = point
if lat is None or lon is None: if lat is None or lon is None:
return return
@@ -127,48 +111,24 @@ class MapViewApp(App):
self.pothole_markers.append(marker) self.pothole_markers.append(marker)
def set_bump_marker(self, point): def set_bump_marker(self, point):
lat, lon = get_lat_lon(point) if isinstance(point, dict):
lat = point.get("lat")
lon = point.get("lon")
else:
lat, lon = point
if lat is None or lon is None: if lat is None or lon is None:
return return
marker = MapMarker( marker = MapMarker(
lat=lat, lat=lat,
lon=lon, lon=lon,
source="images/bump.png" source="images/bump.png"
) )
self.mapview.add_marker(marker) self.mapview.add_marker(marker)
self.bump_markers.append(marker) self.bump_markers.append(marker)
def delete_pothole_marker(self, point):
lat, lon = get_lat_lon(point)
if lat is None or lon is None:
return
clicked_marker_data = self.datasource.map_lat_lon_to_processed_agent_data(lat, lon)
if not clicked_marker_data:
return
clicked_marker = self.map_lat_lon_to_marker(clicked_marker_data.latitude, clicked_marker_data.longitude)
if clicked_marker is None:
return
self.mapview.remove_marker(clicked_marker)
if clicked_marker in self.pothole_markers:
self.pothole_markers.pop(self.pothole_markers.index(clicked_marker))
elif clicked_marker in self.bump_markers:
self.bump_markers.pop(self.bump_markers.index(clicked_marker))
self.datasource.update_db_record_visibility(clicked_marker_data.id)
def on_touch_down(self, widget, touch):
if touch.button == "right":
coordinate = self.mapview.get_latlon_at(touch.x, touch.y, self.mapview.zoom)
self.delete_pothole_marker(coordinate)
return True
def build(self): def build(self):
""" """
@@ -181,8 +141,6 @@ class MapViewApp(App):
lon=30.5234 lon=30.5234
) )
self.mapview.bind(on_touch_down=self.on_touch_down)
return self.mapview return self.mapview

View File

@@ -1,81 +0,0 @@
@startuml
interface MQTT as mqtt1
interface MQTT as mqtt2
interface HTTP as http1
interface HTTP as http2
interface WebSocket
node Agent {
component Accelerometer
component GPS
component CarParkingSlots
component "Data Aggregator" as agent_agg
}
node Edge {
component AgentMQTTAdapter
component AgentDataProcessor
component HubMQTTAdapter
component HubHTTPAdapter
}
node Hub {
component EdgeHTTPAdapter
component EdgeMQTTAdapter
component Aggregator
component StoreHTTPAdapter
component Redis
}
node Store {
component "HTTP Rest API" as store_http
component "SQL Adapter" as store_sql
component "Websocket API" as store_ws
}
node MapView {
component "Store Processor" as mv_sp
component "Input Handler" as mv_ih
component "Map"
}
component PostgreSQL
agent_agg --> Accelerometer
agent_agg --> GPS
agent_agg --> CarParkingSlots
agent_agg -> mqtt1
mqtt1 <- AgentMQTTAdapter
AgentMQTTAdapter -> AgentDataProcessor
AgentDataProcessor --> HubMQTTAdapter
AgentDataProcessor --> HubHTTPAdapter
HubMQTTAdapter --> mqtt2
HubHTTPAdapter --> http1
mqtt2 <-- EdgeMQTTAdapter
http1 <-- EdgeHTTPAdapter
EdgeMQTTAdapter --> Aggregator
EdgeHTTPAdapter --> Aggregator
Aggregator <--> Redis
StoreHTTPAdapter <-- Aggregator
http2 <-- StoreHTTPAdapter
store_http --> http2
store_http -> store_sql
store_sql <- store_ws
store_ws --> WebSocket
store_sql <--> PostgreSQL
WebSocket <-- mv_sp
mv_sp <-- mv_ih
mv_sp --> Map
actor Spectator as user
Map --> user
mv_ih <-- user
@enduml

View File

@@ -1,22 +0,0 @@
@startuml
actor Worker as worker
participant Agent as agent
participant "Edge" as edge
participant "Hub" as hub
participant "Store" as store
participant "MapView" as mapview
actor Spectator as spectator
worker -> agent : Start agent
agent -> edge : Send sensor data
edge -> edge : Classify road state
edge -> hub : Send processed data
hub -> hub : Save processed data
hub -> store : Send processed data batch
store -> store : Save processed data batch
store -> mapview : Send new rows
mapview -> spectator : Show updated paths and markers
spectator -> mapview : Click on markers
mapview -> spectator : Hide clicked markers
mapview -> store : Save display preference
@enduml

View File

@@ -1,15 +0,0 @@
@startuml
|Користувач|
start
:Натискає на іконку ями;
|Система|
:Запитує підтвердження видалення ями;
|Користувач|
:Підтверджує видалення ями;
|Система|
:Модифікує запис про яму;
note right #ff9999: Можлива виключна ситуація 1
:Видаляє яму;
|Користувач|
stop
@enduml

View File

@@ -39,18 +39,15 @@ subscriptions: Set[WebSocket] = set()
# FastAPI WebSocket endpoint # FastAPI WebSocket endpoint
@app.websocket("/ws") @app.websocket("/ws/{user_id}")
async def websocket_endpoint(websocket: WebSocket): async def websocket_endpoint(websocket: WebSocket, user_id: int):
await websocket.accept() await websocket.accept()
subscriptions.add(websocket) subscriptions.add(websocket)
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)
.where(processed_agent_data.c.visible == True) \
.order_by(processed_agent_data.c.timestamp)
session = SessionLocal() session = SessionLocal()
stored_data = session.execute(r).fetchall() stored_data = session.execute(r).fetchall()
session.close() session.close()