This commit is contained in:
legop3
2026-04-29 14:22:44 -04:00
parent b716e40f53
commit 2e140d1bd5
2 changed files with 23 additions and 5 deletions
@@ -10,6 +10,7 @@ const SNAPSHOT_DIR = process.env.ROVER_SNAPSHOT_DIR || '/var/lib/rover-snapshots
const POLL_INTERVAL_MS = 300; const POLL_INTERVAL_MS = 300;
const roverState = new Map(); const roverState = new Map();
const events = new EventEmitter(); const events = new EventEmitter();
const readCounts = new Map();
let pollTimer = null; let pollTimer = null;
function markState(id, updates = {}) { function markState(id, updates = {}) {
@@ -24,24 +25,32 @@ function getSnapshotPath(id) {
} }
function createRoverSnapshotPoller({ roverManager }) { function createRoverSnapshotPoller({ roverManager }) {
async function fetchSnapshot(id) { async function fetchSnapshot(id, options = {}) {
const force = Boolean(options?.force);
const state = roverState.get(id); const state = roverState.get(id);
if (state?.fetching) return; if (state?.fetching) return;
markState(id, { fetching: true }); markState(id, { fetching: true });
try { try {
const filePath = getSnapshotPath(id); const filePath = getSnapshotPath(id);
const stats = await fs.stat(filePath); const stats = await fs.stat(filePath);
if (state?.mtimeMs && stats.mtimeMs <= state.mtimeMs) return; if (!force && state?.mtimeMs && stats.mtimeMs <= state.mtimeMs) return;
const buffer = await fs.readFile(filePath); const buffer = await fs.readFile(filePath);
const ts = stats.mtimeMs || Date.now(); const ts = stats.mtimeMs || Date.now();
markState(id, { frame: buffer, ts, error: null, failures: 0, mtimeMs: stats.mtimeMs }); markState(id, { frame: buffer, ts, error: null, failures: 0, mtimeMs: stats.mtimeMs });
const count = (readCounts.get(id) || 0) + 1;
readCounts.set(id, count);
if (count === 1 || count % 30 === 0) {
logger.warn('Snapshot read ok', { id, count, bytes: buffer.length, ts, mtimeMs: stats.mtimeMs });
}
events.emit('frame', { id, buffer, ts }); events.emit('frame', { id, buffer, ts });
return { frame: buffer, ts };
} catch (err) { } catch (err) {
const failures = (state?.failures || 0) + 1; const failures = (state?.failures || 0) + 1;
const message = err.code === 'ENOENT' ? 'Snapshot missing' : err.message; const message = err.code === 'ENOENT' ? 'Snapshot missing' : err.message;
markState(id, { error: message, failures }); markState(id, { error: message, failures });
events.emit('status', { id, error: message }); events.emit('status', { id, error: message });
if (failures % 20 === 1) logger.warn('Snapshot read failed', { id, err: message }); if (failures % 20 === 1) logger.warn('Snapshot read failed', { id, err: message });
return null;
} finally { } finally {
markState(id, { fetching: false }); markState(id, { fetching: false });
} }
@@ -30,6 +30,7 @@ function registerRoverSnapshotSocketGateway({
const socketSubscriptions = new Map(); const socketSubscriptions = new Map();
const subscribeBuckets = new Map(); const subscribeBuckets = new Map();
const lastSentBySocket = new Map(); const lastSentBySocket = new Map();
const sentCounts = new Map();
function addSubscription(socket, roverId) { function addSubscription(socket, roverId) {
if (!roverSubscribers.has(roverId)) roverSubscribers.set(roverId, new Set()); if (!roverSubscribers.has(roverId)) roverSubscribers.set(roverId, new Set());
@@ -89,6 +90,12 @@ function registerRoverSnapshotSocketGateway({
const now = ts || Date.now(); const now = ts || Date.now();
if (now - lastSent < STREAM_INTERVAL_MS) return; if (now - lastSent < STREAM_INTERVAL_MS) return;
lastMap.set(id, now); lastMap.set(id, now);
const key = `${socketId}:${id}`;
const sentCount = (sentCounts.get(key) || 0) + 1;
sentCounts.set(key, sentCount);
if (sentCount === 1 || sentCount % 30 === 0) {
logger.warn('Snapshot frame sent', { socketId, roverId: id, sentCount, ts, bytes: buffer.length });
}
sendFrame(socket, id, { ts }, buffer); sendFrame(socket, id, { ts }, buffer);
}); });
}); });
@@ -126,11 +133,13 @@ function registerRoverSnapshotSocketGateway({
subscribedIds: validIds, subscribedIds: validIds,
}); });
validIds.forEach((roverId) => addSubscription(socket, roverId)); validIds.forEach((roverId) => addSubscription(socket, roverId));
validIds.forEach((roverId) => { validIds.forEach(async (roverId) => {
if (typeof fetchSnapshotNow === 'function') { if (typeof fetchSnapshotNow === 'function') {
fetchSnapshotNow(String(roverId)).catch((err) => { try {
await fetchSnapshotNow(String(roverId), { force: true });
} catch (err) {
logger.warn('Immediate snapshot fetch failed', { roverId, err: err.message }); logger.warn('Immediate snapshot fetch failed', { roverId, err: err.message });
}); }
} }
const state = getRoverSnapshotState(roverId); const state = getRoverSnapshotState(roverId);
if (state?.frame) sendFrame(socket, roverId, { ts: state.ts }, state.frame); if (state?.frame) sendFrame(socket, roverId, { ts: state.ts }, state.frame);