114 lines
3.4 KiB
JavaScript
114 lines
3.4 KiB
JavaScript
const express = require('express');
|
|
const http = require('http');
|
|
const WebSocket = require('ws');
|
|
const axios = require('axios');
|
|
const path = require('path');
|
|
|
|
const argv = require('yargs').argv;
|
|
const app = express();
|
|
const server = http.createServer(app);
|
|
const wss = new WebSocket.Server({ server });
|
|
|
|
let logCache = new Set();
|
|
|
|
app.use(express.static(path.join(__dirname, 'public')));
|
|
|
|
wss.on('connection', (ws) => {
|
|
console.log('Client connected:', ws._socket.remoteAddress);
|
|
});
|
|
|
|
const debugMode = argv.debug || true;
|
|
|
|
setInterval(async () => {
|
|
let logs;
|
|
if (debugMode) {
|
|
try {
|
|
const fs = require('fs');
|
|
const data = fs.readFileSync('logsexample.out', 'utf8');
|
|
const logLines = data.split('\n');
|
|
const validLogs = [];
|
|
logLines.forEach((line, index) => {
|
|
if (line.trim() !== '') {
|
|
try {
|
|
const parsedLog = JSON.parse(line);
|
|
validLogs.push(parsedLog);
|
|
} catch (parseError) {
|
|
console.error(`Error parsing log line at index ${index}:`, line);
|
|
}
|
|
}
|
|
});
|
|
logs = validLogs;
|
|
} catch (error) {
|
|
console.error('Error reading logsexample.out:', error);
|
|
return;
|
|
}
|
|
} else {
|
|
const query = 'query=_time:4s relay received';
|
|
try {
|
|
const response = await axios.post('https://vmselect.riff.cc/select/logsql/query', {
|
|
data: query,
|
|
headers: {
|
|
'Content-Type': 'application/x-www-form-urlencoded'
|
|
}
|
|
});
|
|
const logLines = response.data.split('\n');
|
|
const validLogs = [];
|
|
logLines.forEach((line, index) => {
|
|
if (line.trim() !== '') {
|
|
try {
|
|
const unescapedLine = line.replace(/\\"/g, '"');
|
|
const parsedLog = JSON.parse(unescapedLine);
|
|
validLogs.push(parsedLog);
|
|
} catch (parseError) {
|
|
console.error(`Error parsing log line at index ${index}:`, line);
|
|
}
|
|
}
|
|
});
|
|
logs = validLogs;
|
|
console.log('Logs response:', logs);
|
|
|
|
if (Array.isArray(logs)) {
|
|
logs.forEach(log => {
|
|
const msgMatch = log._msg.match(/msg_hash=0x[0-9a-fA-F]+/);
|
|
const timeMatch = log._msg.match(/receivedTime=\d+/);
|
|
const nodeIdMatch = log.kubernetes_pod_name.match(/nodes-(\d+)/);
|
|
|
|
if (msgMatch && timeMatch && nodeIdMatch) {
|
|
const msg_hash = msgMatch[0].split('=')[1];
|
|
const receivedTime = timeMatch[0].split('=')[1];
|
|
const nodeId = parseInt(nodeIdMatch[1], 10);
|
|
|
|
const logIdentifier = `${msg_hash}-${receivedTime}`;
|
|
if (!logCache.has(logIdentifier)) {
|
|
logCache.add(logIdentifier);
|
|
const logData = {
|
|
msg_hash,
|
|
receivedTime,
|
|
nodeId,
|
|
newNode: !logCache.has(`node-${nodeId}`)
|
|
};
|
|
|
|
logCache.add(`node-${nodeId}`);
|
|
|
|
console.log('Sending log data to clients:', logData);
|
|
wss.clients.forEach(client => {
|
|
if (client.readyState === WebSocket.OPEN) {
|
|
client.send(JSON.stringify(logData));
|
|
}
|
|
});
|
|
}
|
|
}
|
|
});
|
|
} else {
|
|
console.error('Logs response is not an array:', logs);
|
|
}
|
|
} catch (error) {
|
|
console.error('Error querying logs:', error);
|
|
}
|
|
}
|
|
}, 3000);
|
|
|
|
server.listen(3000, () => {
|
|
console.log('Server is listening on port 3000');
|
|
});
|