67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
"""Log manager for storing device messages."""
|
|
|
|
import time
|
|
from typing import Dict, List, Optional
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class LogEntry:
|
|
"""A single log entry from a device."""
|
|
timestamp: float
|
|
device_id: str
|
|
msg_type: str
|
|
source: str
|
|
payload: str
|
|
request_id: Optional[str] = None
|
|
|
|
|
|
class LogManager:
|
|
"""Manages log storage for device messages."""
|
|
|
|
def __init__(self, max_entries_per_device: int = 1000):
|
|
self.max_entries = max_entries_per_device
|
|
self._logs: Dict[str, List[LogEntry]] = {}
|
|
|
|
def add(self, device_id: str, msg_type: str, source: str, payload: str, request_id: str = None):
|
|
if device_id not in self._logs:
|
|
self._logs[device_id] = []
|
|
|
|
entry = LogEntry(
|
|
timestamp=time.time(),
|
|
device_id=device_id,
|
|
msg_type=msg_type,
|
|
source=source,
|
|
payload=payload,
|
|
request_id=request_id
|
|
)
|
|
|
|
self._logs[device_id].append(entry)
|
|
|
|
if len(self._logs[device_id]) > self.max_entries:
|
|
self._logs[device_id] = self._logs[device_id][-self.max_entries:]
|
|
|
|
def get_logs(self, device_id: str, limit: int = 100) -> List[LogEntry]:
|
|
if device_id not in self._logs:
|
|
return []
|
|
return self._logs[device_id][-limit:]
|
|
|
|
def get_all_logs(self, limit: int = 100) -> List[LogEntry]:
|
|
all_entries = []
|
|
for entries in self._logs.values():
|
|
all_entries.extend(entries)
|
|
all_entries.sort(key=lambda e: e.timestamp)
|
|
return all_entries[-limit:]
|
|
|
|
def clear(self, device_id: str = None):
|
|
if device_id:
|
|
self._logs.pop(device_id, None)
|
|
else:
|
|
self._logs.clear()
|
|
|
|
def device_count(self) -> int:
|
|
return len(self._logs)
|
|
|
|
def total_entries(self) -> int:
|
|
return sum(len(entries) for entries in self._logs.values())
|