079project 1.0.0 → 3.0.0

package/loggerworker.cjs

@@ -0,0 +1,202 @@
+/* 日志子进程：接收主进程发来的 IO 记录与进程注册信息，按周期采样资源并输出文本表格 */
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { exec } = require('child_process');
+
+const LOG_DIR = path.join(__dirname, 'logs');
+fs.mkdirSync(LOG_DIR, { recursive: true });
+
+let watch = new Map(); // key: port -> { pid, groupId, role, port }
+let prevSample = new Map(); // pid -> { cpuSec, readBytes, writeBytes, ts }
+let lastMetricsWriteHour = '';
+let lastIoWriteHour = '';
+let sampleIntervalMs = 5000;
+let cores = os.cpus().length;
+
+// 工具：获取当前小时文件名
+function hourSuffix() {
+  const d = new Date();
+  const y = d.getFullYear();
+  const m = String(d.getMonth() + 1).padStart(2, '0');
+  const dd = String(d.getDate()).padStart(2, '0');
+  const hh = String(d.getHours()).padStart(2, '0');
+  return `${y}${m}${dd}_${hh}`;
+}
+function ioLogPath() { return path.join(LOG_DIR, `io_${hourSuffix()}.log`); }
+function metricsLogPath() { return path.join(LOG_DIR, `metrics_${hourSuffix()}.txt`); }
+
+function appendLine(file, line) {
+  fs.appendFile(file, line + '\n', () => {});
+}
+
+function writeIo(record) {
+  const nowHour = hourSuffix();
+  if (lastIoWriteHour !== nowHour) lastIoWriteHour = nowHour;
+  const file = ioLogPath();
+  appendLine(file, JSON.stringify(record));
+}
+function getProcInfoPowershell(pids) {
+  return new Promise((resolve) => {
+    if (!pids.length) return resolve([]);
+    const pidList = pids.join(',');
+    const cmd = [
+      'powershell.exe -NoProfile -Command',
+      `"Get-Process -Id ${pidList} | Select-Object Id,CPU,WorkingSet64,Name,`,
+      'IOReadBytes,IOWriteBytes | ConvertTo-Json -Compress"'
+    ].join(' ');
+    exec(cmd, { windowsHide: true, maxBuffer: 10 * 1024 * 1024 }, (err, stdout) => {
+      if (err || !stdout) return resolve([]);
+      try {
+        const data = JSON.parse(stdout);
+        resolve(Array.isArray(data) ? data : [data]);
+      } catch {
+        resolve([]);
+      }
+    });
+  });
+}
+
+function getTcpConnCounts() {
+  return new Promise((resolve) => {
+    // 使用 netstat -ano 统计 Owning PID 的 TCP 连接数量
+    exec('netstat -ano', { windowsHide: true, maxBuffer: 10 * 1024 * 1024 }, (err, stdout) => {
+      if (err || !stdout) return resolve(new Map());
+      const lines = stdout.split(/\r?\n/).filter(Boolean);
+      const map = new Map(); // pid -> count
+      for (const line of lines) {
+        // Proto  Local Address  Foreign Address  State  PID
+        const parts = line.trim().split(/\s+/);
+        if (parts.length < 5) continue;
+        const pid = Number(parts[parts.length - 1]);
+        if (!Number.isFinite(pid)) continue;
+        map.set(pid, (map.get(pid) || 0) + 1);
+      }
+      resolve(map);
+    });
+  });
+}
+
+function getListeningPortsByPid() {
+  return new Promise((resolve) => {
+    exec('netstat -ano -p TCP', { windowsHide: true, maxBuffer: 10 * 1024 * 1024 }, (err, stdout) => {
+      const map = new Map(); // pid -> Set(ports)
+      if (err || !stdout) return resolve(map);
+      const lines = stdout.split(/\r?\n/).filter(Boolean);
+      for (const line of lines) {
+        const parts = line.trim().split(/\s+/);
+        if (parts.length < 5) continue;
+        const state = parts[3] || parts[4] || '';
+        const pid = Number(parts[parts.length - 1]);
+        if (!Number.isFinite(pid)) continue;
+        const local = parts[1] || '';
+        const port = Number(local.split(':').pop());
+        if (!Number.isFinite(port)) continue;
+        if (!map.has(pid)) map.set(pid, new Set());
+        if (state.toUpperCase().includes('LISTEN')) {
+          map.get(pid).add(port);
+        }
+      }
+      resolve(map);
+    });
+  });
+}
+
+async function sampleAndWrite() {
+  try {
+    const ports = Array.from(watch.keys());
+    const procs = Array.from(watch.values());
+    const pids = Array.from(new Set(procs.map(p => p.pid))).filter(Boolean);
+
+    const [procInfo, connCounts, listenMap] = await Promise.all([
+      getProcInfoPowershell(pids),
+      getTcpConnCounts(),
+      getListeningPortsByPid()
+    ]);
+
+    const infoByPid = new Map();
+    for (const it of procInfo) {
+      infoByPid.set(Number(it.Id), {
+        cpuSec: Number(it.CPU || 0),
+        rss: Number(it.WorkingSet64 || 0),
+        ioRead: Number(it.IOReadBytes || 0),
+        ioWrite: Number(it.IOWriteBytes || 0),
+        name: it.Name || ''
+      });
+    }
+
+    const nowTs = Date.now();
+    const rows = [];
+    let aliveCount = 0;
+
+    for (const port of ports) {
+      const { pid, role, groupId } = watch.get(port);
+      const info = infoByPid.get(pid) || { cpuSec: 0, rss: 0, ioRead: 0, ioWrite: 0, name: '' };
+      const prev = prevSample.get(pid) || { cpuSec: info.cpuSec, readBytes: info.ioRead, writeBytes: info.ioWrite, ts: nowTs };
+      const dt = Math.max(1, (nowTs - prev.ts) / 1000); // sec
+      const dCpu = Math.max(0, info.cpuSec - prev.cpuSec); // sec
+      const cpuPct = Math.min(100, (dCpu / dt) * (100 / Math.max(1, 1 / cores))); // 近似
+      const dRead = Math.max(0, info.ioRead - prev.readBytes);
+      const dWrite = Math.max(0, info.ioWrite - prev.writeBytes);
+      const conn = connCounts.get(pid) || 0;
+      const rssMb = (info.rss / 1024 / 1024).toFixed(1);
+      const rps = (dRead / dt) | 0;
+      const wps = (dWrite / dt) | 0;
+      const listeningPorts = listenMap.get(pid) || new Set();
+      const up = listeningPorts.has(port) ? 'Y' : 'N';
+      if (up === 'Y') aliveCount++;
+
+      rows.push({
+        port,
+        pid,
+        role,
+        groupId,
+        cpu: cpuPct.toFixed(1),
+        mem: rssMb,
+        conn,
+        rps,
+        wps,
+        up
+      });
+
+      prevSample.set(pid, { cpuSec: info.cpuSec, readBytes: info.ioRead, writeBytes: info.ioWrite, ts: nowTs });
+    }
+
+    // 输出文本表
+    const nowHour = hourSuffix();
+    if (lastMetricsWriteHour !== nowHour) lastMetricsWriteHour = nowHour;
+    const file = metricsLogPath();
+    const header = `# ${new Date().toISOString()}  AlivePorts=${aliveCount}/${ports.length}\n` +
+      'PORT   PID      ROLE       GID  CPU%  MEM(MB)  CONN  DiskR/s  DiskW/s  UP\n' +
+      '-----  -------  ---------  ---  ----  -------  ----  -------  -------  --';
+    const lines = rows
+      .sort((a, b) => a.port - b.port)
+      .map(r =>
+        `${String(r.port).padEnd(5)}  ${String(r.pid).padEnd(7)}  ${String(r.role).padEnd(9)}  ${String(r.groupId).padEnd(3)}  ${String(r.cpu).padStart(4)}  ${String(r.mem).padStart(7)}  ${String(r.conn).padStart(4)}  ${String(r.rps).padStart(7)}  ${String(r.wps).padStart(7)}  ${r.up}`
+      );
+    appendLine(file, [header, ...lines, ''].join('\n'));
+  } catch (e) {
+    // 忽略单次采样错误
+  }
+}
+
+process.on('message', (msg) => {
+  if (!msg || typeof msg !== 'object') return;
+  if (msg.type === 'register' && Array.isArray(msg.processes)) {
+    for (const p of msg.processes) {
+      if (p && p.port) {
+        watch.set(p.port, { pid: p.pid, role: p.role, groupId: p.groupId, port: p.port });
+      }
+    }
+  } else if (msg.type === 'update' && msg.port) {
+    const old = watch.get(msg.port) || {};
+    watch.set(msg.port, { ...old, ...msg });
+  } else if (msg.type === 'io') {
+    // { ts, reqId, input, selectedPorts, responses[] }
+    writeIo(msg);
+  } else if (msg.type === 'config' && msg.sampleIntervalMs) {
+    sampleIntervalMs = Math.max(1000, Number(msg.sampleIntervalMs) || sampleIntervalMs);
+  }
+});
+
+setInterval(sampleAndWrite, sampleIntervalMs);